← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: Martin Pool
  • Date: 2010-03-26 10:25:47 UTC
  • mfrom: (5110.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 5118.
  • Revision ID: mbp@canonical.com-20100326102547-74ta65m7w7ecjebq
merge 2.1.1, including fetch format warning, back to trunk

Show diffs side-by-side

added added

removed removed

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