/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

• browse files at revision 5121.2.2
• compare with another revision
• download diff
• download tarball
• view history from revision 5121.2.2

• reverse the comparison (5121.2.2 to 5200.3.7)
• stop comparing with revision 5200.3.7

Show diffs side-by-side

added

removed

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(), """

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:

1384

1365

        False)`).  May be ignored if the feature is not available on this

1385

1366

        platform or Python version.

1386

1367

    """

1387

 

    try:

1388

 

        import signal

1389

 

        siginterrupt = signal.siginterrupt

1390

 

    except ImportError:

1391

 

        # This python implementation doesn't provide signal support, hence no

1392

 

        # handler exists

1393

 

        return None

1394

 

    except AttributeError:

1395

 

        # siginterrupt doesn't exist on this platform, or for this version

1396

 

        # of Python.

1397

 

        siginterrupt = lambda signum, flag: None

 

1368

    old_handler = signal.signal(signum, handler)

1398

1369

    if restart_syscall:

1399

 

        def sig_handler(*args):

1400

 

            # Python resets the siginterrupt flag when a signal is

1401

 

            # received.  <http://bugs.python.org/issue8354>

1402

 

            # 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:

1403

1376

            siginterrupt(signum, False)

1404

 

            # Now run the handler function passed to set_signal_handler.

1405

 

            handler(*args)

1406

 

    else:

1407

 

        sig_handler = handler

1408

 

    old_handler = signal.signal(signum, sig_handler)

1409

 

    if restart_syscall:

1410

 

        siginterrupt(signum, False)

1411

1377

    return old_handler

1412

1378

 

1413

1379

 

1507

1473

 

1508

1474

 

1509

1475

_registered_sigwinch = False

 

1476

 

1510

1477

def watch_sigwinch():

1511

 

    """Register for SIGWINCH, once and only once.

1512

 

 

1513

 

    Do nothing if the signal module is not available.

1514

 

    """

 

1478

    """Register for SIGWINCH, once and only once."""

1515

1479

    global _registered_sigwinch

1516

1480

    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

 

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

1523

1485

            pass

 

1486

        else:

 

1487

            set_signal_handler(signal.SIGWINCH, _terminal_size_changed)

1524

1488

        _registered_sigwinch = True

1525

1489

 

1526

1490

 

1847

1811

            real_handlers[kind](abspath, relpath)

1848

1812

 

1849

1813

 

1850

 

def copy_ownership_from_path(dst, src=None):

 

1814

def copy_ownership(dst, src=None):

1851

1815

    """Copy usr/grp ownership from src file/dir to dst file/dir.

1852

1816

 

1853

1817

    If src is None, the containing directory is used as source. If chown

1869

1833

        trace.warning("Unable to copy ownership from '%s' to '%s': IOError: %s." % (src, dst, e))

1870

1834

 

1871

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

 

1872

1860

def path_prefix_key(path):

1873

1861

    """Generate a prefix-order path key for path.

1874

1862

 

2303

2291

        return os.fdopen(os.open(filename, flags), mode, bufsize)

2304

2292

else:

2305

2293

    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

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0