1
# Copyright (C) 2005-2011 Canonical Ltd
1
# Copyright (C) 2005, 2006 Canonical Ltd
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
23
from stat import ST_MODE, S_ISDIR, S_IMODE
23
from stat import ST_MODE, S_ISDIR, ST_SIZE, S_IMODE
26
26
from bzrlib.lazy_import import lazy_import
38
from bzrlib.trace import mutter
37
39
from bzrlib.transport import LateReadError
40
42
from bzrlib import transport
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
_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
49
class LocalTransport(transport.Transport):
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 + '/'
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.
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)
108
105
def abspath(self, relpath):
163
160
transport._file_streams[canonical_url].flush()
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)
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')
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
402
400
def rename(self, rel_from, rel_to):
403
401
path_from = self._abspath(rel_from)
404
path_to = self._abspath(rel_to)
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)
519
515
except (IOError, OSError),e:
520
516
self._translate_error(e, path)
522
if osutils.host_os_dereferences_symlinks():
523
def readlink(self, relpath):
524
"""See Transport.readlink."""
525
return osutils.readlink(self._abspath(relpath))
527
if osutils.hardlinks_good():
528
def hardlink(self, source, link_name):
529
"""See Transport.link."""
531
os.link(self._abspath(source), self._abspath(link_name))
532
except (IOError, OSError), e:
533
self._translate_error(e, source)
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))
545
os.symlink(source_rel, self._abspath(link_name))
546
except (IOError, OSError), e:
547
self._translate_error(e, source_rel)
549
518
def _can_roundtrip_unix_modebits(self):
550
519
if sys.platform == 'win32':