← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: Andrew Bennetts
  • Date: 2009-04-02 05:53:12 UTC
  • mto: This revision was merged to the branch mainline in revision 4242.
  • Revision ID: andrew.bennetts@canonical.com-20090402055312-h7mvgumvm7e620mj
Fix nits in spelling and naming.

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)
207
204
        except (IOError, OSError),e:
208
205
            self._translate_error(e, path)
209
206
        try:
210
 
            if bytes:
211
 
                fp.write(bytes)
 
207
            fp.write(bytes)
212
208
            fp.commit()
213
209
        finally:
214
210
            fp.close()
289
285
    def put_bytes_non_atomic(self, relpath, bytes, mode=None,
290
286
                             create_parent_dir=False, dir_mode=None):
291
287
        def writer(fd):
292
 
            if bytes:
293
 
                os.write(fd, bytes)
 
288
            os.write(fd, bytes)
294
289
        self._put_non_atomic_helper(relpath, writer, mode=mode,
295
290
                                    create_parent_dir=create_parent_dir,
296
291
                                    dir_mode=dir_mode)
329
324
 
330
325
    def open_write_stream(self, relpath, mode=None):
331
326
        """See Transport.open_write_stream."""
 
327
        # initialise the file
 
328
        self.put_bytes_non_atomic(relpath, "", mode=mode)
332
329
        abspath = self._abspath(relpath)
333
 
        handle = osutils.open_file(abspath, 'wb')
334
 
        handle.truncate()
 
330
        handle = open(abspath, 'wb')
335
331
        if mode is not None:
336
332
            self._check_mode_and_size(abspath, handle.fileno(), mode)
337
333
        transport._file_streams[self.abspath(relpath)] = handle
374
370
        file_abspath, fd = self._get_append_file(relpath, mode=mode)
375
371
        try:
376
372
            result = self._check_mode_and_size(file_abspath, fd, mode=mode)
377
 
            if bytes:
378
 
                os.write(fd, bytes)
 
373
            os.write(fd, bytes)
379
374
        finally:
380
375
            os.close(fd)
381
376
        return result
401
396
 
402
397
    def rename(self, rel_from, rel_to):
403
398
        path_from = self._abspath(rel_from)
404
 
        path_to = self._abspath(rel_to)
405
399
        try:
406
400
            # *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)
 
401
            # detect errors on rename
 
402
            os.rename(path_from, self._abspath(rel_to))
410
403
        except (IOError, OSError),e:
411
404
            # TODO: What about path_to?
412
405
            self._translate_error(e, path_from)
485
478
        path = relpath
486
479
        try:
487
480
            path = self._abspath(relpath)
488
 
            return os.lstat(path)
 
481
            return os.stat(path)
489
482
        except (IOError, OSError),e:
490
483
            self._translate_error(e, path)
491
484
 
519
512
        except (IOError, OSError),e:
520
513
            self._translate_error(e, path)
521
514
 
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
515
    def _can_roundtrip_unix_modebits(self):
550
516
        if sys.platform == 'win32':
551
517
            # anyone else?
585
551
            return EmulatedWin32LocalTransport(abspath)
586
552
 
587
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
 
588
572
def get_test_permutations():
589
573
    """Return the permutations to be used in testing."""
590
 
    from bzrlib.tests import test_server
591
 
    return [(LocalTransport, test_server.LocalURLServer),]
 
574
    return [
 
575
            (LocalTransport, LocalURLServer),
 
576
            ]