← 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-03-27 22:29:55 UTC
  • mto: (3735.39.2 clean)
  • mto: This revision was merged to the branch mainline in revision 4280.
  • Revision ID: john@arbash-meinel.com-20090327222955-utifmfm888zerixt
Implement apply_delta_to_source which doesn't have to malloc another string.

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
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):
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)
332
327
        # initialise the file
333
328
        self.put_bytes_non_atomic(relpath, "", mode=mode)
334
329
        abspath = self._abspath(relpath)
335
 
        handle = osutils.open_file(abspath, 'wb')
 
330
        handle = open(abspath, 'wb')
336
331
        if mode is not None:
337
332
            self._check_mode_and_size(abspath, handle.fileno(), mode)
338
333
        transport._file_streams[self.abspath(relpath)] = handle
375
370
        file_abspath, fd = self._get_append_file(relpath, mode=mode)
376
371
        try:
377
372
            result = self._check_mode_and_size(file_abspath, fd, mode=mode)
378
 
            if bytes:
379
 
                os.write(fd, bytes)
 
373
            os.write(fd, bytes)
380
374
        finally:
381
375
            os.close(fd)
382
376
        return result
402
396
 
403
397
    def rename(self, rel_from, rel_to):
404
398
        path_from = self._abspath(rel_from)
405
 
        path_to = self._abspath(rel_to)
406
399
        try:
407
400
            # *don't* call bzrlib.osutils.rename, because we want to
408
 
            # detect conflicting names on rename, and osutils.rename tries to
409
 
            # mask cross-platform differences there
410
 
            os.rename(path_from, path_to)
 
401
            # detect errors on rename
 
402
            os.rename(path_from, self._abspath(rel_to))
411
403
        except (IOError, OSError),e:
412
404
            # TODO: What about path_to?
413
405
            self._translate_error(e, path_from)
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?
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
            ]