← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

Merge the 0.17 fixes back into bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/local.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
"""Transport for the local filesystem.
18
18
 
33
33
    osutils,
34
34
    urlutils,
35
35
    symbol_versioning,
36
 
    transport,
37
36
    )
38
37
from bzrlib.trace import mutter
39
 
from bzrlib.transport import LateReadError
40
38
""")
41
39
 
42
 
from bzrlib import transport
 
40
from bzrlib.transport import Transport, Server
43
41
 
44
42
 
45
43
_append_flags = os.O_CREAT | os.O_APPEND | os.O_WRONLY | osutils.O_BINARY
46
44
_put_non_atomic_flags = os.O_CREAT | os.O_TRUNC | os.O_WRONLY | osutils.O_BINARY
47
45
 
48
46
 
49
 
class LocalTransport(transport.Transport):
 
47
class LocalTransport(Transport):
50
48
    """This is the transport agent for local filesystem access."""
51
49
 
52
50
    def __init__(self, base):
63
61
            base = urlutils.local_path_to_url(base)
64
62
        if base[-1] != '/':
65
63
            base = base + '/'
66
 
 
67
 
        # Special case : windows has no "root", but does have
68
 
        # multiple lettered drives inside it. #240910
69
 
        if sys.platform == 'win32' and base == 'file:///':
70
 
            base = ''
71
 
            self._local_base = ''
72
 
            super(LocalTransport, self).__init__(base)
73
 
            return
74
 
 
75
64
        super(LocalTransport, self).__init__(base)
76
65
        self._local_base = urlutils.local_path_from_url(base)
77
66
 
 
67
    def should_cache(self):
 
68
        return False
 
69
 
78
70
    def clone(self, offset=None):
79
71
        """Return a new LocalTransport with root at self.base + offset
80
 
        Because the local filesystem does not require a connection,
 
72
        Because the local filesystem does not require a connection, 
81
73
        we can just return a new object.
82
74
        """
83
75
        if offset is None:
105
97
    def abspath(self, relpath):
106
98
        """Return the full url to the given relative URL."""
107
99
        # TODO: url escape the result. RBC 20060523.
 
100
        assert isinstance(relpath, basestring), (type(relpath), relpath)
108
101
        # jam 20060426 Using normpath on the real path, because that ensures
109
102
        #       proper handling of stuff like
110
103
        path = osutils.normpath(osutils.pathjoin(
111
104
                    self._local_base, urlutils.unescape(relpath)))
112
 
        # on windows, our _local_base may or may not have a drive specified
113
 
        # (ie, it may be "/" or "c:/foo").
114
 
        # If 'relpath' is '/' we *always* get back an abspath without
115
 
        # the drive letter - but if our transport already has a drive letter,
116
 
        # we want our abspaths to have a drive letter too - so handle that
117
 
        # here.
118
 
        if (sys.platform == "win32" and self._local_base[1:2] == ":"
119
 
            and path == '/'):
120
 
            path = self._local_base[:3]
121
 
 
122
105
        return urlutils.local_path_to_url(path)
123
106
 
124
107
    def local_abspath(self, relpath):
127
110
        This function only exists for the LocalTransport, since it is
128
111
        the only one that has direct local access.
129
112
        This is mostly for stuff like WorkingTree which needs to know
130
 
        the local working directory.  The returned path will always contain
131
 
        forward slashes as the path separator, regardless of the platform.
132
 
 
 
113
        the local working directory.
 
114
        
133
115
        This function is quite expensive: it calls realpath which resolves
134
116
        symlinks.
135
117
        """
144
126
            abspath = u'.'
145
127
 
146
128
        return urlutils.file_relpath(
147
 
            urlutils.strip_trailing_slash(self.base),
 
129
            urlutils.strip_trailing_slash(self.base), 
148
130
            urlutils.strip_trailing_slash(abspath))
149
131
 
150
132
    def has(self, relpath):
155
137
 
156
138
        :param relpath: The relative path to the file
157
139
        """
158
 
        canonical_url = self.abspath(relpath)
159
 
        if canonical_url in transport._file_streams:
160
 
            transport._file_streams[canonical_url].flush()
161
140
        try:
162
141
            path = self._abspath(relpath)
163
142
            return open(path, 'rb')
164
143
        except (IOError, OSError),e:
165
 
            if e.errno == errno.EISDIR:
166
 
                return LateReadError(relpath)
167
144
            self._translate_error(e, path)
168
145
 
169
146
    def put_file(self, relpath, f, mode=None):
171
148
 
172
149
        :param relpath: Location to put the contents, relative to base.
173
150
        :param f:       File-like object.
174
 
        :param mode: The mode for the newly created file,
 
151
        :param mode: The mode for the newly created file, 
175
152
                     None means just use the default
176
153
        """
177
154
 
183
160
        except (IOError, OSError),e:
184
161
            self._translate_error(e, path)
185
162
        try:
186
 
            length = self._pump(f, fp)
 
163
            self._pump(f, fp)
187
164
            fp.commit()
188
165
        finally:
189
166
            fp.close()
190
 
        return length
191
167
 
192
168
    def put_bytes(self, relpath, bytes, mode=None):
193
169
        """Copy the string into the location.
204
180
        except (IOError, OSError),e:
205
181
            self._translate_error(e, path)
206
182
        try:
207
 
            if bytes:
208
 
                fp.write(bytes)
 
183
            fp.write(bytes)
209
184
            fp.commit()
210
185
        finally:
211
186
            fp.close()
286
261
    def put_bytes_non_atomic(self, relpath, bytes, mode=None,
287
262
                             create_parent_dir=False, dir_mode=None):
288
263
        def writer(fd):
289
 
            if bytes:
290
 
                os.write(fd, bytes)
 
264
            os.write(fd, bytes)
291
265
        self._put_non_atomic_helper(relpath, writer, mode=mode,
292
266
                                    create_parent_dir=create_parent_dir,
293
267
                                    dir_mode=dir_mode)
324
298
        """Create a directory at the given path."""
325
299
        self._mkdir(self._abspath(relpath), mode=mode)
326
300
 
327
 
    def open_write_stream(self, relpath, mode=None):
328
 
        """See Transport.open_write_stream."""
329
 
        # initialise the file
330
 
        self.put_bytes_non_atomic(relpath, "", mode=mode)
331
 
        abspath = self._abspath(relpath)
332
 
        handle = open(abspath, 'wb')
333
 
        if mode is not None:
334
 
            self._check_mode_and_size(abspath, handle.fileno(), mode)
335
 
        transport._file_streams[self.abspath(relpath)] = handle
336
 
        return transport.FileFileStream(self, relpath, handle)
337
 
 
338
301
    def _get_append_file(self, relpath, mode=None):
339
302
        """Call os.open() for the given relpath"""
340
303
        file_abspath = self._abspath(relpath)
372
335
        file_abspath, fd = self._get_append_file(relpath, mode=mode)
373
336
        try:
374
337
            result = self._check_mode_and_size(file_abspath, fd, mode=mode)
375
 
            if bytes:
376
 
                os.write(fd, bytes)
 
338
            os.write(fd, bytes)
377
339
        finally:
378
340
            os.close(fd)
379
341
        return result
400
362
    def rename(self, rel_from, rel_to):
401
363
        path_from = self._abspath(rel_from)
402
364
        try:
403
 
            # *don't* call bzrlib.osutils.rename, because we want to
 
365
            # *don't* call bzrlib.osutils.rename, because we want to 
404
366
            # detect errors on rename
405
367
            os.rename(path_from, self._abspath(rel_to))
406
368
        except (IOError, OSError),e:
428
390
        except (IOError, OSError),e:
429
391
            self._translate_error(e, path)
430
392
 
431
 
    def external_url(self):
432
 
        """See bzrlib.transport.Transport.external_url."""
433
 
        # File URL's are externally usable.
434
 
        return self.base
435
 
 
436
393
    def copy_to(self, relpaths, other, mode=None, pb=None):
437
394
        """Copy a set of entries from self into another Transport.
438
395
 
533
490
        self._local_base = urlutils._win32_local_path_from_url(base)
534
491
 
535
492
    def abspath(self, relpath):
 
493
        assert isinstance(relpath, basestring), (type(relpath), relpath)
536
494
        path = osutils.normpath(osutils.pathjoin(
537
495
                    self._local_base, urlutils.unescape(relpath)))
538
496
        return urlutils._win32_local_path_to_url(path)
539
497
 
540
498
    def clone(self, offset=None):
541
499
        """Return a new LocalTransport with root at self.base + offset
542
 
        Because the local filesystem does not require a connection,
 
500
        Because the local filesystem does not require a connection, 
543
501
        we can just return a new object.
544
502
        """
545
503
        if offset is None:
554
512
            return EmulatedWin32LocalTransport(abspath)
555
513
 
556
514
 
 
515
class LocalURLServer(Server):
 
516
    """A pretend server for local transports, using file:// urls.
 
517
    
 
518
    Of course no actual server is required to access the local filesystem, so
 
519
    this just exists to tell the test code how to get to it.
 
520
    """
 
521
 
 
522
    def setUp(self):
 
523
        """Setup the server to service requests.
 
524
        
 
525
        :param decorated_transport: ignored by this implementation.
 
526
        """
 
527
 
 
528
    def get_url(self):
 
529
        """See Transport.Server.get_url."""
 
530
        return urlutils.local_path_to_url('')
 
531
 
 
532
 
557
533
def get_test_permutations():
558
534
    """Return the permutations to be used in testing."""
559
 
    from bzrlib.tests import test_server
560
 
    return [(LocalTransport, test_server.LocalURLServer),]
 
535
    return [
 
536
            (LocalTransport, LocalURLServer),
 
537
            ]