/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: Andrew Bennetts
  • Date: 2010-03-26 04:26:55 UTC
  • mto: This revision was merged to the branch mainline in revision 5117.
  • Revision ID: andrew.bennetts@canonical.com-20100326042655-jnusy59k1t8hjs7l
Use YYYYMMDD in XXX comment, not DDMMYYYY.

Show diffs side-by-side

added added

removed removed

Lines of Context:
17
17
import os
18
18
import re
19
19
import stat
20
 
from stat import S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE
 
20
from stat import (S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE,
 
21
                  S_ISCHR, S_ISBLK, S_ISFIFO, S_ISSOCK)
21
22
import sys
22
23
import time
23
24
import codecs
 
25
import warnings
24
26
 
25
27
from bzrlib.lazy_import import lazy_import
26
28
lazy_import(globals(), """
37
39
from shutil import (
38
40
    rmtree,
39
41
    )
 
42
import signal
40
43
import socket
41
44
import subprocess
42
45
import tempfile
48
51
from bzrlib import (
49
52
    cache_utf8,
50
53
    errors,
51
 
    trace,
52
54
    win32utils,
53
55
    )
 
56
from bzrlib.smart import medium
 
57
from bzrlib.tests import test_smart_transport
54
58
""")
55
59
 
56
60
from bzrlib.symbol_versioning import (
360
364
    return _win32_fixdrive(tempfile.mkdtemp(*args, **kwargs).replace('\\', '/'))
361
365
 
362
366
 
363
 
def _add_rename_error_details(e, old, new):
364
 
    new_e = OSError(e.errno, "failed to rename %s to %s: %s"
365
 
        % (old, new, e.strerror))
366
 
    new_e.filename = old
367
 
    new_e.to_filename = new
368
 
    return new_e
369
 
 
370
 
 
371
367
def _win32_rename(old, new):
372
368
    """We expect to be able to atomically replace 'new' with old.
373
369
 
375
371
    and then deleted.
376
372
    """
377
373
    try:
378
 
        fancy_rename(old, new, rename_func=_wrapped_rename, unlink_func=os.unlink)
 
374
        fancy_rename(old, new, rename_func=os.rename, unlink_func=os.unlink)
379
375
    except OSError, e:
380
376
        if e.errno in (errno.EPERM, errno.EACCES, errno.EBUSY, errno.EINVAL):
381
377
            # If we try to rename a non-existant file onto cwd, we get
386
382
        raise
387
383
 
388
384
 
389
 
def _wrapped_rename(old, new):
390
 
    """Rename a file or directory"""
391
 
    try:
392
 
        os.rename(old, new)
393
 
    except (IOError, OSError), e:
394
 
        # this is eventually called by all rename-like functions, so should 
395
 
        # catch all of them
396
 
        raise _add_rename_error_details(e, old, new)
397
 
 
398
 
 
399
385
def _mac_getcwd():
400
386
    return unicodedata.normalize('NFC', os.getcwdu())
401
387
 
406
392
realpath = _posix_realpath
407
393
pathjoin = os.path.join
408
394
normpath = os.path.normpath
409
 
rename = _wrapped_rename # overridden below on win32
410
395
getcwd = os.getcwdu
 
396
rename = os.rename
411
397
dirname = os.path.dirname
412
398
basename = os.path.basename
413
399
split = os.path.split
1147
1133
 
1148
1134
 
1149
1135
def relpath(base, path):
1150
 
    """Return path relative to base, or raise PathNotChild exception.
 
1136
    """Return path relative to base, or raise exception.
1151
1137
 
1152
1138
    The path may be either an absolute path or a path relative to the
1153
1139
    current working directory.
1155
1141
    os.path.commonprefix (python2.4) has a bad bug that it works just
1156
1142
    on string prefixes, assuming that '/u' is a prefix of '/u2'.  This
1157
1143
    avoids that problem.
1158
 
 
1159
 
    NOTE: `base` should not have a trailing slash otherwise you'll get
1160
 
    PathNotChild exceptions regardless of `path`.
1161
1144
    """
1162
1145
 
1163
1146
    if len(base) < MIN_ABS_PATHLENGTH:
1383
1366
        False)`).  May be ignored if the feature is not available on this
1384
1367
        platform or Python version.
1385
1368
    """
1386
 
    try:
1387
 
        import signal
1388
 
        siginterrupt = signal.siginterrupt
1389
 
    except ImportError:
1390
 
        # This python implementation doesn't provide signal support, hence no
1391
 
        # handler exists
1392
 
        return None
1393
 
    except AttributeError:
1394
 
        # siginterrupt doesn't exist on this platform, or for this version
1395
 
        # of Python.
1396
 
        siginterrupt = lambda signum, flag: None
 
1369
    old_handler = signal.signal(signum, handler)
1397
1370
    if restart_syscall:
1398
 
        def sig_handler(*args):
1399
 
            # Python resets the siginterrupt flag when a signal is
1400
 
            # received.  <http://bugs.python.org/issue8354>
1401
 
            # As a workaround for some cases, set it back the way we want it.
 
1371
        try:
 
1372
            siginterrupt = signal.siginterrupt
 
1373
        except AttributeError: # siginterrupt doesn't exist on this platform, or for this version of
 
1374
            # Python.
 
1375
            pass
 
1376
        else:
1402
1377
            siginterrupt(signum, False)
1403
 
            # Now run the handler function passed to set_signal_handler.
1404
 
            handler(*args)
1405
 
    else:
1406
 
        sig_handler = handler
1407
 
    old_handler = signal.signal(signum, sig_handler)
1408
 
    if restart_syscall:
1409
 
        siginterrupt(signum, False)
1410
1378
    return old_handler
1411
1379
 
1412
1380
 
1506
1474
 
1507
1475
 
1508
1476
_registered_sigwinch = False
 
1477
 
1509
1478
def watch_sigwinch():
1510
 
    """Register for SIGWINCH, once and only once.
1511
 
 
1512
 
    Do nothing if the signal module is not available.
1513
 
    """
 
1479
    """Register for SIGWINCH, once and only once."""
1514
1480
    global _registered_sigwinch
1515
1481
    if not _registered_sigwinch:
1516
 
        try:
1517
 
            import signal
1518
 
            if getattr(signal, "SIGWINCH", None) is not None:
1519
 
                set_signal_handler(signal.SIGWINCH, _terminal_size_changed)
1520
 
        except ImportError:
1521
 
            # python doesn't provide signal support, nothing we can do about it
 
1482
        if sys.platform == 'win32':
 
1483
            # Martin (gz) mentioned WINDOW_BUFFER_SIZE_RECORD from
 
1484
            # ReadConsoleInput but I've no idea how to plug that in
 
1485
            # the current design -- vila 20091216
1522
1486
            pass
 
1487
        else:
 
1488
            set_signal_handler(signal.SIGWINCH, _terminal_size_changed)
1523
1489
        _registered_sigwinch = True
1524
1490
 
1525
1491
 
1846
1812
            real_handlers[kind](abspath, relpath)
1847
1813
 
1848
1814
 
1849
 
def copy_ownership_from_path(dst, src=None):
1850
 
    """Copy usr/grp ownership from src file/dir to dst file/dir.
1851
 
 
1852
 
    If src is None, the containing directory is used as source. If chown
1853
 
    fails, the error is ignored and a warning is printed.
1854
 
    """
1855
 
    chown = getattr(os, 'chown', None)
1856
 
    if chown is None:
1857
 
        return
1858
 
 
1859
 
    if src == None:
1860
 
        src = os.path.dirname(dst)
1861
 
        if src == '':
1862
 
            src = '.'
1863
 
 
1864
 
    try:
1865
 
        s = os.stat(src)
1866
 
        chown(dst, s.st_uid, s.st_gid)
1867
 
    except OSError, e:
1868
 
        trace.warning("Unable to copy ownership from '%s' to '%s': IOError: %s." % (src, dst, e))
1869
 
 
1870
 
 
1871
1815
def path_prefix_key(path):
1872
1816
    """Generate a prefix-order path key for path.
1873
1817