/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: Vincent Ladeuil
  • Date: 2010-04-20 14:04:54 UTC
  • mto: (5169.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 5170.
  • Revision ID: v.ladeuil+lp@free.fr-20100420140454-ixobg867y79ink8w
Final import fixes for bt.test_diff.

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(), """
27
29
from datetime import datetime
28
30
import errno
29
 
import getpass
30
31
from ntpath import (abspath as _nt_abspath,
31
32
                    join as _nt_join,
32
33
                    normpath as _nt_normpath,
38
39
from shutil import (
39
40
    rmtree,
40
41
    )
 
42
import signal
41
43
import socket
42
44
import subprocess
43
45
import tempfile
361
363
    return _win32_fixdrive(tempfile.mkdtemp(*args, **kwargs).replace('\\', '/'))
362
364
 
363
365
 
364
 
def _add_rename_error_details(e, old, new):
365
 
    new_e = OSError(e.errno, "failed to rename %s to %s: %s"
366
 
        % (old, new, e.strerror))
367
 
    new_e.filename = old
368
 
    new_e.to_filename = new
369
 
    return new_e
370
 
 
371
 
 
372
366
def _win32_rename(old, new):
373
367
    """We expect to be able to atomically replace 'new' with old.
374
368
 
376
370
    and then deleted.
377
371
    """
378
372
    try:
379
 
        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)
380
374
    except OSError, e:
381
375
        if e.errno in (errno.EPERM, errno.EACCES, errno.EBUSY, errno.EINVAL):
382
376
            # If we try to rename a non-existant file onto cwd, we get
387
381
        raise
388
382
 
389
383
 
390
 
def _wrapped_rename(old, new):
391
 
    """Rename a file or directory"""
392
 
    try:
393
 
        os.rename(old, new)
394
 
    except (IOError, OSError), e:
395
 
        # this is eventually called by all rename-like functions, so should 
396
 
        # catch all of them
397
 
        raise _add_rename_error_details(e, old, new)
398
 
 
399
 
 
400
384
def _mac_getcwd():
401
385
    return unicodedata.normalize('NFC', os.getcwdu())
402
386
 
407
391
realpath = _posix_realpath
408
392
pathjoin = os.path.join
409
393
normpath = os.path.normpath
410
 
rename = _wrapped_rename # overridden below on win32
411
394
getcwd = os.getcwdu
 
395
rename = os.rename
412
396
dirname = os.path.dirname
413
397
basename = os.path.basename
414
398
split = os.path.split
1148
1132
 
1149
1133
 
1150
1134
def relpath(base, path):
1151
 
    """Return path relative to base, or raise PathNotChild exception.
 
1135
    """Return path relative to base, or raise exception.
1152
1136
 
1153
1137
    The path may be either an absolute path or a path relative to the
1154
1138
    current working directory.
1156
1140
    os.path.commonprefix (python2.4) has a bad bug that it works just
1157
1141
    on string prefixes, assuming that '/u' is a prefix of '/u2'.  This
1158
1142
    avoids that problem.
1159
 
 
1160
 
    NOTE: `base` should not have a trailing slash otherwise you'll get
1161
 
    PathNotChild exceptions regardless of `path`.
1162
1143
    """
1163
1144
 
1164
1145
    if len(base) < MIN_ABS_PATHLENGTH:
1385
1366
        platform or Python version.
1386
1367
    """
1387
1368
    try:
1388
 
        import signal
1389
1369
        siginterrupt = signal.siginterrupt
1390
 
    except ImportError:
1391
 
        # This python implementation doesn't provide signal support, hence no
1392
 
        # handler exists
1393
 
        return None
1394
1370
    except AttributeError:
1395
1371
        # siginterrupt doesn't exist on this platform, or for this version
1396
1372
        # of Python.
1507
1483
 
1508
1484
 
1509
1485
_registered_sigwinch = False
 
1486
 
1510
1487
def watch_sigwinch():
1511
 
    """Register for SIGWINCH, once and only once.
1512
 
 
1513
 
    Do nothing if the signal module is not available.
1514
 
    """
 
1488
    """Register for SIGWINCH, once and only once."""
1515
1489
    global _registered_sigwinch
1516
1490
    if not _registered_sigwinch:
1517
 
        try:
1518
 
            import signal
1519
 
            if getattr(signal, "SIGWINCH", None) is not None:
1520
 
                set_signal_handler(signal.SIGWINCH, _terminal_size_changed)
1521
 
        except ImportError:
1522
 
            # 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
1523
1495
            pass
 
1496
        else:
 
1497
            set_signal_handler(signal.SIGWINCH, _terminal_size_changed)
1524
1498
        _registered_sigwinch = True
1525
1499
 
1526
1500
 
1847
1821
            real_handlers[kind](abspath, relpath)
1848
1822
 
1849
1823
 
1850
 
def copy_ownership_from_path(dst, src=None):
 
1824
def copy_ownership(dst, src=None):
1851
1825
    """Copy usr/grp ownership from src file/dir to dst file/dir.
1852
1826
 
1853
1827
    If src is None, the containing directory is used as source. If chown
1869
1843
        trace.warning("Unable to copy ownership from '%s' to '%s': IOError: %s." % (src, dst, e))
1870
1844
 
1871
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
 
1872
1870
def path_prefix_key(path):
1873
1871
    """Generate a prefix-order path key for path.
1874
1872
 
2303
2301
        return os.fdopen(os.open(filename, flags), mode, bufsize)
2304
2302
else:
2305
2303
    open_file = open
2306
 
 
2307
 
 
2308
 
def getuser_unicode():
2309
 
    """Return the username as unicode.
2310
 
    """
2311
 
    try:
2312
 
        user_encoding = get_user_encoding()
2313
 
        username = getpass.getuser().decode(user_encoding)
2314
 
    except UnicodeDecodeError:
2315
 
        raise errors.BzrError("Can't decode username as %s." % \
2316
 
                user_encoding)
2317
 
    return username