← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: John Arbash Meinel
  • Date: 2009-07-30 23:54:26 UTC
  • mto: This revision was merged to the branch mainline in revision 4580.
  • Revision ID: john@arbash-meinel.com-20090730235426-o8h73swbh7seqaf7
Update the breakin support to support CTRL-BREAK on Windows.

The signal handling code is very similar, but the testing code got a bit clumsy.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/local.py
1
 
# Copyright (C) 2005-2011 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
20
20
"""
21
21
 
22
22
import os
23
 
from stat import ST_MODE, S_ISDIR, S_IMODE
 
23
from stat import ST_MODE, S_ISDIR, ST_SIZE, S_IMODE
24
24
import sys
25
25
 
26
26
from bzrlib.lazy_import import lazy_import
33
33
    osutils,
34
34
    urlutils,
35
35
    symbol_versioning,
 
36
    transport,
36
37
    )
 
38
from bzrlib.trace import mutter
37
39
from bzrlib.transport import LateReadError
38
40
""")
39
41
 
40
 
from bzrlib import transport
41
 
 
42
 
 
43
 
_append_flags = os.O_CREAT | os.O_APPEND | os.O_WRONLY | osutils.O_BINARY | osutils.O_NOINHERIT
44
 
_put_non_atomic_flags = os.O_CREAT | os.O_TRUNC | os.O_WRONLY | osutils.O_BINARY | osutils.O_NOINHERIT
45
 
 
46
 
 
47
 
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):
48
50
    """This is the transport agent for local filesystem access."""
49
51
 
50
52
    def __init__(self, base):
72
74
 
73
75
        super(LocalTransport, self).__init__(base)
74
76
        self._local_base = urlutils.local_path_from_url(base)
75
 
        if self._local_base[-1] != '/':
76
 
            self._local_base = self._local_base + '/'
77
77
 
78
78
    def clone(self, offset=None):
79
79
        """Return a new LocalTransport with root at self.base + offset
99
99
         - relative_reference is url escaped.
100
100
        """
101
101
        if relative_reference in ('.', ''):
102
 
            # _local_base normally has a trailing slash; strip it so that stat
103
 
            # on a transport pointing to a symlink reads the link not the
104
 
            # referent but be careful of / and c:\
105
 
            return osutils.split(self._local_base)[0]
 
102
            return self._local_base
106
103
        return self._local_base + urlutils.unescape(relative_reference)
107
104
 
108
105
    def abspath(self, relpath):
163
160
            transport._file_streams[canonical_url].flush()
164
161
        try:
165
162
            path = self._abspath(relpath)
166
 
            return osutils.open_file(path, 'rb')
 
163
            return open(path, 'rb')
167
164
        except (IOError, OSError),e:
168
165
            if e.errno == errno.EISDIR:
169
166
                return LateReadError(relpath)
329
326
 
330
327
    def open_write_stream(self, relpath, mode=None):
331
328
        """See Transport.open_write_stream."""
 
329
        # initialise the file
 
330
        self.put_bytes_non_atomic(relpath, "", mode=mode)
332
331
        abspath = self._abspath(relpath)
333
 
        handle = osutils.open_file(abspath, 'wb')
334
 
        handle.truncate()
 
332
        handle = open(abspath, 'wb')
335
333
        if mode is not None:
336
334
            self._check_mode_and_size(abspath, handle.fileno(), mode)
337
335
        transport._file_streams[self.abspath(relpath)] = handle
401
399
 
402
400
    def rename(self, rel_from, rel_to):
403
401
        path_from = self._abspath(rel_from)
404
 
        path_to = self._abspath(rel_to)
405
402
        try:
406
403
            # *don't* call bzrlib.osutils.rename, because we want to
407
 
            # detect conflicting names on rename, and osutils.rename tries to
408
 
            # mask cross-platform differences there
409
 
            os.rename(path_from, path_to)
 
404
            # detect errors on rename
 
405
            os.rename(path_from, self._abspath(rel_to))
410
406
        except (IOError, OSError),e:
411
407
            # TODO: What about path_to?
412
408
            self._translate_error(e, path_from)
485
481
        path = relpath
486
482
        try:
487
483
            path = self._abspath(relpath)
488
 
            return os.lstat(path)
 
484
            return os.stat(path)
489
485
        except (IOError, OSError),e:
490
486
            self._translate_error(e, path)
491
487
 
519
515
        except (IOError, OSError),e:
520
516
            self._translate_error(e, path)
521
517
 
522
 
    if osutils.host_os_dereferences_symlinks():
523
 
        def readlink(self, relpath):
524
 
            """See Transport.readlink."""
525
 
            return osutils.readlink(self._abspath(relpath))
526
 
 
527
 
    if osutils.hardlinks_good():
528
 
        def hardlink(self, source, link_name):
529
 
            """See Transport.link."""
530
 
            try:
531
 
                os.link(self._abspath(source), self._abspath(link_name))
532
 
            except (IOError, OSError), e:
533
 
                self._translate_error(e, source)
534
 
 
535
 
    if osutils.has_symlinks():
536
 
        def symlink(self, source, link_name):
537
 
            """See Transport.symlink."""
538
 
            abs_link_dirpath = urlutils.dirname(self.abspath(link_name))
539
 
            source_rel = urlutils.file_relpath(
540
 
                urlutils.strip_trailing_slash(abs_link_dirpath),
541
 
                urlutils.strip_trailing_slash(self.abspath(source))
542
 
            )
543
 
 
544
 
            try:
545
 
                os.symlink(source_rel, self._abspath(link_name))
546
 
            except (IOError, OSError), e:
547
 
                self._translate_error(e, source_rel)
548
 
 
549
518
    def _can_roundtrip_unix_modebits(self):
550
519
        if sys.platform == 'win32':
551
520
            # anyone else?
585
554
            return EmulatedWin32LocalTransport(abspath)
586
555
 
587
556
 
 
557
class LocalURLServer(Server):
 
558
    """A pretend server for local transports, using file:// urls.
 
559
 
 
560
    Of course no actual server is required to access the local filesystem, so
 
561
    this just exists to tell the test code how to get to it.
 
562
    """
 
563
 
 
564
    def setUp(self):
 
565
        """Setup the server to service requests.
 
566
 
 
567
        :param decorated_transport: ignored by this implementation.
 
568
        """
 
569
 
 
570
    def get_url(self):
 
571
        """See Transport.Server.get_url."""
 
572
        return urlutils.local_path_to_url('')
 
573
 
 
574
 
588
575
def get_test_permutations():
589
576
    """Return the permutations to be used in testing."""
590
 
    from bzrlib.tests import test_server
591
 
    return [(LocalTransport, test_server.LocalURLServer),]
 
577
    return [
 
578
            (LocalTransport, LocalURLServer),
 
579
            ]