~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/memory.py

  • Committer: Aaron Bentley
  • Date: 2007-07-17 13:27:14 UTC
  • mfrom: (2624 +trunk)
  • mto: This revision was merged to the branch mainline in revision 2631.
  • Revision ID: abentley@panoramicfeedback.com-20070717132714-tmzx9khmg9501k51
Merge from bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2008, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Implementation of Transport that uses memory for its storage.
18
18
 
27
27
from cStringIO import StringIO
28
28
import warnings
29
29
 
30
 
from bzrlib import (
31
 
    urlutils,
32
 
    )
33
 
from bzrlib.errors import (
34
 
    FileExists,
35
 
    LockError,
36
 
    InProcessTransport,
37
 
    NoSuchFile,
38
 
    TransportError,
39
 
    )
 
30
from bzrlib.errors import TransportError, NoSuchFile, FileExists, LockError
40
31
from bzrlib.trace import mutter
41
32
from bzrlib.transport import (
42
 
    AppendBasedFileStream,
43
 
    _file_streams,
44
33
    LateReadError,
45
34
    register_transport,
46
35
    Server,
47
36
    Transport,
48
 
    unregister_transport,
49
37
    )
 
38
import bzrlib.urlutils as urlutils
50
39
 
51
40
 
52
41
 
88
77
        if len(path) == 0 or path[-1] != '/':
89
78
            path += '/'
90
79
        url = self._scheme + path
91
 
        result = self.__class__(url)
 
80
        result = MemoryTransport(url)
92
81
        result._dirs = self._dirs
93
82
        result._files = self._files
94
83
        result._locks = self._locks
133
122
            raise NoSuchFile(relpath)
134
123
        del self._files[_abspath]
135
124
 
136
 
    def external_url(self):
137
 
        """See bzrlib.transport.Transport.external_url."""
138
 
        # MemoryTransport's are only accessible in-process
139
 
        # so we raise here
140
 
        raise InProcessTransport(self)
141
 
 
142
125
    def get(self, relpath):
143
126
        """See Transport.get()."""
144
127
        _abspath = self._abspath(relpath)
161
144
                'undefined', bytes, 0, 1,
162
145
                'put_file must be given a file of bytes, not unicode.')
163
146
        self._files[_abspath] = (bytes, mode)
164
 
        return len(bytes)
165
147
 
166
148
    def mkdir(self, relpath, mode=None):
167
149
        """See Transport.mkdir()."""
171
153
            raise FileExists(relpath)
172
154
        self._dirs[_abspath]=mode
173
155
 
174
 
    def open_write_stream(self, relpath, mode=None):
175
 
        """See Transport.open_write_stream."""
176
 
        self.put_bytes(relpath, "", mode)
177
 
        result = AppendBasedFileStream(self, relpath)
178
 
        _file_streams[self.abspath(relpath)] = result
179
 
        return result
180
 
 
181
156
    def listable(self):
182
157
        """See Transport.listable."""
183
158
        return True
186
161
        for file in self._files:
187
162
            if file.startswith(self._cwd):
188
163
                yield urlutils.escape(file[len(self._cwd):])
189
 
 
 
164
    
190
165
    def list_dir(self, relpath):
191
166
        """See Transport.list_dir()."""
192
167
        _abspath = self._abspath(relpath)
225
200
                    del container[path]
226
201
        do_renames(self._files)
227
202
        do_renames(self._dirs)
228
 
 
 
203
    
229
204
    def rmdir(self, relpath):
230
205
        """See Transport.rmdir."""
231
206
        _abspath = self._abspath(relpath)
246
221
        """See Transport.stat()."""
247
222
        _abspath = self._abspath(relpath)
248
223
        if _abspath in self._files:
249
 
            return MemoryStat(len(self._files[_abspath][0]), False,
 
224
            return MemoryStat(len(self._files[_abspath][0]), False, 
250
225
                              self._files[_abspath][1])
251
226
        elif _abspath in self._dirs:
252
227
            return MemoryStat(0, True, self._dirs[_abspath])
264
239
    def _abspath(self, relpath):
265
240
        """Generate an internal absolute path."""
266
241
        relpath = urlutils.unescape(relpath)
267
 
        if relpath[:1] == '/':
 
242
        if relpath.find('..') != -1:
 
243
            raise AssertionError('relpath contains ..')
 
244
        if relpath == '':
 
245
            return '/'
 
246
        if relpath[0] == '/':
268
247
            return relpath
269
 
        cwd_parts = self._cwd.split('/')
270
 
        rel_parts = relpath.split('/')
271
 
        r = []
272
 
        for i in cwd_parts + rel_parts:
273
 
            if i == '..':
274
 
                if not r:
275
 
                    raise ValueError("illegal relpath %r under %r"
276
 
                        % (relpath, self._cwd))
277
 
                r = r[:-1]
278
 
            elif i == '.' or i == '':
279
 
                pass
280
 
            else:
281
 
                r.append(i)
282
 
        return '/' + '/'.join(r)
 
248
        if relpath == '.':
 
249
            if (self._cwd == '/'):
 
250
                return self._cwd
 
251
            return self._cwd[:-1]
 
252
        if relpath.endswith('/'):
 
253
            relpath = relpath[:-1]
 
254
        if relpath.startswith('./'):
 
255
            relpath = relpath[2:]
 
256
        return self._cwd + relpath
283
257
 
284
258
 
285
259
class _MemoryLock(object):
286
260
    """This makes a lock."""
287
261
 
288
262
    def __init__(self, path, transport):
 
263
        assert isinstance(transport, MemoryTransport)
289
264
        self.path = path
290
265
        self.transport = transport
291
266
        if self.path in self.transport._locks:
306
281
class MemoryServer(Server):
307
282
    """Server for the MemoryTransport for testing with."""
308
283
 
309
 
    def start_server(self):
 
284
    def setUp(self):
 
285
        """See bzrlib.transport.Server.setUp."""
310
286
        self._dirs = {'/':None}
311
287
        self._files = {}
312
288
        self._locks = {}
317
293
            result._files = self._files
318
294
            result._locks = self._locks
319
295
            return result
320
 
        self._memory_factory = memory_factory
321
 
        register_transport(self._scheme, self._memory_factory)
 
296
        register_transport(self._scheme, memory_factory)
322
297
 
323
 
    def stop_server(self):
 
298
    def tearDown(self):
 
299
        """See bzrlib.transport.Server.tearDown."""
324
300
        # unregister this server
325
 
        unregister_transport(self._scheme, self._memory_factory)
326
301
 
327
302
    def get_url(self):
328
303
        """See bzrlib.transport.Server.get_url."""