← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: Tarmac
  • Author(s): Vincent Ladeuil
  • Date: 2017-01-30 14:42:05 UTC
  • mfrom: (6620.1.1 trunk)
  • Revision ID: tarmac-20170130144205-r8fh2xpmiuxyozpv
Merge  2.7 into trunk including fix for bug #1657238 [r=vila]

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-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
25
from stat import ST_MODE, S_ISDIR, S_IMODE
24
26
import sys
50
52
    def __init__(self, base):
51
53
        """Set the base path where files will be stored."""
52
54
        if not base.startswith('file://'):
53
 
            symbol_versioning.warn(
54
 
                "Instantiating LocalTransport with a filesystem path"
55
 
                " is deprecated as of bzr 0.8."
56
 
                " Please use bzrlib.transport.get_transport()"
57
 
                " or pass in a file:// url.",
58
 
                 DeprecationWarning,
59
 
                 stacklevel=2
60
 
                 )
61
 
            base = urlutils.local_path_to_url(base)
 
55
            raise AssertionError("not a file:// url: %r" % base)
62
56
        if base[-1] != '/':
63
57
            base = base + '/'
64
58
 
72
66
 
73
67
        super(LocalTransport, self).__init__(base)
74
68
        self._local_base = urlutils.local_path_from_url(base)
 
69
        if self._local_base[-1] != '/':
 
70
            self._local_base = self._local_base + '/'
75
71
 
76
72
    def clone(self, offset=None):
77
73
        """Return a new LocalTransport with root at self.base + offset
144
140
        if abspath is None:
145
141
            abspath = u'.'
146
142
 
147
 
        return urlutils.file_relpath(
148
 
            urlutils.strip_trailing_slash(self.base),
149
 
            urlutils.strip_trailing_slash(abspath))
 
143
        return urlutils.file_relpath(self.base, abspath)
150
144
 
151
145
    def has(self, relpath):
152
146
        return os.access(self._abspath(relpath), os.F_OK)
190
184
            fp.close()
191
185
        return length
192
186
 
193
 
    def put_bytes(self, relpath, bytes, mode=None):
 
187
    def put_bytes(self, relpath, raw_bytes, mode=None):
194
188
        """Copy the string into the location.
195
189
 
196
190
        :param relpath: Location to put the contents, relative to base.
197
 
        :param bytes:   String
 
191
        :param raw_bytes:   String
198
192
        """
199
 
 
 
193
        if not isinstance(raw_bytes, str):
 
194
            raise TypeError(
 
195
                'raw_bytes must be a plain string, not %s' % type(raw_bytes))
200
196
        path = relpath
201
197
        try:
202
198
            path = self._abspath(relpath)
206
202
            self._translate_error(e, path)
207
203
        try:
208
204
            if bytes:
209
 
                fp.write(bytes)
 
205
                fp.write(raw_bytes)
210
206
            fp.commit()
211
207
        finally:
212
208
            fp.close()
255
251
            if mode is not None and mode != S_IMODE(st.st_mode):
256
252
                # Because of umask, we may still need to chmod the file.
257
253
                # But in the general case, we won't have to
258
 
                os.chmod(abspath, mode)
 
254
                osutils.chmod_if_possible(abspath, mode)
259
255
            writer(fd)
260
256
        finally:
261
257
            os.close(fd)
314
310
            local_mode = mode
315
311
        try:
316
312
            os.mkdir(abspath, local_mode)
317
 
            if mode is not None:
318
 
                # It is probably faster to just do the chmod, rather than
319
 
                # doing a stat, and then trying to compare
320
 
                os.chmod(abspath, mode)
321
313
        except (IOError, OSError),e:
322
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)
323
320
 
324
321
    def mkdir(self, relpath, mode=None):
325
322
        """Create a directory at the given path."""
327
324
 
328
325
    def open_write_stream(self, relpath, mode=None):
329
326
        """See Transport.open_write_stream."""
330
 
        # initialise the file
331
 
        self.put_bytes_non_atomic(relpath, "", mode=mode)
332
327
        abspath = self._abspath(relpath)
333
 
        handle = osutils.open_file(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()
334
333
        if mode is not None:
335
334
            self._check_mode_and_size(abspath, handle.fileno(), mode)
336
335
        transport._file_streams[self.abspath(relpath)] = handle
355
354
        if mode is not None and mode != S_IMODE(st.st_mode):
356
355
            # Because of umask, we may still need to chmod the file.
357
356
            # But in the general case, we won't have to
358
 
            os.chmod(file_abspath, mode)
 
357
            osutils.chmod_if_possible(file_abspath, mode)
359
358
        return st.st_size
360
359
 
361
360
    def append_file(self, relpath, f, mode=None):
454
453
                    otherpath = other._abspath(path)
455
454
                    shutil.copy(mypath, otherpath)
456
455
                    if mode is not None:
457
 
                        os.chmod(otherpath, mode)
 
456
                        osutils.chmod_if_possible(otherpath, mode)
458
457
                except (IOError, OSError),e:
459
458
                    self._translate_error(e, path)
460
459
                count += 1
536
535
            """See Transport.symlink."""
537
536
            abs_link_dirpath = urlutils.dirname(self.abspath(link_name))
538
537
            source_rel = urlutils.file_relpath(
539
 
                urlutils.strip_trailing_slash(abs_link_dirpath),
540
 
                urlutils.strip_trailing_slash(self.abspath(source))
541
 
            )
 
538
                abs_link_dirpath, self.abspath(source))
542
539
 
543
540
            try:
544
541
                os.symlink(source_rel, self._abspath(link_name))
563
560
        self._local_base = urlutils._win32_local_path_from_url(base)
564
561
 
565
562
    def abspath(self, relpath):
566
 
        path = osutils.normpath(osutils.pathjoin(
 
563
        path = osutils._win32_normpath(osutils.pathjoin(
567
564
                    self._local_base, urlutils.unescape(relpath)))
568
565
        return urlutils._win32_local_path_to_url(path)
569
566