← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: Vincent Ladeuil
  • Date: 2010-03-02 10:21:39 UTC
  • mfrom: (4797.2.24 2.1)
  • mto: This revision was merged to the branch mainline in revision 5069.
  • Revision ID: v.ladeuil+lp@free.fr-20100302102139-b5cba7h6xu13mekg
Merge 2.1 into trunk including fixes for #331095, #507557, #185103, #524184 and #369501

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/local.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Transport for the local filesystem.
18
18
 
39
39
from bzrlib.transport import LateReadError
40
40
""")
41
41
 
42
 
from bzrlib.transport import Transport, Server
 
42
from bzrlib import transport
43
43
 
44
44
 
45
45
_append_flags = os.O_CREAT | os.O_APPEND | os.O_WRONLY | osutils.O_BINARY
46
46
_put_non_atomic_flags = os.O_CREAT | os.O_TRUNC | os.O_WRONLY | osutils.O_BINARY
47
47
 
48
48
 
49
 
class LocalTransport(Transport):
 
49
class LocalTransport(transport.Transport):
50
50
    """This is the transport agent for local filesystem access."""
51
51
 
52
52
    def __init__(self, base):
63
63
            base = urlutils.local_path_to_url(base)
64
64
        if base[-1] != '/':
65
65
            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
 
66
75
        super(LocalTransport, self).__init__(base)
67
76
        self._local_base = urlutils.local_path_from_url(base)
68
77
 
69
78
    def clone(self, offset=None):
70
79
        """Return a new LocalTransport with root at self.base + offset
71
 
        Because the local filesystem does not require a connection, 
 
80
        Because the local filesystem does not require a connection,
72
81
        we can just return a new object.
73
82
        """
74
83
        if offset is None:
96
105
    def abspath(self, relpath):
97
106
        """Return the full url to the given relative URL."""
98
107
        # TODO: url escape the result. RBC 20060523.
99
 
        assert isinstance(relpath, basestring), (type(relpath), relpath)
100
108
        # jam 20060426 Using normpath on the real path, because that ensures
101
109
        #       proper handling of stuff like
102
110
        path = osutils.normpath(osutils.pathjoin(
103
111
                    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
 
104
122
        return urlutils.local_path_to_url(path)
105
123
 
106
124
    def local_abspath(self, relpath):
109
127
        This function only exists for the LocalTransport, since it is
110
128
        the only one that has direct local access.
111
129
        This is mostly for stuff like WorkingTree which needs to know
112
 
        the local working directory.
113
 
        
 
130
        the local working directory.  The returned path will always contain
 
131
        forward slashes as the path separator, regardless of the platform.
 
132
 
114
133
        This function is quite expensive: it calls realpath which resolves
115
134
        symlinks.
116
135
        """
152
171
 
153
172
        :param relpath: Location to put the contents, relative to base.
154
173
        :param f:       File-like object.
155
 
        :param mode: The mode for the newly created file, 
 
174
        :param mode: The mode for the newly created file,
156
175
                     None means just use the default
157
176
        """
158
177
 
185
204
        except (IOError, OSError),e:
186
205
            self._translate_error(e, path)
187
206
        try:
188
 
            fp.write(bytes)
 
207
            if bytes:
 
208
                fp.write(bytes)
189
209
            fp.commit()
190
210
        finally:
191
211
            fp.close()
266
286
    def put_bytes_non_atomic(self, relpath, bytes, mode=None,
267
287
                             create_parent_dir=False, dir_mode=None):
268
288
        def writer(fd):
269
 
            os.write(fd, bytes)
 
289
            if bytes:
 
290
                os.write(fd, bytes)
270
291
        self._put_non_atomic_helper(relpath, writer, mode=mode,
271
292
                                    create_parent_dir=create_parent_dir,
272
293
                                    dir_mode=dir_mode)
307
328
        """See Transport.open_write_stream."""
308
329
        # initialise the file
309
330
        self.put_bytes_non_atomic(relpath, "", mode=mode)
310
 
        handle = open(self._abspath(relpath), 'wb')
 
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)
311
335
        transport._file_streams[self.abspath(relpath)] = handle
312
336
        return transport.FileFileStream(self, relpath, handle)
313
337
 
348
372
        file_abspath, fd = self._get_append_file(relpath, mode=mode)
349
373
        try:
350
374
            result = self._check_mode_and_size(file_abspath, fd, mode=mode)
351
 
            os.write(fd, bytes)
 
375
            if bytes:
 
376
                os.write(fd, bytes)
352
377
        finally:
353
378
            os.close(fd)
354
379
        return result
375
400
    def rename(self, rel_from, rel_to):
376
401
        path_from = self._abspath(rel_from)
377
402
        try:
378
 
            # *don't* call bzrlib.osutils.rename, because we want to 
 
403
            # *don't* call bzrlib.osutils.rename, because we want to
379
404
            # detect errors on rename
380
405
            os.rename(path_from, self._abspath(rel_to))
381
406
        except (IOError, OSError),e:
508
533
        self._local_base = urlutils._win32_local_path_from_url(base)
509
534
 
510
535
    def abspath(self, relpath):
511
 
        assert isinstance(relpath, basestring), (type(relpath), relpath)
512
536
        path = osutils.normpath(osutils.pathjoin(
513
537
                    self._local_base, urlutils.unescape(relpath)))
514
538
        return urlutils._win32_local_path_to_url(path)
515
539
 
516
540
    def clone(self, offset=None):
517
541
        """Return a new LocalTransport with root at self.base + offset
518
 
        Because the local filesystem does not require a connection, 
 
542
        Because the local filesystem does not require a connection,
519
543
        we can just return a new object.
520
544
        """
521
545
        if offset is None:
530
554
            return EmulatedWin32LocalTransport(abspath)
531
555
 
532
556
 
533
 
class LocalURLServer(Server):
534
 
    """A pretend server for local transports, using file:// urls.
535
 
    
536
 
    Of course no actual server is required to access the local filesystem, so
537
 
    this just exists to tell the test code how to get to it.
538
 
    """
539
 
 
540
 
    def setUp(self):
541
 
        """Setup the server to service requests.
542
 
        
543
 
        :param decorated_transport: ignored by this implementation.
544
 
        """
545
 
 
546
 
    def get_url(self):
547
 
        """See Transport.Server.get_url."""
548
 
        return urlutils.local_path_to_url('')
549
 
 
550
 
 
551
557
def get_test_permutations():
552
558
    """Return the permutations to be used in testing."""
553
 
    return [
554
 
            (LocalTransport, LocalURLServer),
555
 
            ]
 
559
    from bzrlib.tests import test_server
 
560
    return [(LocalTransport, test_server.LocalURLServer),]