← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/memory.py

  • Committer: Martin Pool
  • Date: 2006-04-20 02:37:21 UTC
  • mto: This revision was merged to the branch mainline in revision 1675.
  • Revision ID: mbp@sourcefrog.net-20060420023721-04d8a3b015987240
Add .hg to default ignore list

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/memory.py
1
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
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
26
 
from stat import S_IFREG, S_IFDIR
 
27
from stat import *
27
28
from cStringIO import StringIO
28
 
import warnings
29
29
 
30
 
from bzrlib.errors import (
31
 
    FileExists,
32
 
    LockError,
33
 
    InProcessTransport,
34
 
    NoSuchFile,
35
 
    TransportError,
36
 
    )
37
30
from bzrlib.trace import mutter
38
 
from bzrlib.transport import (
39
 
    AppendBasedFileStream,
40
 
    _file_streams,
41
 
    LateReadError,
42
 
    register_transport,
43
 
    Server,
44
 
    Transport,
45
 
    )
46
 
import bzrlib.urlutils as urlutils
47
 
 
 
31
from bzrlib.errors import TransportError, NoSuchFile, FileExists, LockError
 
32
from bzrlib.transport import Transport, register_transport, Server
48
33
 
49
34
 
50
35
class MemoryStat(object):
67
52
    def __init__(self, url=""):
68
53
        """Set the 'base' path where files will be stored."""
69
54
        if url == "":
70
 
            url = "memory:///"
 
55
            url = "memory:/"
71
56
        if url[-1] != '/':
72
57
            url = url + '/'
73
58
        super(MemoryTransport, self).__init__(url)
74
 
        split = url.find(':') + 3
75
 
        self._scheme = url[:split]
76
 
        self._cwd = url[split:]
 
59
        self._cwd = url[url.find(':') + 1:]
77
60
        # dictionaries from absolute path to file mode
78
 
        self._dirs = {'/':None}
 
61
        self._dirs = {}
79
62
        self._files = {}
80
63
        self._locks = {}
81
64
 
82
65
    def clone(self, offset=None):
83
66
        """See Transport.clone()."""
84
 
        path = self._combine_paths(self._cwd, offset)
85
 
        if len(path) == 0 or path[-1] != '/':
86
 
            path += '/'
87
 
        url = self._scheme + path
 
67
        if offset is None or offset == '':
 
68
            return copy(self)
 
69
        segments = offset.split('/')
 
70
        cwdsegments = self._cwd.split('/')[:-1]
 
71
        while len(segments):
 
72
            segment = segments.pop(0)
 
73
            if segment == '.':
 
74
                continue
 
75
            if segment == '..':
 
76
                if len(cwdsegments) > 1:
 
77
                    cwdsegments.pop()
 
78
                continue
 
79
            cwdsegments.append(segment)
 
80
        url = self.base[:self.base.find(':') + 1] + '/'.join(cwdsegments) + '/'
88
81
        result = MemoryTransport(url)
89
82
        result._dirs = self._dirs
90
83
        result._files = self._files
97
90
        # current environment - XXX RBC 20060404 move the clone '..' handling
98
91
        # into here and call abspath from clone
99
92
        temp_t = self.clone(relpath)
100
 
        if temp_t.base.count('/') == 3:
 
93
        if temp_t.base.count('/') == 1:
101
94
            return temp_t.base
102
95
        else:
103
96
            return temp_t.base[:-1]
104
97
 
105
 
    def append_file(self, relpath, f, mode=None):
106
 
        """See Transport.append_file()."""
 
98
    def append(self, relpath, f):
 
99
        """See Transport.append()."""
107
100
        _abspath = self._abspath(relpath)
108
101
        self._check_parent(_abspath)
109
102
        orig_content, orig_mode = self._files.get(_abspath, ("", None))
110
 
        if mode is None:
111
 
            mode = orig_mode
112
 
        self._files[_abspath] = (orig_content + f.read(), mode)
 
103
        self._files[_abspath] = (orig_content + f.read(), orig_mode)
113
104
        return len(orig_content)
114
105
 
115
106
    def _check_parent(self, _abspath):
121
112
    def has(self, relpath):
122
113
        """See Transport.has()."""
123
114
        _abspath = self._abspath(relpath)
124
 
        return (_abspath in self._files) or (_abspath in self._dirs)
 
115
        return _abspath in self._files or _abspath in self._dirs
125
116
 
126
117
    def delete(self, relpath):
127
118
        """See Transport.delete()."""
130
121
            raise NoSuchFile(relpath)
131
122
        del self._files[_abspath]
132
123
 
133
 
    def external_url(self):
134
 
        """See bzrlib.transport.Transport.external_url."""
135
 
        # MemoryTransport's are only accessible in-process
136
 
        # so we raise here
137
 
        raise InProcessTransport(self)
138
 
 
139
124
    def get(self, relpath):
140
125
        """See Transport.get()."""
141
126
        _abspath = self._abspath(relpath)
142
127
        if not _abspath in self._files:
143
 
            if _abspath in self._dirs:
144
 
                return LateReadError(relpath)
145
 
            else:
146
 
                raise NoSuchFile(relpath)
 
128
            raise NoSuchFile(relpath)
147
129
        return StringIO(self._files[_abspath][0])
148
130
 
149
 
    def put_file(self, relpath, f, mode=None):
150
 
        """See Transport.put_file()."""
 
131
    def put(self, relpath, f, mode=None):
 
132
        """See Transport.put()."""
151
133
        _abspath = self._abspath(relpath)
152
134
        self._check_parent(_abspath)
153
 
        bytes = f.read()
154
 
        if type(bytes) is not str:
155
 
            # Although not strictly correct, we raise UnicodeEncodeError to be
156
 
            # compatible with other transports.
157
 
            raise UnicodeEncodeError(
158
 
                'undefined', bytes, 0, 1,
159
 
                'put_file must be given a file of bytes, not unicode.')
160
 
        self._files[_abspath] = (bytes, mode)
161
 
        return len(bytes)
 
135
        self._files[_abspath] = (f.read(), mode)
162
136
 
163
137
    def mkdir(self, relpath, mode=None):
164
138
        """See Transport.mkdir()."""
168
142
            raise FileExists(relpath)
169
143
        self._dirs[_abspath]=mode
170
144
 
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
145
    def listable(self):
179
146
        """See Transport.listable."""
180
147
        return True
182
149
    def iter_files_recursive(self):
183
150
        for file in self._files:
184
151
            if file.startswith(self._cwd):
185
 
                yield urlutils.escape(file[len(self._cwd):])
 
152
                yield file[len(self._cwd):]
186
153
    
187
154
    def list_dir(self, relpath):
188
155
        """See Transport.list_dir()."""
190
157
        if _abspath != '/' and _abspath not in self._dirs:
191
158
            raise NoSuchFile(relpath)
192
159
        result = []
193
 
 
194
 
        if not _abspath.endswith('/'):
195
 
            _abspath += '/'
196
 
 
197
 
        for path_group in self._files, self._dirs:
198
 
            for path in path_group:
199
 
                if path.startswith(_abspath):
200
 
                    trailing = path[len(_abspath):]
201
 
                    if trailing and '/' not in trailing:
202
 
                        result.append(trailing)
203
 
        return map(urlutils.escape, result)
 
160
        for path in self._files:
 
161
            if (path.startswith(_abspath) and 
 
162
                path[len(_abspath) + 1:].find('/') == -1 and
 
163
                len(path) > len(_abspath)):
 
164
                result.append(path[len(_abspath) + 1:])
 
165
        for path in self._dirs:
 
166
            if (path.startswith(_abspath) and 
 
167
                path[len(_abspath) + 1:].find('/') == -1 and
 
168
                len(path) > len(_abspath) and
 
169
                path[len(_abspath)] == '/'):
 
170
                result.append(path[len(_abspath) + 1:])
 
171
        return result
204
172
 
205
173
    def rename(self, rel_from, rel_to):
206
174
        """Rename a file or directory; fail if the destination exists"""
229
197
        if _abspath in self._files:
230
198
            self._translate_error(IOError(errno.ENOTDIR, relpath), relpath)
231
199
        for path in self._files:
232
 
            if path.startswith(_abspath + '/'):
 
200
            if path.startswith(_abspath):
233
201
                self._translate_error(IOError(errno.ENOTEMPTY, relpath),
234
202
                                      relpath)
235
203
        for path in self._dirs:
236
 
            if path.startswith(_abspath + '/') and path != _abspath:
 
204
            if path.startswith(_abspath) and path != _abspath:
237
205
                self._translate_error(IOError(errno.ENOTEMPTY, relpath), relpath)
238
206
        if not _abspath in self._dirs:
239
207
            raise NoSuchFile(relpath)
245
213
        if _abspath in self._files:
246
214
            return MemoryStat(len(self._files[_abspath][0]), False, 
247
215
                              self._files[_abspath][1])
 
216
        elif _abspath == '':
 
217
            return MemoryStat(0, True, None)
248
218
        elif _abspath in self._dirs:
249
219
            return MemoryStat(0, True, self._dirs[_abspath])
250
220
        else:
260
230
 
261
231
    def _abspath(self, relpath):
262
232
        """Generate an internal absolute path."""
263
 
        relpath = urlutils.unescape(relpath)
264
 
        if relpath[:1] == '/':
265
 
            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)
 
233
        if relpath.find('..') != -1:
 
234
            raise AssertionError('relpath contains ..')
 
235
        if relpath == '.':
 
236
            return self._cwd[:-1]
 
237
        if relpath.endswith('/'):
 
238
            relpath = relpath[:-1]
 
239
        if relpath.startswith('./'):
 
240
            relpath = relpath[2:]
 
241
        return self._cwd + relpath
280
242
 
281
243
 
282
244
class _MemoryLock(object):
283
245
    """This makes a lock."""
284
246
 
285
247
    def __init__(self, path, transport):
 
248
        assert isinstance(transport, MemoryTransport)
286
249
        self.path = path
287
250
        self.transport = transport
288
251
        if self.path in self.transport._locks:
292
255
    def __del__(self):
293
256
        # Should this warn, or actually try to cleanup?
294
257
        if self.transport:
295
 
            warnings.warn("MemoryLock %r not explicitly unlocked" % (self.path,))
 
258
            warn("MemoryLock %r not explicitly unlocked" % (self.path,))
296
259
            self.unlock()
297
260
 
298
261
    def unlock(self):
305
268
 
306
269
    def setUp(self):
307
270
        """See bzrlib.transport.Server.setUp."""
308
 
        self._dirs = {'/':None}
 
271
        self._dirs = {}
309
272
        self._files = {}
310
273
        self._locks = {}
311
 
        self._scheme = "memory+%s:///" % id(self)
 
274
        self._scheme = "memory+%s:" % id(self)
312
275
        def memory_factory(url):
313
276
            result = MemoryTransport(url)
314
277
            result._dirs = self._dirs