~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: Vincent Ladeuil
  • Date: 2010-04-22 10:20:40 UTC
  • mfrom: (5050.3.4 2.2)
  • mto: This revision was merged to the branch mainline in revision 5173.
  • Revision ID: v.ladeuil+lp@free.fr-20100422102040-3nv01vm15i8qjip0
Merge 2.2 into bzr.dev

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
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:
1384
1366
        platform or Python version.
1385
1367
    """
1386
1368
    try:
1387
 
        import signal
1388
1369
        siginterrupt = signal.siginterrupt
1389
 
    except ImportError:
1390
 
        # This python implementation doesn't provide signal support, hence no
1391
 
        # handler exists
1392
 
        return None
1393
1370
    except AttributeError:
1394
1371
        # siginterrupt doesn't exist on this platform, or for this version
1395
1372
        # of Python.
1506
1483
 
1507
1484
 
1508
1485
_registered_sigwinch = False
 
1486
 
1509
1487
def watch_sigwinch():
1510
 
    """Register for SIGWINCH, once and only once.
1511
 
 
1512
 
    Do nothing if the signal module is not available.
1513
 
    """
 
1488
    """Register for SIGWINCH, once and only once."""
1514
1489
    global _registered_sigwinch
1515
1490
    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
 
1491
        if sys.platform == 'win32':
 
1492
            # Martin (gz) mentioned WINDOW_BUFFER_SIZE_RECORD from
 
1493
            # ReadConsoleInput but I've no idea how to plug that in
 
1494
            # the current design -- vila 20091216
1522
1495
            pass
 
1496
        else:
 
1497
            set_signal_handler(signal.SIGWINCH, _terminal_size_changed)
1523
1498
        _registered_sigwinch = True
1524
1499
 
1525
1500
 
1846
1821
            real_handlers[kind](abspath, relpath)
1847
1822
 
1848
1823
 
1849
 
def copy_ownership_from_path(dst, src=None):
 
1824
def copy_ownership(dst, src=None):
1850
1825
    """Copy usr/grp ownership from src file/dir to dst file/dir.
1851
1826
 
1852
1827
    If src is None, the containing directory is used as source. If chown
1868
1843
        trace.warning("Unable to copy ownership from '%s' to '%s': IOError: %s." % (src, dst, e))
1869
1844
 
1870
1845
 
 
1846
def mkdir_with_ownership(path, ownership_src=None):
 
1847
    """Create the directory 'path' with specified ownership.
 
1848
 
 
1849
    If ownership_src is given, copies (chown) usr/grp ownership
 
1850
    from 'ownership_src' to 'path'. If ownership_src is None, use the
 
1851
    containing dir ownership.
 
1852
    """
 
1853
    os.mkdir(path)
 
1854
    copy_ownership(path, ownership_src)
 
1855
 
 
1856
 
 
1857
def open_with_ownership(filename, mode='r', bufsize=-1, ownership_src=None):
 
1858
    """Open the file 'filename' with the specified ownership.
 
1859
 
 
1860
    If ownership_src is specified, copy usr/grp ownership from ownership_src
 
1861
    to filename. If ownership_src is None, copy ownership from containing
 
1862
    directory.
 
1863
    Returns the opened file object.
 
1864
    """
 
1865
    f = open(filename, mode, bufsize)
 
1866
    copy_ownership(filename, ownership_src)
 
1867
    return f
 
1868
 
 
1869
 
1871
1870
def path_prefix_key(path):
1872
1871
    """Generate a prefix-order path key for path.
1873
1872