← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: Patch Queue Manager
  • Date: 2016-02-01 19:56:05 UTC
  • mfrom: (6615.1.1 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20160201195605-o7rl92wf6uyum3fk
(vila) Open trunk again as 2.8b1 (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/local.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005-2012, 2016 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
19
19
This is a fairly thin wrapper on regular file IO.
20
20
"""
21
21
 
 
22
from __future__ import absolute_import
 
23
 
22
24
import os
23
 
from stat import ST_MODE, S_ISDIR, ST_SIZE, S_IMODE
 
25
from stat import ST_MODE, S_ISDIR, S_IMODE
24
26
import sys
25
27
 
26
28
from bzrlib.lazy_import import lazy_import
33
35
    osutils,
34
36
    urlutils,
35
37
    symbol_versioning,
36
 
    transport,
37
38
    )
38
 
from bzrlib.trace import mutter
39
39
from bzrlib.transport import LateReadError
40
40
""")
41
41
 
42
42
from bzrlib import transport
43
43
 
44
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
 
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
47
 
48
48
 
49
49
class LocalTransport(transport.Transport):
52
52
    def __init__(self, base):
53
53
        """Set the base path where files will be stored."""
54
54
        if not base.startswith('file://'):
55
 
            symbol_versioning.warn(
56
 
                "Instantiating LocalTransport with a filesystem path"
57
 
                " is deprecated as of bzr 0.8."
58
 
                " Please use bzrlib.transport.get_transport()"
59
 
                " or pass in a file:// url.",
60
 
                 DeprecationWarning,
61
 
                 stacklevel=2
62
 
                 )
63
 
            base = urlutils.local_path_to_url(base)
 
55
            raise AssertionError("not a file:// url: %r" % base)
64
56
        if base[-1] != '/':
65
57
            base = base + '/'
66
58
 
74
66
 
75
67
        super(LocalTransport, self).__init__(base)
76
68
        self._local_base = urlutils.local_path_from_url(base)
 
69
        if self._local_base[-1] != '/':
 
70
            self._local_base = self._local_base + '/'
77
71
 
78
72
    def clone(self, offset=None):
79
73
        """Return a new LocalTransport with root at self.base + offset
99
93
         - relative_reference is url escaped.
100
94
        """
101
95
        if relative_reference in ('.', ''):
102
 
            return self._local_base
 
96
            # _local_base normally has a trailing slash; strip it so that stat
 
97
            # on a transport pointing to a symlink reads the link not the
 
98
            # referent but be careful of / and c:\
 
99
            return osutils.split(self._local_base)[0]
103
100
        return self._local_base + urlutils.unescape(relative_reference)
104
101
 
105
102
    def abspath(self, relpath):
143
140
        if abspath is None:
144
141
            abspath = u'.'
145
142
 
146
 
        return urlutils.file_relpath(
147
 
            urlutils.strip_trailing_slash(self.base),
148
 
            urlutils.strip_trailing_slash(abspath))
 
143
        return urlutils.file_relpath(self.base, abspath)
149
144
 
150
145
    def has(self, relpath):
151
146
        return os.access(self._abspath(relpath), os.F_OK)
160
155
            transport._file_streams[canonical_url].flush()
161
156
        try:
162
157
            path = self._abspath(relpath)
163
 
            return open(path, 'rb')
 
158
            return osutils.open_file(path, 'rb')
164
159
        except (IOError, OSError),e:
165
160
            if e.errno == errno.EISDIR:
166
161
                return LateReadError(relpath)
189
184
            fp.close()
190
185
        return length
191
186
 
192
 
    def put_bytes(self, relpath, bytes, mode=None):
 
187
    def put_bytes(self, relpath, raw_bytes, mode=None):
193
188
        """Copy the string into the location.
194
189
 
195
190
        :param relpath: Location to put the contents, relative to base.
196
 
        :param bytes:   String
 
191
        :param raw_bytes:   String
197
192
        """
198
 
 
 
193
        if not isinstance(raw_bytes, str):
 
194
            raise TypeError(
 
195
                'raw_bytes must be a plain string, not %s' % type(raw_bytes))
199
196
        path = relpath
200
197
        try:
201
198
            path = self._abspath(relpath)
205
202
            self._translate_error(e, path)
206
203
        try:
207
204
            if bytes:
208
 
                fp.write(bytes)
 
205
                fp.write(raw_bytes)
209
206
            fp.commit()
210
207
        finally:
211
208
            fp.close()
254
251
            if mode is not None and mode != S_IMODE(st.st_mode):
255
252
                # Because of umask, we may still need to chmod the file.
256
253
                # But in the general case, we won't have to
257
 
                os.chmod(abspath, mode)
 
254
                osutils.chmod_if_possible(abspath, mode)
258
255
            writer(fd)
259
256
        finally:
260
257
            os.close(fd)
313
310
            local_mode = mode
314
311
        try:
315
312
            os.mkdir(abspath, local_mode)
316
 
            if mode is not None:
317
 
                # It is probably faster to just do the chmod, rather than
318
 
                # doing a stat, and then trying to compare
319
 
                os.chmod(abspath, mode)
320
313
        except (IOError, OSError),e:
321
314
            self._translate_error(e, abspath)
 
315
        if mode is not None:
 
316
            try:
 
317
                osutils.chmod_if_possible(abspath, mode)
 
318
            except (IOError, OSError), e:
 
319
                self._translate_error(e, abspath)
322
320
 
323
321
    def mkdir(self, relpath, mode=None):
324
322
        """Create a directory at the given path."""
326
324
 
327
325
    def open_write_stream(self, relpath, mode=None):
328
326
        """See Transport.open_write_stream."""
329
 
        # initialise the file
330
 
        self.put_bytes_non_atomic(relpath, "", mode=mode)
331
327
        abspath = self._abspath(relpath)
332
 
        handle = open(abspath, 'wb')
 
328
        try:
 
329
            handle = osutils.open_file(abspath, 'wb')
 
330
        except (IOError, OSError),e:
 
331
            self._translate_error(e, abspath)
 
332
        handle.truncate()
333
333
        if mode is not None:
334
334
            self._check_mode_and_size(abspath, handle.fileno(), mode)
335
335
        transport._file_streams[self.abspath(relpath)] = handle
354
354
        if mode is not None and mode != S_IMODE(st.st_mode):
355
355
            # Because of umask, we may still need to chmod the file.
356
356
            # But in the general case, we won't have to
357
 
            os.chmod(file_abspath, mode)
 
357
            osutils.chmod_if_possible(file_abspath, mode)
358
358
        return st.st_size
359
359
 
360
360
    def append_file(self, relpath, f, mode=None):
399
399
 
400
400
    def rename(self, rel_from, rel_to):
401
401
        path_from = self._abspath(rel_from)
 
402
        path_to = self._abspath(rel_to)
402
403
        try:
403
404
            # *don't* call bzrlib.osutils.rename, because we want to
404
 
            # detect errors on rename
405
 
            os.rename(path_from, self._abspath(rel_to))
 
405
            # detect conflicting names on rename, and osutils.rename tries to
 
406
            # mask cross-platform differences there
 
407
            os.rename(path_from, path_to)
406
408
        except (IOError, OSError),e:
407
409
            # TODO: What about path_to?
408
410
            self._translate_error(e, path_from)
451
453
                    otherpath = other._abspath(path)
452
454
                    shutil.copy(mypath, otherpath)
453
455
                    if mode is not None:
454
 
                        os.chmod(otherpath, mode)
 
456
                        osutils.chmod_if_possible(otherpath, mode)
455
457
                except (IOError, OSError),e:
456
458
                    self._translate_error(e, path)
457
459
                count += 1
481
483
        path = relpath
482
484
        try:
483
485
            path = self._abspath(relpath)
484
 
            return os.stat(path)
 
486
            return os.lstat(path)
485
487
        except (IOError, OSError),e:
486
488
            self._translate_error(e, path)
487
489
 
515
517
        except (IOError, OSError),e:
516
518
            self._translate_error(e, path)
517
519
 
 
520
    if osutils.host_os_dereferences_symlinks():
 
521
        def readlink(self, relpath):
 
522
            """See Transport.readlink."""
 
523
            return osutils.readlink(self._abspath(relpath))
 
524
 
 
525
    if osutils.hardlinks_good():
 
526
        def hardlink(self, source, link_name):
 
527
            """See Transport.link."""
 
528
            try:
 
529
                os.link(self._abspath(source), self._abspath(link_name))
 
530
            except (IOError, OSError), e:
 
531
                self._translate_error(e, source)
 
532
 
 
533
    if osutils.has_symlinks():
 
534
        def symlink(self, source, link_name):
 
535
            """See Transport.symlink."""
 
536
            abs_link_dirpath = urlutils.dirname(self.abspath(link_name))
 
537
            source_rel = urlutils.file_relpath(
 
538
                abs_link_dirpath, self.abspath(source))
 
539
 
 
540
            try:
 
541
                os.symlink(source_rel, self._abspath(link_name))
 
542
            except (IOError, OSError), e:
 
543
                self._translate_error(e, source_rel)
 
544
 
518
545
    def _can_roundtrip_unix_modebits(self):
519
546
        if sys.platform == 'win32':
520
547
            # anyone else?
533
560
        self._local_base = urlutils._win32_local_path_from_url(base)
534
561
 
535
562
    def abspath(self, relpath):
536
 
        path = osutils.normpath(osutils.pathjoin(
 
563
        path = osutils._win32_normpath(osutils.pathjoin(
537
564
                    self._local_base, urlutils.unescape(relpath)))
538
565
        return urlutils._win32_local_path_to_url(path)
539
566