~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/osutils.py

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

• reverse the comparison (5462.5.3 to 5304)
• stop comparing with revision 5304

Show diffs side-by-side

added

removed

bzrlib/osutils.py

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

 

441

435

    getcwd = _mac_getcwd

442

436

 

443

437

 

444

 

def get_terminal_encoding():

 

438

def get_terminal_encoding(trace=False):

445

439

    """Find the best encoding for printing to the screen.

446

440

 

447

441

    This attempts to check both sys.stdout and sys.stdin to see

453

447

 

454

448

    On my standard US Windows XP, the preferred encoding is

455

449

    cp1252, but the console is cp437

 

450

 

 

451

    :param trace: If True trace the selected encoding via mutter().

456

452

    """

457

453

    from bzrlib.trace import mutter

458

454

    output_encoding = getattr(sys.stdout, 'encoding', None)

460

456

        input_encoding = getattr(sys.stdin, 'encoding', None)

461

457

        if not input_encoding:

462

458

            output_encoding = get_user_encoding()

463

 

            mutter('encoding stdout as osutils.get_user_encoding() %r',

 

459

            if trace:

 

460

                mutter('encoding stdout as osutils.get_user_encoding() %r',

464

461

                   output_encoding)

465

462

        else:

466

463

            output_encoding = input_encoding

467

 

            mutter('encoding stdout as sys.stdin encoding %r', output_encoding)

 

464

            if trace:

 

465

                mutter('encoding stdout as sys.stdin encoding %r',

 

466

                    output_encoding)

468

467

    else:

469

 

        mutter('encoding stdout as sys.stdout encoding %r', output_encoding)

 

468

        if trace:

 

469

            mutter('encoding stdout as sys.stdout encoding %r', output_encoding)

470

470

    if output_encoding == 'cp0':

471

471

        # invalid encoding (cp0 means 'no codepage' on Windows)

472

472

        output_encoding = get_user_encoding()

473

 

        mutter('cp0 is invalid encoding.'

 

473

        if trace:

 

474

            mutter('cp0 is invalid encoding.'

474

475

               ' encoding stdout as osutils.get_user_encoding() %r',

475

476

               output_encoding)

476

477

    # check encoding

502

503

def isdir(f):

503

504

    """True if f is an accessible directory."""

504

505

    try:

505

 

        return S_ISDIR(os.lstat(f)[ST_MODE])

 

506

        return stat.S_ISDIR(os.lstat(f)[stat.ST_MODE])

506

507

    except OSError:

507

508

        return False

508

509

 

510

511

def isfile(f):

511

512

    """True if f is a regular file."""

512

513

    try:

513

 

        return S_ISREG(os.lstat(f)[ST_MODE])

 

514

        return stat.S_ISREG(os.lstat(f)[stat.ST_MODE])

514

515

    except OSError:

515

516

        return False

516

517

 

517

518

def islink(f):

518

519

    """True if f is a symlink."""

519

520

    try:

520

 

        return S_ISLNK(os.lstat(f)[ST_MODE])

 

521

        return stat.S_ISLNK(os.lstat(f)[stat.ST_MODE])

521

522

    except OSError:

522

523

        return False

523

524

 

863

864

 

864

865

def filesize(f):

865

866

    """Return size of given open file."""

866

 

    return os.fstat(f.fileno())[ST_SIZE]

 

867

    return os.fstat(f.fileno())[stat.ST_SIZE]

867

868

 

868

869

 

869

870

# Define rand_bytes based on platform.

1968

1969

    return user_encoding

1969

1970

 

1970

1971

 

 

1972

def get_diff_header_encoding():

 

1973

    return get_terminal_encoding()

 

1974

 

 

1975

 

1971

1976

def get_host_name():

1972

1977

    """Return the current unicode host name.

1973

1978

 

2060

2065

            report_activity(sent, 'write')

2061

2066

 

2062

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

 

2063

2091

def dereference_path(path):

2064

2092

    """Determine the real path to a file.

2065

2093

 

2158

2186

            raise

2159

2187

 

2160

2188

 

 

2189

@deprecated_function(deprecated_in((2, 2, 0)))

2161

2190

def re_compile_checked(re_string, flags=0, where=""):

2162

2191

    """Return a compiled re, or raise a sensible error.

2163

2192

 

2173

2202

        re_obj = re.compile(re_string, flags)

2174

2203

        re_obj.search("")

2175

2204

        return re_obj

2176

 

    except re.error, e:

 

2205

    except errors.InvalidPattern, e:

2177

2206

        if where:

2178

2207

            where = ' in ' + where

2179

2208

        # despite the name 'error' is a type

2180

 

        raise errors.BzrCommandError('Invalid regular expression%s: %r: %s'

2181

 

            % (where, re_string, e))

 

2209

        raise errors.BzrCommandError('Invalid regular expression%s: %s'

 

2210

            % (where, e.msg))

2182

2211

 

2183

2212

 

2184

2213

if sys.platform == "win32":

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1