~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
18
18
import os
19
19
import re
20
20
import stat
21
 
from stat import S_ISREG, S_ISDIR, S_ISLNK, ST_MODE, ST_SIZE
22
21
import sys
23
22
import time
24
23
import codecs
27
26
lazy_import(globals(), """
28
27
from datetime import datetime
29
28
import getpass
30
 
from ntpath import (abspath as _nt_abspath,
31
 
                    join as _nt_join,
32
 
                    normpath as _nt_normpath,
33
 
                    realpath as _nt_realpath,
34
 
                    splitdrive as _nt_splitdrive,
35
 
                    )
 
29
import ntpath
36
30
import posixpath
 
31
# We need to import both shutil and rmtree as we export the later on posix
 
32
# and need the former on windows
37
33
import shutil
38
 
from shutil import (
39
 
    rmtree,
40
 
    )
 
34
from shutil import rmtree
41
35
import socket
42
36
import subprocess
 
37
# We need to import both tempfile and mkdtemp as we export the later on posix
 
38
# and need the former on windows
43
39
import tempfile
44
 
from tempfile import (
45
 
    mkdtemp,
46
 
    )
 
40
from tempfile import mkdtemp
47
41
import unicodedata
48
42
 
49
43
from bzrlib import (
304
298
    running python.exe under cmd.exe return capital C:\\
305
299
    running win32 python inside a cygwin shell returns lowercase c:\\
306
300
    """
307
 
    drive, path = _nt_splitdrive(path)
 
301
    drive, path = ntpath.splitdrive(path)
308
302
    return drive.upper() + path
309
303
 
310
304
 
311
305
def _win32_abspath(path):
312
 
    # Real _nt_abspath doesn't have a problem with a unicode cwd
313
 
    return _win32_fixdrive(_nt_abspath(unicode(path)).replace('\\', '/'))
 
306
    # Real ntpath.abspath doesn't have a problem with a unicode cwd
 
307
    return _win32_fixdrive(ntpath.abspath(unicode(path)).replace('\\', '/'))
314
308
 
315
309
 
316
310
def _win98_abspath(path):
327
321
    #   /path       => C:/path
328
322
    path = unicode(path)
329
323
    # check for absolute path
330
 
    drive = _nt_splitdrive(path)[0]
 
324
    drive = ntpath.splitdrive(path)[0]
331
325
    if drive == '' and path[:2] not in('//','\\\\'):
332
326
        cwd = os.getcwdu()
333
327
        # we cannot simply os.path.join cwd and path
334
328
        # because os.path.join('C:','/path') produce '/path'
335
329
        # and this is incorrect
336
330
        if path[:1] in ('/','\\'):
337
 
            cwd = _nt_splitdrive(cwd)[0]
 
331
            cwd = ntpath.splitdrive(cwd)[0]
338
332
            path = path[1:]
339
333
        path = cwd + '\\' + path
340
 
    return _win32_fixdrive(_nt_normpath(path).replace('\\', '/'))
 
334
    return _win32_fixdrive(ntpath.normpath(path).replace('\\', '/'))
341
335
 
342
336
 
343
337
def _win32_realpath(path):
344
 
    # Real _nt_realpath doesn't have a problem with a unicode cwd
345
 
    return _win32_fixdrive(_nt_realpath(unicode(path)).replace('\\', '/'))
 
338
    # Real ntpath.realpath doesn't have a problem with a unicode cwd
 
339
    return _win32_fixdrive(ntpath.realpath(unicode(path)).replace('\\', '/'))
346
340
 
347
341
 
348
342
def _win32_pathjoin(*args):
349
 
    return _nt_join(*args).replace('\\', '/')
 
343
    return ntpath.join(*args).replace('\\', '/')
350
344
 
351
345
 
352
346
def _win32_normpath(path):
353
 
    return _win32_fixdrive(_nt_normpath(unicode(path)).replace('\\', '/'))
 
347
    return _win32_fixdrive(ntpath.normpath(unicode(path)).replace('\\', '/'))
354
348
 
355
349
 
356
350
def _win32_getcwd():
395
389
basename = os.path.basename
396
390
split = os.path.split
397
391
splitext = os.path.splitext
398
 
# These were already imported into local scope
 
392
# These were already lazily imported into local scope
399
393
# mkdtemp = tempfile.mkdtemp
400
394
# rmtree = shutil.rmtree
401
395
 
509
503
def isdir(f):
510
504
    """True if f is an accessible directory."""
511
505
    try:
512
 
        return S_ISDIR(os.lstat(f)[ST_MODE])
 
506
        return stat.S_ISDIR(os.lstat(f)[stat.ST_MODE])
513
507
    except OSError:
514
508
        return False
515
509
 
517
511
def isfile(f):
518
512
    """True if f is a regular file."""
519
513
    try:
520
 
        return S_ISREG(os.lstat(f)[ST_MODE])
 
514
        return stat.S_ISREG(os.lstat(f)[stat.ST_MODE])
521
515
    except OSError:
522
516
        return False
523
517
 
524
518
def islink(f):
525
519
    """True if f is a symlink."""
526
520
    try:
527
 
        return S_ISLNK(os.lstat(f)[ST_MODE])
 
521
        return stat.S_ISLNK(os.lstat(f)[stat.ST_MODE])
528
522
    except OSError:
529
523
        return False
530
524
 
870
864
 
871
865
def filesize(f):
872
866
    """Return size of given open file."""
873
 
    return os.fstat(f.fileno())[ST_SIZE]
 
867
    return os.fstat(f.fileno())[stat.ST_SIZE]
874
868
 
875
869
 
876
870
# Define rand_bytes based on platform.
2071
2065
            report_activity(sent, 'write')
2072
2066
 
2073
2067
 
 
2068
def connect_socket(address):
 
2069
    # Slight variation of the socket.create_connection() function (provided by
 
2070
    # python-2.6) that can fail if getaddrinfo returns an empty list. We also
 
2071
    # provide it for previous python versions. Also, we don't use the timeout
 
2072
    # parameter (provided by the python implementation) so we don't implement
 
2073
    # it either).
 
2074
    err = socket.error('getaddrinfo returns an empty list')
 
2075
    host, port = address
 
2076
    for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM):
 
2077
        af, socktype, proto, canonname, sa = res
 
2078
        sock = None
 
2079
        try:
 
2080
            sock = socket.socket(af, socktype, proto)
 
2081
            sock.connect(sa)
 
2082
            return sock
 
2083
 
 
2084
        except socket.error, err:
 
2085
            # 'err' is now the most recent error
 
2086
            if sock is not None:
 
2087
                sock.close()
 
2088
    raise err
 
2089
 
 
2090
 
2074
2091
def dereference_path(path):
2075
2092
    """Determine the real path to a file.
2076
2093
 
2169
2186
            raise
2170
2187
 
2171
2188
 
 
2189
@deprecated_function(deprecated_in((2, 2, 0)))
2172
2190
def re_compile_checked(re_string, flags=0, where=""):
2173
2191
    """Return a compiled re, or raise a sensible error.
2174
2192
 
2184
2202
        re_obj = re.compile(re_string, flags)
2185
2203
        re_obj.search("")
2186
2204
        return re_obj
2187
 
    except re.error, e:
 
2205
    except errors.InvalidPattern, e:
2188
2206
        if where:
2189
2207
            where = ' in ' + where
2190
2208
        # despite the name 'error' is a type
2191
 
        raise errors.BzrCommandError('Invalid regular expression%s: %r: %s'
2192
 
            % (where, re_string, e))
 
2209
        raise errors.BzrCommandError('Invalid regular expression%s: %s'
 
2210
            % (where, e.msg))
2193
2211
 
2194
2212
 
2195
2213
if sys.platform == "win32":