← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/local.py

  • Committer: Vincent Ladeuil
  • Date: 2012-03-13 17:25:29 UTC
  • mfrom: (6499 +trunk)
  • mto: This revision was merged to the branch mainline in revision 6501.
  • Revision ID: v.ladeuil+lp@free.fr-20120313172529-i0suyjnepsor25i7
Merge trunk

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-2011 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
 
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
146
140
        if abspath is None:
147
141
            abspath = u'.'
148
142
 
149
 
        return urlutils.file_relpath(
150
 
            urlutils.strip_trailing_slash(self.base),
151
 
            urlutils.strip_trailing_slash(abspath))
 
143
        return urlutils.file_relpath(self.base, abspath)
152
144
 
153
145
    def has(self, relpath):
154
146
        return os.access(self._abspath(relpath), os.F_OK)
257
249
            if mode is not None and mode != S_IMODE(st.st_mode):
258
250
                # Because of umask, we may still need to chmod the file.
259
251
                # But in the general case, we won't have to
260
 
                os.chmod(abspath, mode)
 
252
                osutils.chmod_if_possible(abspath, mode)
261
253
            writer(fd)
262
254
        finally:
263
255
            os.close(fd)
316
308
            local_mode = mode
317
309
        try:
318
310
            os.mkdir(abspath, local_mode)
319
 
            if mode is not None:
320
 
                # It is probably faster to just do the chmod, rather than
321
 
                # doing a stat, and then trying to compare
322
 
                os.chmod(abspath, mode)
323
311
        except (IOError, OSError),e:
324
312
            self._translate_error(e, abspath)
 
313
        if mode is not None:
 
314
            try:
 
315
                osutils.chmod_if_possible(abspath, mode)
 
316
            except (IOError, OSError), e:
 
317
                self._translate_error(e, abspath)
325
318
 
326
319
    def mkdir(self, relpath, mode=None):
327
320
        """Create a directory at the given path."""
329
322
 
330
323
    def open_write_stream(self, relpath, mode=None):
331
324
        """See Transport.open_write_stream."""
332
 
        # initialise the file
333
 
        self.put_bytes_non_atomic(relpath, "", mode=mode)
334
325
        abspath = self._abspath(relpath)
335
 
        handle = osutils.open_file(abspath, 'wb')
 
326
        try:
 
327
            handle = osutils.open_file(abspath, 'wb')
 
328
        except (IOError, OSError),e:
 
329
            self._translate_error(e, abspath)
 
330
        handle.truncate()
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
357
352
        if mode is not None and mode != S_IMODE(st.st_mode):
358
353
            # Because of umask, we may still need to chmod the file.
359
354
            # But in the general case, we won't have to
360
 
            os.chmod(file_abspath, mode)
 
355
            osutils.chmod_if_possible(file_abspath, mode)
361
356
        return st.st_size
362
357
 
363
358
    def append_file(self, relpath, f, mode=None):
456
451
                    otherpath = other._abspath(path)
457
452
                    shutil.copy(mypath, otherpath)
458
453
                    if mode is not None:
459
 
                        os.chmod(otherpath, mode)
 
454
                        osutils.chmod_if_possible(otherpath, mode)
460
455
                except (IOError, OSError),e:
461
456
                    self._translate_error(e, path)
462
457
                count += 1
538
533
            """See Transport.symlink."""
539
534
            abs_link_dirpath = urlutils.dirname(self.abspath(link_name))
540
535
            source_rel = urlutils.file_relpath(
541
 
                urlutils.strip_trailing_slash(abs_link_dirpath),
542
 
                urlutils.strip_trailing_slash(self.abspath(source))
543
 
            )
 
536
                abs_link_dirpath, self.abspath(source))
544
537
 
545
538
            try:
546
539
                os.symlink(source_rel, self._abspath(link_name))
565
558
        self._local_base = urlutils._win32_local_path_from_url(base)
566
559
 
567
560
    def abspath(self, relpath):
568
 
        path = osutils.normpath(osutils.pathjoin(
 
561
        path = osutils._win32_normpath(osutils.pathjoin(
569
562
                    self._local_base, urlutils.unescape(relpath)))
570
563
        return urlutils._win32_local_path_to_url(path)
571
564