Mercurial > hg
annotate mercurial/win32.py @ 11322:3d6915f5a2bb
improve --branch processing (and differentiate from # syntax)
Previously #foo and --branch foo were handled identically.
The behavior of #foo hasn't changed, but --branch now works like this:
1) If branchmap is not supported on the remote, the operation fails.
2) If branch is '.', substitute with branch of the working dir parent.
3) If branch exists remotely, its heads are expanded.
4) Otherwise, the operation fails.
Tests have been added for the new cases.
author | Sune Foldager <cryo@cyanite.org> |
---|---|
date | Thu, 10 Jun 2010 12:46:09 +0200 |
parents | 8c377f2feee1 |
children | 50523b4407f6 |
rev | line source |
---|---|
8226
8b2cd04a6e97
put license and copyright info into comment blocks
Martin Geisler <mg@lazybytes.net>
parents:
8225
diff
changeset
|
1 # win32.py - utility functions that use win32 API |
8b2cd04a6e97
put license and copyright info into comment blocks
Martin Geisler <mg@lazybytes.net>
parents:
8225
diff
changeset
|
2 # |
8b2cd04a6e97
put license and copyright info into comment blocks
Martin Geisler <mg@lazybytes.net>
parents:
8225
diff
changeset
|
3 # Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others |
8b2cd04a6e97
put license and copyright info into comment blocks
Martin Geisler <mg@lazybytes.net>
parents:
8225
diff
changeset
|
4 # |
8b2cd04a6e97
put license and copyright info into comment blocks
Martin Geisler <mg@lazybytes.net>
parents:
8225
diff
changeset
|
5 # This software may be used and distributed according to the terms of the |
10263 | 6 # GNU General Public License version 2 or any later version. |
8227
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
7 |
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
8 """Utility functions that use win32 API. |
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
9 |
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
10 Mark Hammond's win32all package allows better functionality on |
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
11 Windows. This module overrides definitions in util.py. If not |
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
12 available, import of this module will fail, and generic code will be |
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
13 used. |
0a9542703300
turn some comments back into module docstrings
Martin Geisler <mg@lazybytes.net>
parents:
8226
diff
changeset
|
14 """ |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
15 |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
16 import win32api |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
17 |
6216 | 18 import errno, os, sys, pywintypes, win32con, win32file, win32process |
11012
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
19 import winerror, win32gui, win32console |
7948
de377b1a9a84
move encoding bits from util to encoding
Matt Mackall <mpm@selenic.com>
parents:
7890
diff
changeset
|
20 import osutil, encoding |
9198
061eeb602354
coding style: use a space after comma
Martin Geisler <mg@lazybytes.net>
parents:
8951
diff
changeset
|
21 from win32com.shell import shell, shellcon |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
22 |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
23 def os_link(src, dst): |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
24 try: |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
25 win32file.CreateHardLink(dst, src) |
5879
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
26 # CreateHardLink sometimes succeeds on mapped drives but |
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
27 # following nlinks() returns 1. Check it now and bail out. |
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
28 if nlinks(src) < 2: |
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
29 try: |
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
30 win32file.DeleteFile(dst) |
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
31 except: |
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
32 pass |
cacfeee38870
util_win32: make os_link more robust (issue 761)
Patrick Mezard <pmezard@gmail.com>
parents:
5611
diff
changeset
|
33 # Fake hardlinking error |
8951
835a51e63c5b
windows: fix use of undefined exception (issue1707)
Henrik Stuart <hg@hstuart.dk>
parents:
8656
diff
changeset
|
34 raise OSError(errno.EINVAL, 'Hardlinking not supported') |
11304
8c377f2feee1
cleanups: unused variables
Dirkjan Ochtman <dirkjan@ochtman.nl>
parents:
11012
diff
changeset
|
35 except pywintypes.error: |
8951
835a51e63c5b
windows: fix use of undefined exception (issue1707)
Henrik Stuart <hg@hstuart.dk>
parents:
8656
diff
changeset
|
36 raise OSError(errno.EINVAL, 'target implements hardlinks improperly') |
7778
82f7145b304c
Don't fail on clone on win98 (issue1492)
Matt Mackall <mpm@selenic.com>
parents:
7473
diff
changeset
|
37 except NotImplementedError: # Another fake error win Win98 |
8951
835a51e63c5b
windows: fix use of undefined exception (issue1707)
Henrik Stuart <hg@hstuart.dk>
parents:
8656
diff
changeset
|
38 raise OSError(errno.EINVAL, 'Hardlinking not supported') |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
39 |
10218
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
40 def _getfileinfo(pathname): |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
41 """Return number of hardlinks for the given file.""" |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
42 try: |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
43 fh = win32file.CreateFile(pathname, |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
44 win32file.GENERIC_READ, win32file.FILE_SHARE_READ, |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
45 None, win32file.OPEN_EXISTING, 0, None) |
10219
3b94120864fc
win32: close file when leaving _getfileinfo()
Patrick Mezard <pmezard@gmail.com>
parents:
10218
diff
changeset
|
46 try: |
3b94120864fc
win32: close file when leaving _getfileinfo()
Patrick Mezard <pmezard@gmail.com>
parents:
10218
diff
changeset
|
47 return win32file.GetFileInformationByHandle(fh) |
3b94120864fc
win32: close file when leaving _getfileinfo()
Patrick Mezard <pmezard@gmail.com>
parents:
10218
diff
changeset
|
48 finally: |
3b94120864fc
win32: close file when leaving _getfileinfo()
Patrick Mezard <pmezard@gmail.com>
parents:
10218
diff
changeset
|
49 fh.Close() |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
50 except pywintypes.error: |
10218
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
51 return None |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
52 |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
53 def nlinks(pathname): |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
54 """Return number of hardlinks for the given file.""" |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
55 res = _getfileinfo(pathname) |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
56 if res is not None: |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
57 return res[7] |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
58 else: |
2448
b77a2ef61b81
replace os.stat with os.lstat in some where.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
2313
diff
changeset
|
59 return os.lstat(pathname).st_nlink |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
60 |
10218
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
61 def samefile(fpath1, fpath2): |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
62 """Returns whether fpath1 and fpath2 refer to the same file. This is only |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
63 guaranteed to work for files, not directories.""" |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
64 res1 = _getfileinfo(fpath1) |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
65 res2 = _getfileinfo(fpath2) |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
66 if res1 is not None and res2 is not None: |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
67 # Index 4 is the volume serial number, and 8 and 9 contain the file ID |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
68 return res1[4] == res2[4] and res1[8] == res2[8] and res1[9] == res2[9] |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
69 else: |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
70 return False |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
71 |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
72 def samedevice(fpath1, fpath2): |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
73 """Returns whether fpath1 and fpath2 are on the same device. This is only |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
74 guaranteed to work for files, not directories.""" |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
75 res1 = _getfileinfo(fpath1) |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
76 res2 = _getfileinfo(fpath2) |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
77 if res1 is not None and res2 is not None: |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
78 return res1[4] == res2[4] |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
79 else: |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
80 return False |
750b7a4f01f6
Add support for relinking on Windows.
Siddharth Agarwal <sid.bugzilla@gmail.com>
parents:
9198
diff
changeset
|
81 |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
82 def testpid(pid): |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
83 '''return True if pid is still running or unable to |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
84 determine, False otherwise''' |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
85 try: |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
86 handle = win32api.OpenProcess( |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
87 win32con.PROCESS_QUERY_INFORMATION, False, pid) |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
88 if handle: |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
89 status = win32process.GetExitCodeProcess(handle) |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
90 return status == win32con.STILL_ACTIVE |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
91 except pywintypes.error, details: |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
92 return details[0] != winerror.ERROR_INVALID_PARAMETER |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
93 return True |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
94 |
6006
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
95 def lookup_reg(key, valname=None, scope=None): |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
96 ''' Look up a key/value name in the Windows registry. |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
97 |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
98 valname: value name. If unspecified, the default value for the key |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
99 is used. |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
100 scope: optionally specify scope for registry lookup, this can be |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
101 a sequence of scopes to look up in order. Default (CURRENT_USER, |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
102 LOCAL_MACHINE). |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
103 ''' |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
104 try: |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
105 from _winreg import HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE, \ |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
106 QueryValueEx, OpenKey |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
107 except ImportError: |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
108 return None |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
109 |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
110 if scope is None: |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
111 scope = (HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE) |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
112 elif not isinstance(scope, (list, tuple)): |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
113 scope = (scope,) |
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
114 for s in scope: |
6880
892806b3fc0f
util: disinfect lookup_reg strings (issue1126)
Matt Mackall <mpm@selenic.com>
parents:
6216
diff
changeset
|
115 try: |
6881
d2375bbee6d4
Folding correction and missing import
Remy Roy <remyroy@remyroy.com>
parents:
6880
diff
changeset
|
116 val = QueryValueEx(OpenKey(s, key), valname)[0] |
6880
892806b3fc0f
util: disinfect lookup_reg strings (issue1126)
Matt Mackall <mpm@selenic.com>
parents:
6216
diff
changeset
|
117 # never let a Unicode string escape into the wild |
7948
de377b1a9a84
move encoding bits from util to encoding
Matt Mackall <mpm@selenic.com>
parents:
7890
diff
changeset
|
118 return encoding.tolocal(val.encode('UTF-8')) |
6880
892806b3fc0f
util: disinfect lookup_reg strings (issue1126)
Matt Mackall <mpm@selenic.com>
parents:
6216
diff
changeset
|
119 except EnvironmentError: |
892806b3fc0f
util: disinfect lookup_reg strings (issue1126)
Matt Mackall <mpm@selenic.com>
parents:
6216
diff
changeset
|
120 pass |
6006
3c9dbb743d20
merge: add registry look up bits to tool search
Matt Mackall <mpm@selenic.com>
parents:
5969
diff
changeset
|
121 |
2117 | 122 def system_rcpath_win32(): |
2054
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
123 '''return default os-specific hgrc search path''' |
e18beba54a7e
fix exception handling on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
diff
changeset
|
124 proc = win32api.GetCurrentProcess() |
2285
0912f807b7ff
win98: fall back to win32api.GetModuleFileName if needed.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2284
diff
changeset
|
125 try: |
0912f807b7ff
win98: fall back to win32api.GetModuleFileName if needed.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2284
diff
changeset
|
126 # This will fail on windows < NT |
0912f807b7ff
win98: fall back to win32api.GetModuleFileName if needed.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2284
diff
changeset
|
127 filename = win32process.GetModuleFileNameEx(proc, 0) |
0912f807b7ff
win98: fall back to win32api.GetModuleFileName if needed.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2284
diff
changeset
|
128 except: |
0912f807b7ff
win98: fall back to win32api.GetModuleFileName if needed.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2284
diff
changeset
|
129 filename = win32api.GetModuleFileName(0) |
5619
55d3e845736a
win32: favor local mercurial.ini over registry key
Steve Borho <steve@borho.org>
parents:
5612
diff
changeset
|
130 # Use mercurial.ini found in directory with hg.exe |
55d3e845736a
win32: favor local mercurial.ini over registry key
Steve Borho <steve@borho.org>
parents:
5612
diff
changeset
|
131 progrc = os.path.join(os.path.dirname(filename), 'mercurial.ini') |
55d3e845736a
win32: favor local mercurial.ini over registry key
Steve Borho <steve@borho.org>
parents:
5612
diff
changeset
|
132 if os.path.isfile(progrc): |
55d3e845736a
win32: favor local mercurial.ini over registry key
Steve Borho <steve@borho.org>
parents:
5612
diff
changeset
|
133 return [progrc] |
10388
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
134 # Use hgrc.d found in directory with hg.exe |
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
135 progrcd = os.path.join(os.path.dirname(filename), 'hgrc.d') |
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
136 if os.path.isdir(progrcd): |
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
137 rcpath = [] |
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
138 for f, kind in osutil.listdir(progrcd): |
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
139 if f.endswith('.rc'): |
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
140 rcpath.append(os.path.join(progrcd, f)) |
07bd7608a0ea
win32: allow hgrc.d on Windows
Steve Borho <steve@borho.org>
parents:
10264
diff
changeset
|
141 return rcpath |
5619
55d3e845736a
win32: favor local mercurial.ini over registry key
Steve Borho <steve@borho.org>
parents:
5612
diff
changeset
|
142 # else look for a system rcpath in the registry |
5583
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
143 try: |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
144 value = win32api.RegQueryValue( |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
145 win32con.HKEY_LOCAL_MACHINE, 'SOFTWARE\\Mercurial') |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
146 rcpath = [] |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
147 for p in value.split(os.pathsep): |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
148 if p.lower().endswith('mercurial.ini'): |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
149 rcpath.append(p) |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
150 elif os.path.isdir(p): |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
151 for f, kind in osutil.listdir(p): |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
152 if f.endswith('.rc'): |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
153 rcpath.append(os.path.join(p, f)) |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
154 return rcpath |
1b5b81d9039b
win32: read system rcpath from registry
Steve Borho <steve@borho.org>
parents:
4998
diff
changeset
|
155 except pywintypes.error: |
5619
55d3e845736a
win32: favor local mercurial.ini over registry key
Steve Borho <steve@borho.org>
parents:
5612
diff
changeset
|
156 return [] |
2176
9b42304d9896
fix file handling bugs on windows.
Vadim Gelfer <vadim.gelfer@gmail.com>
parents:
2117
diff
changeset
|
157 |
4098
c08b6af023bc
util_win32.py: fix user_rcpath
Alexis S. L. Carvalho <alexis@cecm.usp.br>
parents:
3891
diff
changeset
|
158 def user_rcpath_win32(): |
2284
d6392a7c03dd
On win98 os.path.expanuser('~') does not result in a useable directory.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2250
diff
changeset
|
159 '''return os-specific hgrc search path to the user dir''' |
d6392a7c03dd
On win98 os.path.expanuser('~') does not result in a useable directory.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2250
diff
changeset
|
160 userdir = os.path.expanduser('~') |
7427
f21e3d0e335b
util_win32: fix Windows version checking (issue1358)
Patrick Mezard <pmezard@gmail.com>
parents:
7390
diff
changeset
|
161 if sys.getwindowsversion()[3] != 2 and userdir == '~': |
2284
d6392a7c03dd
On win98 os.path.expanuser('~') does not result in a useable directory.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2250
diff
changeset
|
162 # We are on win < nt: fetch the APPDATA directory location and use |
d6392a7c03dd
On win98 os.path.expanuser('~') does not result in a useable directory.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2250
diff
changeset
|
163 # the parent directory as the user home dir. |
d6392a7c03dd
On win98 os.path.expanuser('~') does not result in a useable directory.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2250
diff
changeset
|
164 appdir = shell.SHGetPathFromIDList( |
2313
a600d9997521
Fixed typo (qshell instead of shell) in win98 code (see issue244).
Thomas Arendsen Hein <thomas@intevation.de>
parents:
2285
diff
changeset
|
165 shell.SHGetSpecialFolderLocation(0, shellcon.CSIDL_APPDATA)) |
2284
d6392a7c03dd
On win98 os.path.expanuser('~') does not result in a useable directory.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2250
diff
changeset
|
166 userdir = os.path.dirname(appdir) |
6153
09a8be3e5bfb
Also search for .hgrc if mercurial.ini not found on windows
Stefan Rank <strank(AT)strank(DOT)info>
parents:
6012
diff
changeset
|
167 return [os.path.join(userdir, 'mercurial.ini'), |
09a8be3e5bfb
Also search for .hgrc if mercurial.ini not found on windows
Stefan Rank <strank(AT)strank(DOT)info>
parents:
6012
diff
changeset
|
168 os.path.join(userdir, '.hgrc')] |
2284
d6392a7c03dd
On win98 os.path.expanuser('~') does not result in a useable directory.
Volker Kleinfeld <Volker.Kleinfeld@gmx.de>
parents:
2250
diff
changeset
|
169 |
7890
e710f0f592b2
util: split out posix, windows, and win32 modules
Matt Mackall <mpm@selenic.com>
parents:
7778
diff
changeset
|
170 def getuser(): |
e710f0f592b2
util: split out posix, windows, and win32 modules
Matt Mackall <mpm@selenic.com>
parents:
7778
diff
changeset
|
171 '''return name of current user''' |
e710f0f592b2
util: split out posix, windows, and win32 modules
Matt Mackall <mpm@selenic.com>
parents:
7778
diff
changeset
|
172 return win32api.GetUserName() |
4672
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
173 |
4803
7549cd526b7f
Fix serve on Windows without win32* modules.
Nathan Jones <nathanj@insightbb.com>
parents:
4672
diff
changeset
|
174 def set_signal_handler_win32(): |
4672
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
175 """Register a termination handler for console events including |
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
176 CTRL+C. python signal handlers do not work well with socket |
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
177 operations. |
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
178 """ |
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
179 def handler(event): |
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
180 win32process.ExitProcess(1) |
272c0a09b203
Handle CTRL+C in serve under Windows.
Marcos Chaves <marcos.nospam@gmail.com>
parents:
4407
diff
changeset
|
181 win32api.SetConsoleCtrlHandler(handler) |
7890
e710f0f592b2
util: split out posix, windows, and win32 modules
Matt Mackall <mpm@selenic.com>
parents:
7778
diff
changeset
|
182 |
10240
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
183 def hidewindow(): |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
184 def callback(*args, **kwargs): |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
185 hwnd, pid = args |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
186 wpid = win32process.GetWindowThreadProcessId(hwnd)[1] |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
187 if pid == wpid: |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
188 win32gui.ShowWindow(hwnd, win32con.SW_HIDE) |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
189 |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
190 pid = win32process.GetCurrentProcessId() |
3af4b39afe2a
cmdutil: hide child window created by win32 spawndetached()
Patrick Mezard <pmezard@gmail.com>
parents:
10219
diff
changeset
|
191 win32gui.EnumWindows(callback, pid) |
11012
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
192 |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
193 def termwidth_(): |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
194 try: |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
195 # Query stderr to avoid problems with redirections |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
196 screenbuf = win32console.GetStdHandle(win32console.STD_ERROR_HANDLE) |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
197 try: |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
198 window = screenbuf.GetConsoleScreenBufferInfo()['Window'] |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
199 width = window.Right - window.Left |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
200 return width |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
201 finally: |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
202 screenbuf.Detach() |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
203 except pywintypes.error: |
81631f0cf13b
win32: detect console width on Windows
Patrick Mezard <pmezard@gmail.com>
parents:
10388
diff
changeset
|
204 return 79 |