/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: Aaron Bentley
  • Date: 2008-10-28 10:31:32 UTC
  • mto: (0.14.36 prepare-shelf)
  • mto: This revision was merged to the branch mainline in revision 3820.
  • Revision ID: aaron@aaronbentley.com-20081028103132-syota6mcye41kvqh
Update to use Tree.readlines, if possible.

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
 
39
39
from bzrlib.transport import LateReadError
40
40
""")
41
41
 
42
 
from bzrlib import transport
43
 
 
44
 
 
45
 
_append_flags = os.O_CREAT | os.O_APPEND | os.O_WRONLY | osutils.O_BINARY | osutils.O_NOINHERIT
46
 
_put_non_atomic_flags = os.O_CREAT | os.O_TRUNC | os.O_WRONLY | osutils.O_BINARY | osutils.O_NOINHERIT
47
 
 
48
 
 
49
 
class LocalTransport(transport.Transport):
 
42
from bzrlib.transport import Transport, Server
 
43
 
 
44
 
 
45
_append_flags = os.O_CREAT | os.O_APPEND | os.O_WRONLY | osutils.O_BINARY
 
46
_put_non_atomic_flags = os.O_CREAT | os.O_TRUNC | os.O_WRONLY | osutils.O_BINARY
 
47
 
 
48
 
 
49
class LocalTransport(Transport):
50
50
    """This is the transport agent for local filesystem access."""
51
51
 
52
52
    def __init__(self, base):
71
71
            self._local_base = ''
72
72
            super(LocalTransport, self).__init__(base)
73
73
            return
74
 
 
 
74
            
75
75
        super(LocalTransport, self).__init__(base)
76
76
        self._local_base = urlutils.local_path_from_url(base)
77
77
 
78
78
    def clone(self, offset=None):
79
79
        """Return a new LocalTransport with root at self.base + offset
80
 
        Because the local filesystem does not require a connection,
 
80
        Because the local filesystem does not require a connection, 
81
81
        we can just return a new object.
82
82
        """
83
83
        if offset is None:
160
160
            transport._file_streams[canonical_url].flush()
161
161
        try:
162
162
            path = self._abspath(relpath)
163
 
            return osutils.open_file(path, 'rb')
 
163
            return open(path, 'rb')
164
164
        except (IOError, OSError),e:
165
165
            if e.errno == errno.EISDIR:
166
166
                return LateReadError(relpath)
171
171
 
172
172
        :param relpath: Location to put the contents, relative to base.
173
173
        :param f:       File-like object.
174
 
        :param mode: The mode for the newly created file,
 
174
        :param mode: The mode for the newly created file, 
175
175
                     None means just use the default
176
176
        """
177
177
 
204
204
        except (IOError, OSError),e:
205
205
            self._translate_error(e, path)
206
206
        try:
207
 
            if bytes:
208
 
                fp.write(bytes)
 
207
            fp.write(bytes)
209
208
            fp.commit()
210
209
        finally:
211
210
            fp.close()
286
285
    def put_bytes_non_atomic(self, relpath, bytes, mode=None,
287
286
                             create_parent_dir=False, dir_mode=None):
288
287
        def writer(fd):
289
 
            if bytes:
290
 
                os.write(fd, bytes)
 
288
            os.write(fd, bytes)
291
289
        self._put_non_atomic_helper(relpath, writer, mode=mode,
292
290
                                    create_parent_dir=create_parent_dir,
293
291
                                    dir_mode=dir_mode)
329
327
        # initialise the file
330
328
        self.put_bytes_non_atomic(relpath, "", mode=mode)
331
329
        abspath = self._abspath(relpath)
332
 
        handle = osutils.open_file(abspath, 'wb')
 
330
        handle = open(abspath, 'wb')
333
331
        if mode is not None:
334
332
            self._check_mode_and_size(abspath, handle.fileno(), mode)
335
333
        transport._file_streams[self.abspath(relpath)] = handle
372
370
        file_abspath, fd = self._get_append_file(relpath, mode=mode)
373
371
        try:
374
372
            result = self._check_mode_and_size(file_abspath, fd, mode=mode)
375
 
            if bytes:
376
 
                os.write(fd, bytes)
 
373
            os.write(fd, bytes)
377
374
        finally:
378
375
            os.close(fd)
379
376
        return result
399
396
 
400
397
    def rename(self, rel_from, rel_to):
401
398
        path_from = self._abspath(rel_from)
402
 
        path_to = self._abspath(rel_to)
403
399
        try:
404
 
            # *don't* call bzrlib.osutils.rename, because we want to
405
 
            # detect conflicting names on rename, and osutils.rename tries to
406
 
            # mask cross-platform differences there; however we do update the
407
 
            # exception to include the filenames
408
 
            os.rename(path_from, path_to)
 
400
            # *don't* call bzrlib.osutils.rename, because we want to 
 
401
            # detect errors on rename
 
402
            os.rename(path_from, self._abspath(rel_to))
409
403
        except (IOError, OSError),e:
410
404
            # TODO: What about path_to?
411
 
            self._translate_error(
412
 
                osutils._add_rename_error_details(e, path_from, path_to),
413
 
                path_from)
 
405
            self._translate_error(e, path_from)
414
406
 
415
407
    def move(self, rel_from, rel_to):
416
408
        """Move the item at rel_from to the location at rel_to"""
486
478
        path = relpath
487
479
        try:
488
480
            path = self._abspath(relpath)
489
 
            return os.lstat(path)
 
481
            return os.stat(path)
490
482
        except (IOError, OSError),e:
491
483
            self._translate_error(e, path)
492
484
 
520
512
        except (IOError, OSError),e:
521
513
            self._translate_error(e, path)
522
514
 
523
 
    if osutils.host_os_dereferences_symlinks():
524
 
        def readlink(self, relpath):
525
 
            """See Transport.readlink."""
526
 
            return osutils.readlink(self._abspath(relpath))
527
 
 
528
 
    if osutils.hardlinks_good():
529
 
        def hardlink(self, source, link_name):
530
 
            """See Transport.link."""
531
 
            try:
532
 
                os.link(self._abspath(source), self._abspath(link_name))
533
 
            except (IOError, OSError), e:
534
 
                self._translate_error(e, source)
535
 
 
536
 
    if osutils.has_symlinks():
537
 
        def symlink(self, source, link_name):
538
 
            """See Transport.symlink."""
539
 
            abs_link_dirpath = urlutils.dirname(self.abspath(link_name))
540
 
            source_rel = urlutils.file_relpath(
541
 
                urlutils.strip_trailing_slash(abs_link_dirpath),
542
 
                urlutils.strip_trailing_slash(self.abspath(source))
543
 
            )
544
 
 
545
 
            try:
546
 
                os.symlink(source_rel, self._abspath(link_name))
547
 
            except (IOError, OSError), e:
548
 
                self._translate_error(e, source_rel)
549
 
 
550
515
    def _can_roundtrip_unix_modebits(self):
551
516
        if sys.platform == 'win32':
552
517
            # anyone else?
571
536
 
572
537
    def clone(self, offset=None):
573
538
        """Return a new LocalTransport with root at self.base + offset
574
 
        Because the local filesystem does not require a connection,
 
539
        Because the local filesystem does not require a connection, 
575
540
        we can just return a new object.
576
541
        """
577
542
        if offset is None:
586
551
            return EmulatedWin32LocalTransport(abspath)
587
552
 
588
553
 
 
554
class LocalURLServer(Server):
 
555
    """A pretend server for local transports, using file:// urls.
 
556
    
 
557
    Of course no actual server is required to access the local filesystem, so
 
558
    this just exists to tell the test code how to get to it.
 
559
    """
 
560
 
 
561
    def setUp(self):
 
562
        """Setup the server to service requests.
 
563
        
 
564
        :param decorated_transport: ignored by this implementation.
 
565
        """
 
566
 
 
567
    def get_url(self):
 
568
        """See Transport.Server.get_url."""
 
569
        return urlutils.local_path_to_url('')
 
570
 
 
571
 
589
572
def get_test_permutations():
590
573
    """Return the permutations to be used in testing."""
591
 
    from bzrlib.tests import test_server
592
 
    return [(LocalTransport, test_server.LocalURLServer),]
 
574
    return [
 
575
            (LocalTransport, LocalURLServer),
 
576
            ]