← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/memory.py

  • Committer: Alexander Belchenko
  • Date: 2006-07-30 16:43:12 UTC
  • mto: (1711.2.111 jam-integration)
  • mto: This revision was merged to the branch mainline in revision 1906.
  • Revision ID: bialix@ukr.net-20060730164312-b025fd3ff0cee59e
rename  gpl.txt => COPYING.txt

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/memory.py
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
 
20
20
so this is primarily useful for testing.
21
21
"""
22
22
 
 
23
from copy import copy
23
24
import os
24
25
import errno
25
26
import re
27
28
from cStringIO import StringIO
28
29
import warnings
29
30
 
30
 
from bzrlib import (
31
 
    urlutils,
32
 
    )
33
 
from bzrlib.errors import (
34
 
    FileExists,
35
 
    LockError,
36
 
    InProcessTransport,
37
 
    NoSuchFile,
38
 
    TransportError,
39
 
    )
 
31
from bzrlib.errors import TransportError, NoSuchFile, FileExists, LockError
40
32
from bzrlib.trace import mutter
41
 
from bzrlib.transport import (
42
 
    AppendBasedFileStream,
43
 
    _file_streams,
44
 
    LateReadError,
45
 
    register_transport,
46
 
    Server,
47
 
    Transport,
48
 
    unregister_transport,
49
 
    )
 
33
from bzrlib.transport import (Transport, register_transport, Server)
 
34
import bzrlib.urlutils as urlutils
50
35
 
51
36
 
52
37
 
74
59
        if url[-1] != '/':
75
60
            url = url + '/'
76
61
        super(MemoryTransport, self).__init__(url)
77
 
        split = url.find(':') + 3
78
 
        self._scheme = url[:split]
79
 
        self._cwd = url[split:]
 
62
        self._cwd = url[url.find(':') + 3:]
80
63
        # dictionaries from absolute path to file mode
81
64
        self._dirs = {'/':None}
82
65
        self._files = {}
84
67
 
85
68
    def clone(self, offset=None):
86
69
        """See Transport.clone()."""
87
 
        path = self._combine_paths(self._cwd, offset)
88
 
        if len(path) == 0 or path[-1] != '/':
89
 
            path += '/'
90
 
        url = self._scheme + path
91
 
        result = self.__class__(url)
 
70
        if offset is None or offset == '':
 
71
            return copy(self)
 
72
        segments = offset.split('/')
 
73
        cwdsegments = self._cwd.split('/')[:-1]
 
74
        while len(segments):
 
75
            segment = segments.pop(0)
 
76
            if segment == '.':
 
77
                continue
 
78
            if segment == '..':
 
79
                if len(cwdsegments) > 1:
 
80
                    cwdsegments.pop()
 
81
                continue
 
82
            cwdsegments.append(segment)
 
83
        url = self.base[:self.base.find(':') + 3] + '/'.join(cwdsegments) + '/'
 
84
        result = MemoryTransport(url)
92
85
        result._dirs = self._dirs
93
86
        result._files = self._files
94
87
        result._locks = self._locks
105
98
        else:
106
99
            return temp_t.base[:-1]
107
100
 
108
 
    def append_file(self, relpath, f, mode=None):
109
 
        """See Transport.append_file()."""
 
101
    def append(self, relpath, f, mode=None):
 
102
        """See Transport.append()."""
110
103
        _abspath = self._abspath(relpath)
111
104
        self._check_parent(_abspath)
112
105
        orig_content, orig_mode = self._files.get(_abspath, ("", None))
124
117
    def has(self, relpath):
125
118
        """See Transport.has()."""
126
119
        _abspath = self._abspath(relpath)
127
 
        return (_abspath in self._files) or (_abspath in self._dirs)
 
120
        return _abspath in self._files or _abspath in self._dirs
128
121
 
129
122
    def delete(self, relpath):
130
123
        """See Transport.delete()."""
133
126
            raise NoSuchFile(relpath)
134
127
        del self._files[_abspath]
135
128
 
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
129
    def get(self, relpath):
143
130
        """See Transport.get()."""
144
131
        _abspath = self._abspath(relpath)
145
132
        if not _abspath in self._files:
146
 
            if _abspath in self._dirs:
147
 
                return LateReadError(relpath)
148
 
            else:
149
 
                raise NoSuchFile(relpath)
 
133
            raise NoSuchFile(relpath)
150
134
        return StringIO(self._files[_abspath][0])
151
135
 
152
 
    def put_file(self, relpath, f, mode=None):
153
 
        """See Transport.put_file()."""
 
136
    def put(self, relpath, f, mode=None):
 
137
        """See Transport.put()."""
154
138
        _abspath = self._abspath(relpath)
155
139
        self._check_parent(_abspath)
156
 
        bytes = f.read()
157
 
        if type(bytes) is not str:
158
 
            # Although not strictly correct, we raise UnicodeEncodeError to be
159
 
            # compatible with other transports.
160
 
            raise UnicodeEncodeError(
161
 
                'undefined', bytes, 0, 1,
162
 
                'put_file must be given a file of bytes, not unicode.')
163
 
        self._files[_abspath] = (bytes, mode)
164
 
        return len(bytes)
 
140
        self._files[_abspath] = (f.read(), mode)
165
141
 
166
142
    def mkdir(self, relpath, mode=None):
167
143
        """See Transport.mkdir()."""
171
147
            raise FileExists(relpath)
172
148
        self._dirs[_abspath]=mode
173
149
 
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
150
    def listable(self):
182
151
        """See Transport.listable."""
183
152
        return True
185
154
    def iter_files_recursive(self):
186
155
        for file in self._files:
187
156
            if file.startswith(self._cwd):
188
 
                yield urlutils.escape(file[len(self._cwd):])
189
 
 
 
157
                yield file[len(self._cwd):]
 
158
    
190
159
    def list_dir(self, relpath):
191
160
        """See Transport.list_dir()."""
192
161
        _abspath = self._abspath(relpath)
193
162
        if _abspath != '/' and _abspath not in self._dirs:
194
163
            raise NoSuchFile(relpath)
195
164
        result = []
196
 
 
197
 
        if not _abspath.endswith('/'):
198
 
            _abspath += '/'
199
 
 
200
 
        for path_group in self._files, self._dirs:
201
 
            for path in path_group:
202
 
                if path.startswith(_abspath):
203
 
                    trailing = path[len(_abspath):]
204
 
                    if trailing and '/' not in trailing:
205
 
                        result.append(trailing)
206
 
        return map(urlutils.escape, result)
 
165
        for path in self._files:
 
166
            if (path.startswith(_abspath) and 
 
167
                path[len(_abspath) + 1:].find('/') == -1 and
 
168
                len(path) > len(_abspath)):
 
169
                result.append(path[len(_abspath) + 1:])
 
170
        for path in self._dirs:
 
171
            if (path.startswith(_abspath) and 
 
172
                path[len(_abspath) + 1:].find('/') == -1 and
 
173
                len(path) > len(_abspath) and
 
174
                path[len(_abspath)] == '/'):
 
175
                result.append(path[len(_abspath) + 1:])
 
176
        return result
207
177
 
208
178
    def rename(self, rel_from, rel_to):
209
179
        """Rename a file or directory; fail if the destination exists"""
225
195
                    del container[path]
226
196
        do_renames(self._files)
227
197
        do_renames(self._dirs)
228
 
 
 
198
    
229
199
    def rmdir(self, relpath):
230
200
        """See Transport.rmdir."""
231
201
        _abspath = self._abspath(relpath)
232
202
        if _abspath in self._files:
233
203
            self._translate_error(IOError(errno.ENOTDIR, relpath), relpath)
234
204
        for path in self._files:
235
 
            if path.startswith(_abspath + '/'):
 
205
            if path.startswith(_abspath):
236
206
                self._translate_error(IOError(errno.ENOTEMPTY, relpath),
237
207
                                      relpath)
238
208
        for path in self._dirs:
239
 
            if path.startswith(_abspath + '/') and path != _abspath:
 
209
            if path.startswith(_abspath) and path != _abspath:
240
210
                self._translate_error(IOError(errno.ENOTEMPTY, relpath), relpath)
241
211
        if not _abspath in self._dirs:
242
212
            raise NoSuchFile(relpath)
246
216
        """See Transport.stat()."""
247
217
        _abspath = self._abspath(relpath)
248
218
        if _abspath in self._files:
249
 
            return MemoryStat(len(self._files[_abspath][0]), False,
 
219
            return MemoryStat(len(self._files[_abspath][0]), False, 
250
220
                              self._files[_abspath][1])
251
221
        elif _abspath in self._dirs:
252
222
            return MemoryStat(0, True, self._dirs[_abspath])
264
234
    def _abspath(self, relpath):
265
235
        """Generate an internal absolute path."""
266
236
        relpath = urlutils.unescape(relpath)
267
 
        if relpath[:1] == '/':
268
 
            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)
 
237
        if relpath.find('..') != -1:
 
238
            raise AssertionError('relpath contains ..')
 
239
        if relpath == '.':
 
240
            if (self._cwd == '/'):
 
241
                return self._cwd
 
242
            return self._cwd[:-1]
 
243
        if relpath.endswith('/'):
 
244
            relpath = relpath[:-1]
 
245
        if relpath.startswith('./'):
 
246
            relpath = relpath[2:]
 
247
        return self._cwd + relpath
283
248
 
284
249
 
285
250
class _MemoryLock(object):
286
251
    """This makes a lock."""
287
252
 
288
253
    def __init__(self, path, transport):
 
254
        assert isinstance(transport, MemoryTransport)
289
255
        self.path = path
290
256
        self.transport = transport
291
257
        if self.path in self.transport._locks:
306
272
class MemoryServer(Server):
307
273
    """Server for the MemoryTransport for testing with."""
308
274
 
309
 
    def start_server(self):
 
275
    def setUp(self):
 
276
        """See bzrlib.transport.Server.setUp."""
310
277
        self._dirs = {'/':None}
311
278
        self._files = {}
312
279
        self._locks = {}
317
284
            result._files = self._files
318
285
            result._locks = self._locks
319
286
            return result
320
 
        self._memory_factory = memory_factory
321
 
        register_transport(self._scheme, self._memory_factory)
 
287
        register_transport(self._scheme, memory_factory)
322
288
 
323
 
    def stop_server(self):
 
289
    def tearDown(self):
 
290
        """See bzrlib.transport.Server.tearDown."""
324
291
        # unregister this server
325
 
        unregister_transport(self._scheme, self._memory_factory)
326
292
 
327
293
    def get_url(self):
328
294
        """See bzrlib.transport.Server.get_url."""