/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/win32utils.py

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

• reverse the comparison (5200.3.2 to 4505.5.9)
• stop comparing with revision 4505.5.9

Show diffs side-by-side

added

removed

bzrlib/win32utils.py

1

 

# Copyright (C) 2006, 2007 Canonical Ltd

 

1

# Copyright (C) 2005-2010 Canonical Ltd

2

2

#

3

3

# This program is free software; you can redistribute it and/or modify

4

4

# it under the terms of the GNU General Public License as published by

19

19

Only one dependency: ctypes should be installed.

20

20

"""

21

21

 

 

22

import glob

22

23

import os

23

24

import struct

24

25

import sys

25

26

 

 

27

from bzrlib import cmdline

26

28

 

27

29

# Windows version

28

30

if sys.platform == 'win32':

66

68

        suffix = 'W'

67

69

try:

68

70

    import win32file

 

71

    import pywintypes

69

72

    has_win32file = True

70

73

except ImportError:

71

74

    has_win32file = False

132

135

                'WorkingSetSize': mem_struct.WorkingSetSize,

133

136

                'QuotaPeakPagedPoolUsage': mem_struct.QuotaPeakPagedPoolUsage,

134

137

                'QuotaPagedPoolUsage': mem_struct.QuotaPagedPoolUsage,

135

 

                'QuotaPeakNonPagedPoolUsage': mem_struct.QuotaPeakNonPagedPoolUsage,

 

138

                'QuotaPeakNonPagedPoolUsage':

 

139

                    mem_struct.QuotaPeakNonPagedPoolUsage,

136

140

                'QuotaNonPagedPoolUsage': mem_struct.QuotaNonPagedPoolUsage,

137

141

                'PagefileUsage': mem_struct.PagefileUsage,

138

142

                'PeakPagefileUsage': mem_struct.PeakPagefileUsage,

149

153

                   ' or win32process')

150

154

        return

151

155

    if short:

152

 

        trace.note('WorkingSize %7dKB'

153

 

                   '\tPeakWorking %7dKB\t%s',

 

156

        # using base-2 units (see HACKING.txt).

 

157

        trace.note('WorkingSize %7dKiB'

 

158

                   '\tPeakWorking %7dKiB\t%s',

154

159

                   info['WorkingSetSize'] / 1024,

155

160

                   info['PeakWorkingSetSize'] / 1024,

156

161

                   message)

157

162

        return

158

163

    if message:

159

164

        trace.note('%s', message)

160

 

    trace.note('WorkingSize       %8d KB', info['WorkingSetSize'] / 1024)

161

 

    trace.note('PeakWorking       %8d KB', info['PeakWorkingSetSize'] / 1024)

162

 

    trace.note('PagefileUsage     %8d KB', info.get('PagefileUsage', 0) / 1024)

163

 

    trace.note('PeakPagefileUsage %8d KB', info.get('PeakPagefileUsage', 0) / 1024)

164

 

    trace.note('PrivateUsage      %8d KB', info.get('PrivateUsage', 0) / 1024)

 

165

    trace.note('WorkingSize       %8d KiB', info['WorkingSetSize'] / 1024)

 

166

    trace.note('PeakWorking       %8d KiB', info['PeakWorkingSetSize'] / 1024)

 

167

    trace.note('PagefileUsage     %8d KiB', info.get('PagefileUsage', 0) / 1024)

 

168

    trace.note('PeakPagefileUsage %8d KiB',

 

169

               info.get('PeakPagefileUsage', 0) / 1024)

 

170

    trace.note('PrivateUsage      %8d KiB', info.get('PrivateUsage', 0) / 1024)

165

171

    trace.note('PageFaultCount    %8d', info.get('PageFaultCount', 0))

166

172

 

167

173

 

177

183

        return (defaultx, defaulty)

178

184

 

179

185

    # To avoid problem with redirecting output via pipe

180

 

    # need to use stderr instead of stdout

 

186

    # we need to use stderr instead of stdout

181

187

    h = ctypes.windll.kernel32.GetStdHandle(WIN32_STDERR_HANDLE)

182

188

    csbi = ctypes.create_string_buffer(22)

183

189

    res = ctypes.windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)

184

190

 

185

191

    if res:

186

192

        (bufx, bufy, curx, cury, wattr,

187

 

        left, top, right, bottom, maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)

 

193

        left, top, right, bottom, maxx, maxy) = struct.unpack(

 

194

            "hhhhHhhhhhh", csbi.raw)

188

195

        sizex = right - left + 1

189

196

        sizey = bottom - top + 1

190

197

        return (sizex, sizey)

387

394

 

388

395

 

389

396

def _ensure_unicode(s):

390

 

    from bzrlib import osutils

391

397

    if s and type(s) != unicode:

392

398

        from bzrlib import osutils

393

399

        s = s.decode(osutils.get_user_encoding())

408

414

 

409

415

 

410

416

def _ensure_with_dir(path):

411

 

    if not os.path.split(path)[0] or path.startswith(u'*') or path.startswith(u'?'):

 

417

    if (not os.path.split(path)[0] or path.startswith(u'*')

 

418

        or path.startswith(u'?')):

412

419

        return u'./' + path, True

413

420

    else:

414

421

        return path, False

421

428

 

422

429

 

423

430

 

 

431

def glob_one(possible_glob):

 

432

    """Same as glob.glob().

 

433

 

 

434

    work around bugs in glob.glob()

 

435

    - Python bug #1001604 ("glob doesn't return unicode with ...")

 

436

    - failing expansion for */* with non-iso-8859-* chars

 

437

    """

 

438

    corrected_glob, corrected = _ensure_with_dir(possible_glob)

 

439

    glob_files = glob.glob(corrected_glob)

 

440

 

 

441

    if not glob_files:

 

442

        # special case to let the normal code path handle

 

443

        # files that do not exist, etc.

 

444

        glob_files = [possible_glob]

 

445

    elif corrected:

 

446

        glob_files = [_undo_ensure_with_dir(elem, corrected)

 

447

                      for elem in glob_files]

 

448

    return [elem.replace(u'\\', u'/') for elem in glob_files]

 

449

 

 

450

 

424

451

def glob_expand(file_list):

425

452

    """Replacement for glob expansion by the shell.

426

453

 

434

461

    """

435

462

    if not file_list:

436

463

        return []

437

 

    import glob

438

464

    expanded_file_list = []

439

465

    for possible_glob in file_list:

440

 

        # work around bugs in glob.glob()

441

 

        # - Python bug #1001604 ("glob doesn't return unicode with ...")

442

 

        # - failing expansion for */* with non-iso-8859-* chars

443

 

        possible_glob, corrected = _ensure_with_dir(possible_glob)

444

 

        glob_files = glob.glob(possible_glob)

445

 

 

446

 

        if glob_files == []:

447

 

            # special case to let the normal code path handle

448

 

            # files that do not exists

449

 

            expanded_file_list.append(

450

 

                _undo_ensure_with_dir(possible_glob, corrected))

451

 

        else:

452

 

            glob_files = [_undo_ensure_with_dir(elem, corrected) for elem in glob_files]

453

 

            expanded_file_list += glob_files

454

 

 

455

 

    return [elem.replace(u'\\', u'/') for elem in expanded_file_list]

 

466

        expanded_file_list.extend(glob_one(possible_glob))

 

467

    return expanded_file_list

456

468

 

457

469

 

458

470

def get_app_path(appname):

499

511

def set_file_attr_hidden(path):

500

512

    """Set file attributes to hidden if possible"""

501

513

    if has_win32file:

502

 

        win32file.SetFileAttributes(path, win32file.FILE_ATTRIBUTE_HIDDEN)

 

514

        if winver != 'Windows 98':

 

515

            SetFileAttributes = win32file.SetFileAttributesW

 

516

        else:

 

517

            SetFileAttributes = win32file.SetFileAttributes

 

518

        try:

 

519

            SetFileAttributes(path, win32file.FILE_ATTRIBUTE_HIDDEN)

 

520

        except pywintypes.error, e:

 

521

            from bzrlib import trace

 

522

            trace.mutter('Unable to set hidden attribute on %r: %s', path, e)

 

523

 

 

524

 

 

525

def _command_line_to_argv(command_line, single_quotes_allowed=False):

 

526

    """Convert a Unicode command line into a list of argv arguments.

 

527

 

 

528

    It performs wildcard expansion to make wildcards act closer to how they

 

529

    work in posix shells, versus how they work by default on Windows. Quoted

 

530

    arguments are left untouched.

 

531

 

 

532

    :param command_line: The unicode string to split into an arg list.

 

533

    :param single_quotes_allowed: Whether single quotes are accepted as quoting

 

534

                                  characters like double quotes. False by

 

535

                                  default.

 

536

    :return: A list of unicode strings.

 

537

    """

 

538

    s = cmdline.Splitter(command_line, single_quotes_allowed=single_quotes_allowed)

 

539

    # Now that we've split the content, expand globs if necessary

 

540

    # TODO: Use 'globbing' instead of 'glob.glob', this gives us stuff like

 

541

    #       '**/' style globs

 

542

    args = []

 

543

    for is_quoted, arg in s:

 

544

        if is_quoted or not glob.has_magic(arg):

 

545

            args.append(arg)

 

546

        else:

 

547

            args.extend(glob_one(arg))

 

548

    return args

503

549

 

504

550

 

505

551

if has_ctypes and winver != 'Windows 98':

506

552

    def get_unicode_argv():

507

 

        LPCWSTR = ctypes.c_wchar_p

508

 

        INT = ctypes.c_int

509

 

        POINTER = ctypes.POINTER

510

 

        prototype = ctypes.WINFUNCTYPE(LPCWSTR)

511

 

        GetCommandLine = prototype(("GetCommandLineW",

512

 

                                    ctypes.windll.kernel32))

513

 

        prototype = ctypes.WINFUNCTYPE(POINTER(LPCWSTR), LPCWSTR, POINTER(INT))

514

 

        CommandLineToArgv = prototype(("CommandLineToArgvW",

515

 

                                       ctypes.windll.shell32))

516

 

        c = INT(0)

517

 

        pargv = CommandLineToArgv(GetCommandLine(), ctypes.byref(c))

 

553

        prototype = ctypes.WINFUNCTYPE(ctypes.c_wchar_p)

 

554

        GetCommandLineW = prototype(("GetCommandLineW",

 

555

                                     ctypes.windll.kernel32))

 

556

        command_line = GetCommandLineW()

 

557

        if command_line is None:

 

558

            raise ctypes.WinError()

518

559

        # Skip the first argument, since we only care about parameters

519

 

        argv = [pargv[i] for i in range(1, c.value)]

 

560

        argv = _command_line_to_argv(command_line)[1:]

520

561

        if getattr(sys, 'frozen', None) is None:

521

562

            # Invoked via 'python.exe' which takes the form:

522

563

            #   python.exe [PYTHON_OPTIONS] C:\Path\bzr [BZR_OPTIONS]

523

564

            # we need to get only BZR_OPTIONS part,

524

 

            # so let's using sys.argv[1:] as reference to get the tail

525

 

            # of unicode argv

526

 

            tail_len = len(sys.argv[1:])

527

 

            ix = len(argv) - tail_len

528

 

            argv = argv[ix:]

 

565

            # We already removed 'python.exe' so we remove everything up to and

 

566

            # including the first non-option ('-') argument.

 

567

            for idx in xrange(len(argv)):

 

568

                if argv[idx][:1] != '-':

 

569

                    break

 

570

            argv = argv[idx+1:]

529

571

        return argv

530

572

else:

531

573

    get_unicode_argv = None

• Older »

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