/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 4145.1.6)
• stop comparing with revision 4145.1.6

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

12

12

#

13

13

# You should have received a copy of the GNU General Public License

14

14

# along with this program; if not, write to the Free Software

15

 

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 

15

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

16

16

 

17

17

"""Win32-specific helper functions

18

18

 

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

96

99

UNLEN = 256

97

100

MAX_COMPUTERNAME_LENGTH = 31

98

101

 

 

102

# Registry data type ids

 

103

REG_SZ = 1

 

104

REG_EXPAND_SZ = 2

 

105

 

99

106

 

100

107

def debug_memory_win32api(message='', short=True):

101

108

    """Use trace.note() to dump the running memory info."""

128

135

                'WorkingSetSize': mem_struct.WorkingSetSize,

129

136

                'QuotaPeakPagedPoolUsage': mem_struct.QuotaPeakPagedPoolUsage,

130

137

                'QuotaPagedPoolUsage': mem_struct.QuotaPagedPoolUsage,

131

 

                'QuotaPeakNonPagedPoolUsage': mem_struct.QuotaPeakNonPagedPoolUsage,

 

138

                'QuotaPeakNonPagedPoolUsage':

 

139

                    mem_struct.QuotaPeakNonPagedPoolUsage,

132

140

                'QuotaNonPagedPoolUsage': mem_struct.QuotaNonPagedPoolUsage,

133

141

                'PagefileUsage': mem_struct.PagefileUsage,

134

142

                'PeakPagefileUsage': mem_struct.PeakPagefileUsage,

144

152

        trace.note('Cannot debug memory on win32 without ctypes'

145

153

                   ' or win32process')

146

154

        return

147

 

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

148

 

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

149

155

    if short:

 

156

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

 

157

        trace.note('WorkingSize %7dKiB'

 

158

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

 

159

                   info['WorkingSetSize'] / 1024,

 

160

                   info['PeakWorkingSetSize'] / 1024,

 

161

                   message)

150

162

        return

151

 

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

152

 

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

153

 

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

 

163

    if message:

 

164

        trace.note('%s', message)

 

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)

154

171

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

155

172

 

156

173

 

166

183

        return (defaultx, defaulty)

167

184

 

168

185

    # To avoid problem with redirecting output via pipe

169

 

    # need to use stderr instead of stdout

 

186

    # we need to use stderr instead of stdout

170

187

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

171

188

    csbi = ctypes.create_string_buffer(22)

172

189

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

173

190

 

174

191

    if res:

175

192

        (bufx, bufy, curx, cury, wattr,

176

 

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

 

193

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

 

194

            "hhhhHhhhhhh", csbi.raw)

177

195

        sizex = right - left + 1

178

196

        sizey = bottom - top + 1

179

197

        return (sizex, sizey)

260

278

 

261

279

    Returned value can be unicode or plain string.

262

280

    To convert plain string to unicode use

263

 

    s.decode(bzrlib.user_encoding)

 

281

    s.decode(osutils.get_user_encoding())

264

282

    (XXX - but see bug 262874, which asserts the correct encoding is 'mbcs')

265

283

    """

266

284

    local = _get_sh_special_folder_path(CSIDL_LOCAL_APPDATA)

279

297

    If location cannot be obtained return system drive root,

280

298

    i.e. C:\

281

299

 

282

 

    Returned value can be unicode or plain sring.

 

300

    Returned value can be unicode or plain string.

283

301

    To convert plain string to unicode use

284

302

    s.decode(osutils.get_user_encoding())

285

303

    """

302

320

    """Return user name as login name.

303

321

    If name cannot be obtained return None.

304

322

 

305

 

    Returned value can be unicode or plain sring.

 

323

    Returned value can be unicode or plain string.

306

324

    To convert plain string to unicode use

307

325

    s.decode(osutils.get_user_encoding())

308

326

    """

376

394

 

377

395

 

378

396

def _ensure_unicode(s):

379

 

    from bzrlib import osutils

380

397

    if s and type(s) != unicode:

381

398

        from bzrlib import osutils

382

399

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

397

414

 

398

415

 

399

416

def _ensure_with_dir(path):

400

 

    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'?')):

401

419

        return u'./' + path, True

402

420

    else:

403

421

        return path, False

410

428

 

411

429

 

412

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

 

413

451

def glob_expand(file_list):

414

452

    """Replacement for glob expansion by the shell.

415

453

 

423

461

    """

424

462

    if not file_list:

425

463

        return []

426

 

    import glob

427

464

    expanded_file_list = []

428

465

    for possible_glob in file_list:

429

 

 

430

 

        # work around bugs in glob.glob()

431

 

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

432

 

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

433

 

        possible_glob, corrected = _ensure_with_dir(possible_glob)

434

 

        glob_files = glob.glob(possible_glob)

435

 

 

436

 

        if glob_files == []:

437

 

            # special case to let the normal code path handle

438

 

            # files that do not exists

439

 

            expanded_file_list.append(

440

 

                _undo_ensure_with_dir(possible_glob, corrected))

441

 

        else:

442

 

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

443

 

            expanded_file_list += glob_files

444

 

 

445

 

    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

446

468

 

447

469

 

448

470

def get_app_path(appname):

449

471

    """Look up in Windows registry for full path to application executable.

450

 

    Typicaly, applications create subkey with their basename

 

472

    Typically, applications create subkey with their basename

451

473

    in HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\App Paths\

452

474

 

453

475

    :param  appname:    name of application (if no filename extension

456

478

                or appname itself if nothing found.

457

479

    """

458

480

    import _winreg

 

481

 

 

482

    basename = appname

 

483

    if not os.path.splitext(basename)[1]:

 

484

        basename = appname + '.exe'

 

485

 

459

486

    try:

460

487

        hkey = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE,

461

 

                               r'SOFTWARE\Microsoft\Windows'

462

 

                               r'\CurrentVersion\App Paths')

 

488

            'SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\App Paths\\' +

 

489

            basename)

463

490

    except EnvironmentError:

464

491

        return appname

465

492

 

466

 

    basename = appname

467

 

    if not os.path.splitext(basename)[1]:

468

 

        basename = appname + '.exe'

469

493

    try:

470

494

        try:

471

 

            fullpath = _winreg.QueryValue(hkey, basename)

 

495

            path, type_id = _winreg.QueryValueEx(hkey, '')

472

496

        except WindowsError:

473

 

            fullpath = appname

 

497

            return appname

474

498

    finally:

475

499

        _winreg.CloseKey(hkey)

476

500

 

477

 

    return fullpath

 

501

    if type_id == REG_SZ:

 

502

        return path

 

503

    if type_id == REG_EXPAND_SZ and has_win32api:

 

504

        fullpath = win32api.ExpandEnvironmentStrings(path)

 

505

        if len(fullpath) > 1 and fullpath[0] == '"' and fullpath[-1] == '"':

 

506

            fullpath = fullpath[1:-1]   # remove quotes around value

 

507

        return fullpath

 

508

    return appname

478

509

 

479

510

 

480

511

def set_file_attr_hidden(path):

481

512

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

482

513

    if has_win32file:

483

 

        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

 

549

 

 

550

 

 

551

if has_ctypes and winver != 'Windows 98':

 

552

    def get_unicode_argv():

 

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

 

559

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

 

560

        argv = _command_line_to_argv(command_line)[1:]

 

561

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

 

562

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

 

563

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

 

564

            # we need to get only BZR_OPTIONS part,

 

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

 

571

        return argv

 

572

else:

 

573

    get_unicode_argv = None

• Older »

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