/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: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
37
37
from shutil import (
38
38
    rmtree,
39
39
    )
40
 
import signal
41
40
import socket
42
41
import subprocess
43
42
import tempfile
361
360
    return _win32_fixdrive(tempfile.mkdtemp(*args, **kwargs).replace('\\', '/'))
362
361
 
363
362
 
 
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
 
364
371
def _win32_rename(old, new):
365
372
    """We expect to be able to atomically replace 'new' with old.
366
373
 
368
375
    and then deleted.
369
376
    """
370
377
    try:
371
 
        fancy_rename(old, new, rename_func=os.rename, unlink_func=os.unlink)
 
378
        fancy_rename(old, new, rename_func=_wrapped_rename, unlink_func=os.unlink)
372
379
    except OSError, e:
373
380
        if e.errno in (errno.EPERM, errno.EACCES, errno.EBUSY, errno.EINVAL):
374
381
            # If we try to rename a non-existant file onto cwd, we get
379
386
        raise
380
387
 
381
388
 
 
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
 
382
399
def _mac_getcwd():
383
400
    return unicodedata.normalize('NFC', os.getcwdu())
384
401
 
389
406
realpath = _posix_realpath
390
407
pathjoin = os.path.join
391
408
normpath = os.path.normpath
 
409
rename = _wrapped_rename # overridden below on win32
392
410
getcwd = os.getcwdu
393
 
rename = os.rename
394
411
dirname = os.path.dirname
395
412
basename = os.path.basename
396
413
split = os.path.split
1130
1147
 
1131
1148
 
1132
1149
def relpath(base, path):
1133
 
    """Return path relative to base, or raise exception.
 
1150
    """Return path relative to base, or raise PathNotChild exception.
1134
1151
 
1135
1152
    The path may be either an absolute path or a path relative to the
1136
1153
    current working directory.
1138
1155
    os.path.commonprefix (python2.4) has a bad bug that it works just
1139
1156
    on string prefixes, assuming that '/u' is a prefix of '/u2'.  This
1140
1157
    avoids that problem.
 
1158
 
 
1159
    NOTE: `base` should not have a trailing slash otherwise you'll get
 
1160
    PathNotChild exceptions regardless of `path`.
1141
1161
    """
1142
1162
 
1143
1163
    if len(base) < MIN_ABS_PATHLENGTH:
1363
1383
        False)`).  May be ignored if the feature is not available on this
1364
1384
        platform or Python version.
1365
1385
    """
1366
 
    old_handler = signal.signal(signum, handler)
 
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
1367
1397
    if restart_syscall:
1368
 
        try:
1369
 
            siginterrupt = signal.siginterrupt
1370
 
        except AttributeError: # siginterrupt doesn't exist on this platform, or for this version of
1371
 
            # Python.
1372
 
            pass
1373
 
        else:
 
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.
1374
1402
            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)
1375
1410
    return old_handler
1376
1411
 
1377
1412
 
1471
1506
 
1472
1507
 
1473
1508
_registered_sigwinch = False
1474
 
 
1475
1509
def watch_sigwinch():
1476
 
    """Register for SIGWINCH, once and only once."""
 
1510
    """Register for SIGWINCH, once and only once.
 
1511
 
 
1512
    Do nothing if the signal module is not available.
 
1513
    """
1477
1514
    global _registered_sigwinch
1478
1515
    if not _registered_sigwinch:
1479
 
        if sys.platform == 'win32':
1480
 
            # Martin (gz) mentioned WINDOW_BUFFER_SIZE_RECORD from
1481
 
            # ReadConsoleInput but I've no idea how to plug that in
1482
 
            # the current design -- vila 20091216
 
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
1483
1522
            pass
1484
 
        else:
1485
 
            set_signal_handler(signal.SIGWINCH, _terminal_size_changed)
1486
1523
        _registered_sigwinch = True
1487
1524
 
1488
1525
 
1809
1846
            real_handlers[kind](abspath, relpath)
1810
1847
 
1811
1848
 
1812
 
def copy_ownership(dst, src=None):
 
1849
def copy_ownership_from_path(dst, src=None):
1813
1850
    """Copy usr/grp ownership from src file/dir to dst file/dir.
1814
1851
 
1815
1852
    If src is None, the containing directory is used as source. If chown
1831
1868
        trace.warning("Unable to copy ownership from '%s' to '%s': IOError: %s." % (src, dst, e))
1832
1869
 
1833
1870
 
1834
 
def mkdir_with_ownership(path, ownership_src=None):
1835
 
    """Create the directory 'path' with specified ownership.
1836
 
 
1837
 
    If ownership_src is given, copies (chown) usr/grp ownership
1838
 
    from 'ownership_src' to 'path'. If ownership_src is None, use the
1839
 
    containing dir ownership.
1840
 
    """
1841
 
    os.mkdir(path)
1842
 
    copy_ownership(path, ownership_src)
1843
 
 
1844
 
 
1845
 
def open_with_ownership(filename, mode='r', bufsize=-1, ownership_src=None):
1846
 
    """Open the file 'filename' with the specified ownership.
1847
 
 
1848
 
    If ownership_src is specified, copy usr/grp ownership from ownership_src
1849
 
    to filename. If ownership_src is None, copy ownership from containing
1850
 
    directory.
1851
 
    Returns the opened file object.
1852
 
    """
1853
 
    f = open(filename, mode, bufsize)
1854
 
    copy_ownership(filename, ownership_src)
1855
 
    return f
1856
 
 
1857
 
 
1858
1871
def path_prefix_key(path):
1859
1872
    """Generate a prefix-order path key for path.
1860
1873