← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/memory.py

  • Committer: Ian Clatworthy
  • Date: 2007-08-13 14:33:10 UTC
  • mto: (2733.1.1 ianc-integration)
  • mto: This revision was merged to the branch mainline in revision 2734.
  • Revision ID: ian.clatworthy@internode.on.net-20070813143310-twhj4la0qnupvze8
Added Quick Start Summary

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/memory.py
1
 
# Copyright (C) 2005-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
 
    transport,
32
 
    urlutils,
33
 
    )
34
30
from bzrlib.errors import (
35
31
    FileExists,
36
32
    LockError,
40
36
    )
41
37
from bzrlib.trace import mutter
42
38
from bzrlib.transport import (
43
 
    AppendBasedFileStream,
44
 
    _file_streams,
45
39
    LateReadError,
 
40
    register_transport,
 
41
    Server,
 
42
    Transport,
46
43
    )
 
44
import bzrlib.urlutils as urlutils
47
45
 
48
46
 
49
47
 
61
59
            self.st_mode = S_IFDIR | perms
62
60
 
63
61
 
64
 
class MemoryTransport(transport.Transport):
 
62
class MemoryTransport(Transport):
65
63
    """This is an in memory file system for transient data storage."""
66
64
 
67
65
    def __init__(self, url=""):
81
79
 
82
80
    def clone(self, offset=None):
83
81
        """See Transport.clone()."""
84
 
        path = urlutils.URL._combine_paths(self._cwd, offset)
 
82
        path = self._combine_paths(self._cwd, offset)
85
83
        if len(path) == 0 or path[-1] != '/':
86
84
            path += '/'
87
85
        url = self._scheme + path
88
 
        result = self.__class__(url)
 
86
        result = MemoryTransport(url)
89
87
        result._dirs = self._dirs
90
88
        result._files = self._files
91
89
        result._locks = self._locks
158
156
                'undefined', bytes, 0, 1,
159
157
                'put_file must be given a file of bytes, not unicode.')
160
158
        self._files[_abspath] = (bytes, mode)
161
 
        return len(bytes)
162
159
 
163
160
    def mkdir(self, relpath, mode=None):
164
161
        """See Transport.mkdir()."""
168
165
            raise FileExists(relpath)
169
166
        self._dirs[_abspath]=mode
170
167
 
171
 
    def open_write_stream(self, relpath, mode=None):
172
 
        """See Transport.open_write_stream."""
173
 
        self.put_bytes(relpath, "", mode)
174
 
        result = AppendBasedFileStream(self, relpath)
175
 
        _file_streams[self.abspath(relpath)] = result
176
 
        return result
177
 
 
178
168
    def listable(self):
179
169
        """See Transport.listable."""
180
170
        return True
183
173
        for file in self._files:
184
174
            if file.startswith(self._cwd):
185
175
                yield urlutils.escape(file[len(self._cwd):])
186
 
 
 
176
    
187
177
    def list_dir(self, relpath):
188
178
        """See Transport.list_dir()."""
189
179
        _abspath = self._abspath(relpath)
222
212
                    del container[path]
223
213
        do_renames(self._files)
224
214
        do_renames(self._dirs)
225
 
 
 
215
    
226
216
    def rmdir(self, relpath):
227
217
        """See Transport.rmdir."""
228
218
        _abspath = self._abspath(relpath)
243
233
        """See Transport.stat()."""
244
234
        _abspath = self._abspath(relpath)
245
235
        if _abspath in self._files:
246
 
            return MemoryStat(len(self._files[_abspath][0]), False,
 
236
            return MemoryStat(len(self._files[_abspath][0]), False, 
247
237
                              self._files[_abspath][1])
248
238
        elif _abspath in self._dirs:
249
239
            return MemoryStat(0, True, self._dirs[_abspath])
261
251
    def _abspath(self, relpath):
262
252
        """Generate an internal absolute path."""
263
253
        relpath = urlutils.unescape(relpath)
264
 
        if relpath[:1] == '/':
 
254
        if relpath.find('..') != -1:
 
255
            raise AssertionError('relpath contains ..')
 
256
        if relpath == '':
 
257
            return '/'
 
258
        if relpath[0] == '/':
265
259
            return relpath
266
 
        cwd_parts = self._cwd.split('/')
267
 
        rel_parts = relpath.split('/')
268
 
        r = []
269
 
        for i in cwd_parts + rel_parts:
270
 
            if i == '..':
271
 
                if not r:
272
 
                    raise ValueError("illegal relpath %r under %r"
273
 
                        % (relpath, self._cwd))
274
 
                r = r[:-1]
275
 
            elif i == '.' or i == '':
276
 
                pass
277
 
            else:
278
 
                r.append(i)
279
 
        return '/' + '/'.join(r)
 
260
        if relpath == '.':
 
261
            if (self._cwd == '/'):
 
262
                return self._cwd
 
263
            return self._cwd[:-1]
 
264
        if relpath.endswith('/'):
 
265
            relpath = relpath[:-1]
 
266
        if relpath.startswith('./'):
 
267
            relpath = relpath[2:]
 
268
        return self._cwd + relpath
280
269
 
281
270
 
282
271
class _MemoryLock(object):
283
272
    """This makes a lock."""
284
273
 
285
274
    def __init__(self, path, transport):
 
275
        assert isinstance(transport, MemoryTransport)
286
276
        self.path = path
287
277
        self.transport = transport
288
278
        if self.path in self.transport._locks:
289
279
            raise LockError('File %r already locked' % (self.path,))
290
280
        self.transport._locks[self.path] = self
291
281
 
 
282
    def __del__(self):
 
283
        # Should this warn, or actually try to cleanup?
 
284
        if self.transport:
 
285
            warnings.warn("MemoryLock %r not explicitly unlocked" % (self.path,))
 
286
            self.unlock()
 
287
 
292
288
    def unlock(self):
293
289
        del self.transport._locks[self.path]
294
290
        self.transport = None
295
291
 
296
292
 
297
 
class MemoryServer(transport.Server):
 
293
class MemoryServer(Server):
298
294
    """Server for the MemoryTransport for testing with."""
299
295
 
300
 
    def start_server(self):
 
296
    def setUp(self):
 
297
        """See bzrlib.transport.Server.setUp."""
301
298
        self._dirs = {'/':None}
302
299
        self._files = {}
303
300
        self._locks = {}
304
301
        self._scheme = "memory+%s:///" % id(self)
305
302
        def memory_factory(url):
306
 
            from bzrlib.transport import memory
307
 
            result = memory.MemoryTransport(url)
 
303
            result = MemoryTransport(url)
308
304
            result._dirs = self._dirs
309
305
            result._files = self._files
310
306
            result._locks = self._locks
311
307
            return result
312
 
        self._memory_factory = memory_factory
313
 
        transport.register_transport(self._scheme, self._memory_factory)
 
308
        register_transport(self._scheme, memory_factory)
314
309
 
315
 
    def stop_server(self):
 
310
    def tearDown(self):
 
311
        """See bzrlib.transport.Server.tearDown."""
316
312
        # unregister this server
317
 
        transport.unregister_transport(self._scheme, self._memory_factory)
318
313
 
319
314
    def get_url(self):
320
315
        """See bzrlib.transport.Server.get_url."""
321
316
        return self._scheme
322
317
 
323
 
    def get_bogus_url(self):
324
 
        raise NotImplementedError
325
 
 
326
318
 
327
319
def get_test_permutations():
328
320
    """Return the permutations to be used in testing."""