##// END OF EJS Templates
windows: provide filename in IOError exceptions...
Steve Borho -
r9448:bc6b0fef default
parent child Browse files
Show More
@@ -1,283 +1,283 b''
1 # windows.py - Windows utility function implementations for Mercurial
1 # windows.py - Windows utility function implementations for Mercurial
2 #
2 #
3 # Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others
3 # Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2, incorporated herein by reference.
6 # GNU General Public License version 2, incorporated herein by reference.
7
7
8 from i18n import _
8 from i18n import _
9 import osutil, error
9 import osutil, error
10 import errno, msvcrt, os, re, sys
10 import errno, msvcrt, os, re, sys
11
11
12 nulldev = 'NUL:'
12 nulldev = 'NUL:'
13 umask = 002
13 umask = 002
14
14
15 # wrap osutil.posixfile to provide friendlier exceptions
15 # wrap osutil.posixfile to provide friendlier exceptions
16 def posixfile(name, mode='r', buffering=-1):
16 def posixfile(name, mode='r', buffering=-1):
17 try:
17 try:
18 return osutil.posixfile(name, mode, buffering)
18 return osutil.posixfile(name, mode, buffering)
19 except WindowsError, err:
19 except WindowsError, err:
20 raise IOError(err.errno, err.strerror)
20 raise IOError(err.errno, '%s: %s' % (name, err.strerror))
21 posixfile.__doc__ = osutil.posixfile.__doc__
21 posixfile.__doc__ = osutil.posixfile.__doc__
22
22
23 class winstdout(object):
23 class winstdout(object):
24 '''stdout on windows misbehaves if sent through a pipe'''
24 '''stdout on windows misbehaves if sent through a pipe'''
25
25
26 def __init__(self, fp):
26 def __init__(self, fp):
27 self.fp = fp
27 self.fp = fp
28
28
29 def __getattr__(self, key):
29 def __getattr__(self, key):
30 return getattr(self.fp, key)
30 return getattr(self.fp, key)
31
31
32 def close(self):
32 def close(self):
33 try:
33 try:
34 self.fp.close()
34 self.fp.close()
35 except: pass
35 except: pass
36
36
37 def write(self, s):
37 def write(self, s):
38 try:
38 try:
39 # This is workaround for "Not enough space" error on
39 # This is workaround for "Not enough space" error on
40 # writing large size of data to console.
40 # writing large size of data to console.
41 limit = 16000
41 limit = 16000
42 l = len(s)
42 l = len(s)
43 start = 0
43 start = 0
44 self.softspace = 0;
44 self.softspace = 0;
45 while start < l:
45 while start < l:
46 end = start + limit
46 end = start + limit
47 self.fp.write(s[start:end])
47 self.fp.write(s[start:end])
48 start = end
48 start = end
49 except IOError, inst:
49 except IOError, inst:
50 if inst.errno != 0: raise
50 if inst.errno != 0: raise
51 self.close()
51 self.close()
52 raise IOError(errno.EPIPE, 'Broken pipe')
52 raise IOError(errno.EPIPE, 'Broken pipe')
53
53
54 def flush(self):
54 def flush(self):
55 try:
55 try:
56 return self.fp.flush()
56 return self.fp.flush()
57 except IOError, inst:
57 except IOError, inst:
58 if inst.errno != errno.EINVAL: raise
58 if inst.errno != errno.EINVAL: raise
59 self.close()
59 self.close()
60 raise IOError(errno.EPIPE, 'Broken pipe')
60 raise IOError(errno.EPIPE, 'Broken pipe')
61
61
62 sys.stdout = winstdout(sys.stdout)
62 sys.stdout = winstdout(sys.stdout)
63
63
64 def _is_win_9x():
64 def _is_win_9x():
65 '''return true if run on windows 95, 98 or me.'''
65 '''return true if run on windows 95, 98 or me.'''
66 try:
66 try:
67 return sys.getwindowsversion()[3] == 1
67 return sys.getwindowsversion()[3] == 1
68 except AttributeError:
68 except AttributeError:
69 return 'command' in os.environ.get('comspec', '')
69 return 'command' in os.environ.get('comspec', '')
70
70
71 def openhardlinks():
71 def openhardlinks():
72 return not _is_win_9x() and "win32api" in globals()
72 return not _is_win_9x() and "win32api" in globals()
73
73
74 def system_rcpath():
74 def system_rcpath():
75 try:
75 try:
76 return system_rcpath_win32()
76 return system_rcpath_win32()
77 except:
77 except:
78 return [r'c:\mercurial\mercurial.ini']
78 return [r'c:\mercurial\mercurial.ini']
79
79
80 def user_rcpath():
80 def user_rcpath():
81 '''return os-specific hgrc search path to the user dir'''
81 '''return os-specific hgrc search path to the user dir'''
82 try:
82 try:
83 path = user_rcpath_win32()
83 path = user_rcpath_win32()
84 except:
84 except:
85 home = os.path.expanduser('~')
85 home = os.path.expanduser('~')
86 path = [os.path.join(home, 'mercurial.ini'),
86 path = [os.path.join(home, 'mercurial.ini'),
87 os.path.join(home, '.hgrc')]
87 os.path.join(home, '.hgrc')]
88 userprofile = os.environ.get('USERPROFILE')
88 userprofile = os.environ.get('USERPROFILE')
89 if userprofile:
89 if userprofile:
90 path.append(os.path.join(userprofile, 'mercurial.ini'))
90 path.append(os.path.join(userprofile, 'mercurial.ini'))
91 path.append(os.path.join(userprofile, '.hgrc'))
91 path.append(os.path.join(userprofile, '.hgrc'))
92 return path
92 return path
93
93
94 def parse_patch_output(output_line):
94 def parse_patch_output(output_line):
95 """parses the output produced by patch and returns the filename"""
95 """parses the output produced by patch and returns the filename"""
96 pf = output_line[14:]
96 pf = output_line[14:]
97 if pf[0] == '`':
97 if pf[0] == '`':
98 pf = pf[1:-1] # Remove the quotes
98 pf = pf[1:-1] # Remove the quotes
99 return pf
99 return pf
100
100
101 def sshargs(sshcmd, host, user, port):
101 def sshargs(sshcmd, host, user, port):
102 '''Build argument list for ssh or Plink'''
102 '''Build argument list for ssh or Plink'''
103 pflag = 'plink' in sshcmd.lower() and '-P' or '-p'
103 pflag = 'plink' in sshcmd.lower() and '-P' or '-p'
104 args = user and ("%s@%s" % (user, host)) or host
104 args = user and ("%s@%s" % (user, host)) or host
105 return port and ("%s %s %s" % (args, pflag, port)) or args
105 return port and ("%s %s %s" % (args, pflag, port)) or args
106
106
107 def testpid(pid):
107 def testpid(pid):
108 '''return False if pid dead, True if running or not known'''
108 '''return False if pid dead, True if running or not known'''
109 return True
109 return True
110
110
111 def set_flags(f, l, x):
111 def set_flags(f, l, x):
112 pass
112 pass
113
113
114 def set_binary(fd):
114 def set_binary(fd):
115 # When run without console, pipes may expose invalid
115 # When run without console, pipes may expose invalid
116 # fileno(), usually set to -1.
116 # fileno(), usually set to -1.
117 if hasattr(fd, 'fileno') and fd.fileno() >= 0:
117 if hasattr(fd, 'fileno') and fd.fileno() >= 0:
118 msvcrt.setmode(fd.fileno(), os.O_BINARY)
118 msvcrt.setmode(fd.fileno(), os.O_BINARY)
119
119
120 def pconvert(path):
120 def pconvert(path):
121 return '/'.join(path.split(os.sep))
121 return '/'.join(path.split(os.sep))
122
122
123 def localpath(path):
123 def localpath(path):
124 return path.replace('/', '\\')
124 return path.replace('/', '\\')
125
125
126 def normpath(path):
126 def normpath(path):
127 return pconvert(os.path.normpath(path))
127 return pconvert(os.path.normpath(path))
128
128
129 def samestat(s1, s2):
129 def samestat(s1, s2):
130 return False
130 return False
131
131
132 # A sequence of backslashes is special iff it precedes a double quote:
132 # A sequence of backslashes is special iff it precedes a double quote:
133 # - if there's an even number of backslashes, the double quote is not
133 # - if there's an even number of backslashes, the double quote is not
134 # quoted (i.e. it ends the quoted region)
134 # quoted (i.e. it ends the quoted region)
135 # - if there's an odd number of backslashes, the double quote is quoted
135 # - if there's an odd number of backslashes, the double quote is quoted
136 # - in both cases, every pair of backslashes is unquoted into a single
136 # - in both cases, every pair of backslashes is unquoted into a single
137 # backslash
137 # backslash
138 # (See http://msdn2.microsoft.com/en-us/library/a1y7w461.aspx )
138 # (See http://msdn2.microsoft.com/en-us/library/a1y7w461.aspx )
139 # So, to quote a string, we must surround it in double quotes, double
139 # So, to quote a string, we must surround it in double quotes, double
140 # the number of backslashes that preceed double quotes and add another
140 # the number of backslashes that preceed double quotes and add another
141 # backslash before every double quote (being careful with the double
141 # backslash before every double quote (being careful with the double
142 # quote we've appended to the end)
142 # quote we've appended to the end)
143 _quotere = None
143 _quotere = None
144 def shellquote(s):
144 def shellquote(s):
145 global _quotere
145 global _quotere
146 if _quotere is None:
146 if _quotere is None:
147 _quotere = re.compile(r'(\\*)("|\\$)')
147 _quotere = re.compile(r'(\\*)("|\\$)')
148 return '"%s"' % _quotere.sub(r'\1\1\\\2', s)
148 return '"%s"' % _quotere.sub(r'\1\1\\\2', s)
149
149
150 def quotecommand(cmd):
150 def quotecommand(cmd):
151 """Build a command string suitable for os.popen* calls."""
151 """Build a command string suitable for os.popen* calls."""
152 # The extra quotes are needed because popen* runs the command
152 # The extra quotes are needed because popen* runs the command
153 # through the current COMSPEC. cmd.exe suppress enclosing quotes.
153 # through the current COMSPEC. cmd.exe suppress enclosing quotes.
154 return '"' + cmd + '"'
154 return '"' + cmd + '"'
155
155
156 def popen(command, mode='r'):
156 def popen(command, mode='r'):
157 # Work around "popen spawned process may not write to stdout
157 # Work around "popen spawned process may not write to stdout
158 # under windows"
158 # under windows"
159 # http://bugs.python.org/issue1366
159 # http://bugs.python.org/issue1366
160 command += " 2> %s" % nulldev
160 command += " 2> %s" % nulldev
161 return os.popen(quotecommand(command), mode)
161 return os.popen(quotecommand(command), mode)
162
162
163 def explain_exit(code):
163 def explain_exit(code):
164 return _("exited with status %d") % code, code
164 return _("exited with status %d") % code, code
165
165
166 # if you change this stub into a real check, please try to implement the
166 # if you change this stub into a real check, please try to implement the
167 # username and groupname functions above, too.
167 # username and groupname functions above, too.
168 def isowner(st):
168 def isowner(st):
169 return True
169 return True
170
170
171 def find_exe(command):
171 def find_exe(command):
172 '''Find executable for command searching like cmd.exe does.
172 '''Find executable for command searching like cmd.exe does.
173 If command is a basename then PATH is searched for command.
173 If command is a basename then PATH is searched for command.
174 PATH isn't searched if command is an absolute or relative path.
174 PATH isn't searched if command is an absolute or relative path.
175 An extension from PATHEXT is found and added if not present.
175 An extension from PATHEXT is found and added if not present.
176 If command isn't found None is returned.'''
176 If command isn't found None is returned.'''
177 pathext = os.environ.get('PATHEXT', '.COM;.EXE;.BAT;.CMD')
177 pathext = os.environ.get('PATHEXT', '.COM;.EXE;.BAT;.CMD')
178 pathexts = [ext for ext in pathext.lower().split(os.pathsep)]
178 pathexts = [ext for ext in pathext.lower().split(os.pathsep)]
179 if os.path.splitext(command)[1].lower() in pathexts:
179 if os.path.splitext(command)[1].lower() in pathexts:
180 pathexts = ['']
180 pathexts = ['']
181
181
182 def findexisting(pathcommand):
182 def findexisting(pathcommand):
183 'Will append extension (if needed) and return existing file'
183 'Will append extension (if needed) and return existing file'
184 for ext in pathexts:
184 for ext in pathexts:
185 executable = pathcommand + ext
185 executable = pathcommand + ext
186 if os.path.exists(executable):
186 if os.path.exists(executable):
187 return executable
187 return executable
188 return None
188 return None
189
189
190 if os.sep in command:
190 if os.sep in command:
191 return findexisting(command)
191 return findexisting(command)
192
192
193 for path in os.environ.get('PATH', '').split(os.pathsep):
193 for path in os.environ.get('PATH', '').split(os.pathsep):
194 executable = findexisting(os.path.join(path, command))
194 executable = findexisting(os.path.join(path, command))
195 if executable is not None:
195 if executable is not None:
196 return executable
196 return executable
197 return None
197 return None
198
198
199 def set_signal_handler():
199 def set_signal_handler():
200 try:
200 try:
201 set_signal_handler_win32()
201 set_signal_handler_win32()
202 except NameError:
202 except NameError:
203 pass
203 pass
204
204
205 def statfiles(files):
205 def statfiles(files):
206 '''Stat each file in files and yield stat or None if file does not exist.
206 '''Stat each file in files and yield stat or None if file does not exist.
207 Cluster and cache stat per directory to minimize number of OS stat calls.'''
207 Cluster and cache stat per directory to minimize number of OS stat calls.'''
208 ncase = os.path.normcase
208 ncase = os.path.normcase
209 sep = os.sep
209 sep = os.sep
210 dircache = {} # dirname -> filename -> status | None if file does not exist
210 dircache = {} # dirname -> filename -> status | None if file does not exist
211 for nf in files:
211 for nf in files:
212 nf = ncase(nf)
212 nf = ncase(nf)
213 dir, base = os.path.split(nf)
213 dir, base = os.path.split(nf)
214 if not dir:
214 if not dir:
215 dir = '.'
215 dir = '.'
216 cache = dircache.get(dir, None)
216 cache = dircache.get(dir, None)
217 if cache is None:
217 if cache is None:
218 try:
218 try:
219 dmap = dict([(ncase(n), s)
219 dmap = dict([(ncase(n), s)
220 for n, k, s in osutil.listdir(dir, True)])
220 for n, k, s in osutil.listdir(dir, True)])
221 except OSError, err:
221 except OSError, err:
222 # handle directory not found in Python version prior to 2.5
222 # handle directory not found in Python version prior to 2.5
223 # Python <= 2.4 returns native Windows code 3 in errno
223 # Python <= 2.4 returns native Windows code 3 in errno
224 # Python >= 2.5 returns ENOENT and adds winerror field
224 # Python >= 2.5 returns ENOENT and adds winerror field
225 # EINVAL is raised if dir is not a directory.
225 # EINVAL is raised if dir is not a directory.
226 if err.errno not in (3, errno.ENOENT, errno.EINVAL,
226 if err.errno not in (3, errno.ENOENT, errno.EINVAL,
227 errno.ENOTDIR):
227 errno.ENOTDIR):
228 raise
228 raise
229 dmap = {}
229 dmap = {}
230 cache = dircache.setdefault(dir, dmap)
230 cache = dircache.setdefault(dir, dmap)
231 yield cache.get(base, None)
231 yield cache.get(base, None)
232
232
233 def getuser():
233 def getuser():
234 '''return name of current user'''
234 '''return name of current user'''
235 raise error.Abort(_('user name not available - set USERNAME '
235 raise error.Abort(_('user name not available - set USERNAME '
236 'environment variable'))
236 'environment variable'))
237
237
238 def username(uid=None):
238 def username(uid=None):
239 """Return the name of the user with the given uid.
239 """Return the name of the user with the given uid.
240
240
241 If uid is None, return the name of the current user."""
241 If uid is None, return the name of the current user."""
242 return None
242 return None
243
243
244 def groupname(gid=None):
244 def groupname(gid=None):
245 """Return the name of the group with the given gid.
245 """Return the name of the group with the given gid.
246
246
247 If gid is None, return the name of the current group."""
247 If gid is None, return the name of the current group."""
248 return None
248 return None
249
249
250 def _removedirs(name):
250 def _removedirs(name):
251 """special version of os.removedirs that does not remove symlinked
251 """special version of os.removedirs that does not remove symlinked
252 directories or junction points if they actually contain files"""
252 directories or junction points if they actually contain files"""
253 if osutil.listdir(name):
253 if osutil.listdir(name):
254 return
254 return
255 os.rmdir(name)
255 os.rmdir(name)
256 head, tail = os.path.split(name)
256 head, tail = os.path.split(name)
257 if not tail:
257 if not tail:
258 head, tail = os.path.split(head)
258 head, tail = os.path.split(head)
259 while head and tail:
259 while head and tail:
260 try:
260 try:
261 if osutil.listdir(name):
261 if osutil.listdir(name):
262 return
262 return
263 os.rmdir(head)
263 os.rmdir(head)
264 except:
264 except:
265 break
265 break
266 head, tail = os.path.split(head)
266 head, tail = os.path.split(head)
267
267
268 def unlink(f):
268 def unlink(f):
269 """unlink and remove the directory if it is empty"""
269 """unlink and remove the directory if it is empty"""
270 os.unlink(f)
270 os.unlink(f)
271 # try removing directories that might now be empty
271 # try removing directories that might now be empty
272 try:
272 try:
273 _removedirs(os.path.dirname(f))
273 _removedirs(os.path.dirname(f))
274 except OSError:
274 except OSError:
275 pass
275 pass
276
276
277 try:
277 try:
278 # override functions with win32 versions if possible
278 # override functions with win32 versions if possible
279 from win32 import *
279 from win32 import *
280 except ImportError:
280 except ImportError:
281 pass
281 pass
282
282
283 expandglobs = True
283 expandglobs = True
General Comments 0
You need to be logged in to leave comments. Login now