##// END OF EJS Templates
chunkbuffer: removed unused method and arg
Matt Mackall -
r5446:fa836e05 default
parent child Browse files
Show More
@@ -1,1701 +1,1688 b''
1 """
1 """
2 util.py - Mercurial utility functions and platform specfic implementations
2 util.py - Mercurial utility functions and platform specfic implementations
3
3
4 Copyright 2005 K. Thananchayan <thananck@yahoo.com>
4 Copyright 2005 K. Thananchayan <thananck@yahoo.com>
5 Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
5 Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
6 Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
6 Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
7
7
8 This software may be used and distributed according to the terms
8 This software may be used and distributed according to the terms
9 of the GNU General Public License, incorporated herein by reference.
9 of the GNU General Public License, incorporated herein by reference.
10
10
11 This contains helper routines that are independent of the SCM core and hide
11 This contains helper routines that are independent of the SCM core and hide
12 platform-specific details from the core.
12 platform-specific details from the core.
13 """
13 """
14
14
15 from i18n import _
15 from i18n import _
16 import cStringIO, errno, getpass, popen2, re, shutil, sys, tempfile, strutil
16 import cStringIO, errno, getpass, popen2, re, shutil, sys, tempfile, strutil
17 import os, stat, threading, time, calendar, ConfigParser, locale, glob, osutil
17 import os, stat, threading, time, calendar, ConfigParser, locale, glob, osutil
18
18
19 try:
19 try:
20 set = set
20 set = set
21 frozenset = frozenset
21 frozenset = frozenset
22 except NameError:
22 except NameError:
23 from sets import Set as set, ImmutableSet as frozenset
23 from sets import Set as set, ImmutableSet as frozenset
24
24
25 try:
25 try:
26 _encoding = os.environ.get("HGENCODING")
26 _encoding = os.environ.get("HGENCODING")
27 if sys.platform == 'darwin' and not _encoding:
27 if sys.platform == 'darwin' and not _encoding:
28 # On darwin, getpreferredencoding ignores the locale environment and
28 # On darwin, getpreferredencoding ignores the locale environment and
29 # always returns mac-roman. We override this if the environment is
29 # always returns mac-roman. We override this if the environment is
30 # not C (has been customized by the user).
30 # not C (has been customized by the user).
31 locale.setlocale(locale.LC_CTYPE, '')
31 locale.setlocale(locale.LC_CTYPE, '')
32 _encoding = locale.getlocale()[1]
32 _encoding = locale.getlocale()[1]
33 if not _encoding:
33 if not _encoding:
34 _encoding = locale.getpreferredencoding() or 'ascii'
34 _encoding = locale.getpreferredencoding() or 'ascii'
35 except locale.Error:
35 except locale.Error:
36 _encoding = 'ascii'
36 _encoding = 'ascii'
37 _encodingmode = os.environ.get("HGENCODINGMODE", "strict")
37 _encodingmode = os.environ.get("HGENCODINGMODE", "strict")
38 _fallbackencoding = 'ISO-8859-1'
38 _fallbackencoding = 'ISO-8859-1'
39
39
40 def tolocal(s):
40 def tolocal(s):
41 """
41 """
42 Convert a string from internal UTF-8 to local encoding
42 Convert a string from internal UTF-8 to local encoding
43
43
44 All internal strings should be UTF-8 but some repos before the
44 All internal strings should be UTF-8 but some repos before the
45 implementation of locale support may contain latin1 or possibly
45 implementation of locale support may contain latin1 or possibly
46 other character sets. We attempt to decode everything strictly
46 other character sets. We attempt to decode everything strictly
47 using UTF-8, then Latin-1, and failing that, we use UTF-8 and
47 using UTF-8, then Latin-1, and failing that, we use UTF-8 and
48 replace unknown characters.
48 replace unknown characters.
49 """
49 """
50 for e in ('UTF-8', _fallbackencoding):
50 for e in ('UTF-8', _fallbackencoding):
51 try:
51 try:
52 u = s.decode(e) # attempt strict decoding
52 u = s.decode(e) # attempt strict decoding
53 return u.encode(_encoding, "replace")
53 return u.encode(_encoding, "replace")
54 except LookupError, k:
54 except LookupError, k:
55 raise Abort(_("%s, please check your locale settings") % k)
55 raise Abort(_("%s, please check your locale settings") % k)
56 except UnicodeDecodeError:
56 except UnicodeDecodeError:
57 pass
57 pass
58 u = s.decode("utf-8", "replace") # last ditch
58 u = s.decode("utf-8", "replace") # last ditch
59 return u.encode(_encoding, "replace")
59 return u.encode(_encoding, "replace")
60
60
61 def fromlocal(s):
61 def fromlocal(s):
62 """
62 """
63 Convert a string from the local character encoding to UTF-8
63 Convert a string from the local character encoding to UTF-8
64
64
65 We attempt to decode strings using the encoding mode set by
65 We attempt to decode strings using the encoding mode set by
66 HGENCODINGMODE, which defaults to 'strict'. In this mode, unknown
66 HGENCODINGMODE, which defaults to 'strict'. In this mode, unknown
67 characters will cause an error message. Other modes include
67 characters will cause an error message. Other modes include
68 'replace', which replaces unknown characters with a special
68 'replace', which replaces unknown characters with a special
69 Unicode character, and 'ignore', which drops the character.
69 Unicode character, and 'ignore', which drops the character.
70 """
70 """
71 try:
71 try:
72 return s.decode(_encoding, _encodingmode).encode("utf-8")
72 return s.decode(_encoding, _encodingmode).encode("utf-8")
73 except UnicodeDecodeError, inst:
73 except UnicodeDecodeError, inst:
74 sub = s[max(0, inst.start-10):inst.start+10]
74 sub = s[max(0, inst.start-10):inst.start+10]
75 raise Abort("decoding near '%s': %s!" % (sub, inst))
75 raise Abort("decoding near '%s': %s!" % (sub, inst))
76 except LookupError, k:
76 except LookupError, k:
77 raise Abort(_("%s, please check your locale settings") % k)
77 raise Abort(_("%s, please check your locale settings") % k)
78
78
79 def locallen(s):
79 def locallen(s):
80 """Find the length in characters of a local string"""
80 """Find the length in characters of a local string"""
81 return len(s.decode(_encoding, "replace"))
81 return len(s.decode(_encoding, "replace"))
82
82
83 def localsub(s, a, b=None):
83 def localsub(s, a, b=None):
84 try:
84 try:
85 u = s.decode(_encoding, _encodingmode)
85 u = s.decode(_encoding, _encodingmode)
86 if b is not None:
86 if b is not None:
87 u = u[a:b]
87 u = u[a:b]
88 else:
88 else:
89 u = u[:a]
89 u = u[:a]
90 return u.encode(_encoding, _encodingmode)
90 return u.encode(_encoding, _encodingmode)
91 except UnicodeDecodeError, inst:
91 except UnicodeDecodeError, inst:
92 sub = s[max(0, inst.start-10), inst.start+10]
92 sub = s[max(0, inst.start-10), inst.start+10]
93 raise Abort(_("decoding near '%s': %s!") % (sub, inst))
93 raise Abort(_("decoding near '%s': %s!") % (sub, inst))
94
94
95 # used by parsedate
95 # used by parsedate
96 defaultdateformats = (
96 defaultdateformats = (
97 '%Y-%m-%d %H:%M:%S',
97 '%Y-%m-%d %H:%M:%S',
98 '%Y-%m-%d %I:%M:%S%p',
98 '%Y-%m-%d %I:%M:%S%p',
99 '%Y-%m-%d %H:%M',
99 '%Y-%m-%d %H:%M',
100 '%Y-%m-%d %I:%M%p',
100 '%Y-%m-%d %I:%M%p',
101 '%Y-%m-%d',
101 '%Y-%m-%d',
102 '%m-%d',
102 '%m-%d',
103 '%m/%d',
103 '%m/%d',
104 '%m/%d/%y',
104 '%m/%d/%y',
105 '%m/%d/%Y',
105 '%m/%d/%Y',
106 '%a %b %d %H:%M:%S %Y',
106 '%a %b %d %H:%M:%S %Y',
107 '%a %b %d %I:%M:%S%p %Y',
107 '%a %b %d %I:%M:%S%p %Y',
108 '%a, %d %b %Y %H:%M:%S', # GNU coreutils "/bin/date --rfc-2822"
108 '%a, %d %b %Y %H:%M:%S', # GNU coreutils "/bin/date --rfc-2822"
109 '%b %d %H:%M:%S %Y',
109 '%b %d %H:%M:%S %Y',
110 '%b %d %I:%M:%S%p %Y',
110 '%b %d %I:%M:%S%p %Y',
111 '%b %d %H:%M:%S',
111 '%b %d %H:%M:%S',
112 '%b %d %I:%M:%S%p',
112 '%b %d %I:%M:%S%p',
113 '%b %d %H:%M',
113 '%b %d %H:%M',
114 '%b %d %I:%M%p',
114 '%b %d %I:%M%p',
115 '%b %d %Y',
115 '%b %d %Y',
116 '%b %d',
116 '%b %d',
117 '%H:%M:%S',
117 '%H:%M:%S',
118 '%I:%M:%SP',
118 '%I:%M:%SP',
119 '%H:%M',
119 '%H:%M',
120 '%I:%M%p',
120 '%I:%M%p',
121 )
121 )
122
122
123 extendeddateformats = defaultdateformats + (
123 extendeddateformats = defaultdateformats + (
124 "%Y",
124 "%Y",
125 "%Y-%m",
125 "%Y-%m",
126 "%b",
126 "%b",
127 "%b %Y",
127 "%b %Y",
128 )
128 )
129
129
130 class SignalInterrupt(Exception):
130 class SignalInterrupt(Exception):
131 """Exception raised on SIGTERM and SIGHUP."""
131 """Exception raised on SIGTERM and SIGHUP."""
132
132
133 # differences from SafeConfigParser:
133 # differences from SafeConfigParser:
134 # - case-sensitive keys
134 # - case-sensitive keys
135 # - allows values that are not strings (this means that you may not
135 # - allows values that are not strings (this means that you may not
136 # be able to save the configuration to a file)
136 # be able to save the configuration to a file)
137 class configparser(ConfigParser.SafeConfigParser):
137 class configparser(ConfigParser.SafeConfigParser):
138 def optionxform(self, optionstr):
138 def optionxform(self, optionstr):
139 return optionstr
139 return optionstr
140
140
141 def set(self, section, option, value):
141 def set(self, section, option, value):
142 return ConfigParser.ConfigParser.set(self, section, option, value)
142 return ConfigParser.ConfigParser.set(self, section, option, value)
143
143
144 def _interpolate(self, section, option, rawval, vars):
144 def _interpolate(self, section, option, rawval, vars):
145 if not isinstance(rawval, basestring):
145 if not isinstance(rawval, basestring):
146 return rawval
146 return rawval
147 return ConfigParser.SafeConfigParser._interpolate(self, section,
147 return ConfigParser.SafeConfigParser._interpolate(self, section,
148 option, rawval, vars)
148 option, rawval, vars)
149
149
150 def cachefunc(func):
150 def cachefunc(func):
151 '''cache the result of function calls'''
151 '''cache the result of function calls'''
152 # XXX doesn't handle keywords args
152 # XXX doesn't handle keywords args
153 cache = {}
153 cache = {}
154 if func.func_code.co_argcount == 1:
154 if func.func_code.co_argcount == 1:
155 # we gain a small amount of time because
155 # we gain a small amount of time because
156 # we don't need to pack/unpack the list
156 # we don't need to pack/unpack the list
157 def f(arg):
157 def f(arg):
158 if arg not in cache:
158 if arg not in cache:
159 cache[arg] = func(arg)
159 cache[arg] = func(arg)
160 return cache[arg]
160 return cache[arg]
161 else:
161 else:
162 def f(*args):
162 def f(*args):
163 if args not in cache:
163 if args not in cache:
164 cache[args] = func(*args)
164 cache[args] = func(*args)
165 return cache[args]
165 return cache[args]
166
166
167 return f
167 return f
168
168
169 def pipefilter(s, cmd):
169 def pipefilter(s, cmd):
170 '''filter string S through command CMD, returning its output'''
170 '''filter string S through command CMD, returning its output'''
171 (pin, pout) = os.popen2(cmd, 'b')
171 (pin, pout) = os.popen2(cmd, 'b')
172 def writer():
172 def writer():
173 try:
173 try:
174 pin.write(s)
174 pin.write(s)
175 pin.close()
175 pin.close()
176 except IOError, inst:
176 except IOError, inst:
177 if inst.errno != errno.EPIPE:
177 if inst.errno != errno.EPIPE:
178 raise
178 raise
179
179
180 # we should use select instead on UNIX, but this will work on most
180 # we should use select instead on UNIX, but this will work on most
181 # systems, including Windows
181 # systems, including Windows
182 w = threading.Thread(target=writer)
182 w = threading.Thread(target=writer)
183 w.start()
183 w.start()
184 f = pout.read()
184 f = pout.read()
185 pout.close()
185 pout.close()
186 w.join()
186 w.join()
187 return f
187 return f
188
188
189 def tempfilter(s, cmd):
189 def tempfilter(s, cmd):
190 '''filter string S through a pair of temporary files with CMD.
190 '''filter string S through a pair of temporary files with CMD.
191 CMD is used as a template to create the real command to be run,
191 CMD is used as a template to create the real command to be run,
192 with the strings INFILE and OUTFILE replaced by the real names of
192 with the strings INFILE and OUTFILE replaced by the real names of
193 the temporary files generated.'''
193 the temporary files generated.'''
194 inname, outname = None, None
194 inname, outname = None, None
195 try:
195 try:
196 infd, inname = tempfile.mkstemp(prefix='hg-filter-in-')
196 infd, inname = tempfile.mkstemp(prefix='hg-filter-in-')
197 fp = os.fdopen(infd, 'wb')
197 fp = os.fdopen(infd, 'wb')
198 fp.write(s)
198 fp.write(s)
199 fp.close()
199 fp.close()
200 outfd, outname = tempfile.mkstemp(prefix='hg-filter-out-')
200 outfd, outname = tempfile.mkstemp(prefix='hg-filter-out-')
201 os.close(outfd)
201 os.close(outfd)
202 cmd = cmd.replace('INFILE', inname)
202 cmd = cmd.replace('INFILE', inname)
203 cmd = cmd.replace('OUTFILE', outname)
203 cmd = cmd.replace('OUTFILE', outname)
204 code = os.system(cmd)
204 code = os.system(cmd)
205 if sys.platform == 'OpenVMS' and code & 1:
205 if sys.platform == 'OpenVMS' and code & 1:
206 code = 0
206 code = 0
207 if code: raise Abort(_("command '%s' failed: %s") %
207 if code: raise Abort(_("command '%s' failed: %s") %
208 (cmd, explain_exit(code)))
208 (cmd, explain_exit(code)))
209 return open(outname, 'rb').read()
209 return open(outname, 'rb').read()
210 finally:
210 finally:
211 try:
211 try:
212 if inname: os.unlink(inname)
212 if inname: os.unlink(inname)
213 except: pass
213 except: pass
214 try:
214 try:
215 if outname: os.unlink(outname)
215 if outname: os.unlink(outname)
216 except: pass
216 except: pass
217
217
218 filtertable = {
218 filtertable = {
219 'tempfile:': tempfilter,
219 'tempfile:': tempfilter,
220 'pipe:': pipefilter,
220 'pipe:': pipefilter,
221 }
221 }
222
222
223 def filter(s, cmd):
223 def filter(s, cmd):
224 "filter a string through a command that transforms its input to its output"
224 "filter a string through a command that transforms its input to its output"
225 for name, fn in filtertable.iteritems():
225 for name, fn in filtertable.iteritems():
226 if cmd.startswith(name):
226 if cmd.startswith(name):
227 return fn(s, cmd[len(name):].lstrip())
227 return fn(s, cmd[len(name):].lstrip())
228 return pipefilter(s, cmd)
228 return pipefilter(s, cmd)
229
229
230 def binary(s):
230 def binary(s):
231 """return true if a string is binary data using diff's heuristic"""
231 """return true if a string is binary data using diff's heuristic"""
232 if s and '\0' in s[:4096]:
232 if s and '\0' in s[:4096]:
233 return True
233 return True
234 return False
234 return False
235
235
236 def unique(g):
236 def unique(g):
237 """return the uniq elements of iterable g"""
237 """return the uniq elements of iterable g"""
238 seen = {}
238 seen = {}
239 l = []
239 l = []
240 for f in g:
240 for f in g:
241 if f not in seen:
241 if f not in seen:
242 seen[f] = 1
242 seen[f] = 1
243 l.append(f)
243 l.append(f)
244 return l
244 return l
245
245
246 class Abort(Exception):
246 class Abort(Exception):
247 """Raised if a command needs to print an error and exit."""
247 """Raised if a command needs to print an error and exit."""
248
248
249 class UnexpectedOutput(Abort):
249 class UnexpectedOutput(Abort):
250 """Raised to print an error with part of output and exit."""
250 """Raised to print an error with part of output and exit."""
251
251
252 def always(fn): return True
252 def always(fn): return True
253 def never(fn): return False
253 def never(fn): return False
254
254
255 def expand_glob(pats):
255 def expand_glob(pats):
256 '''On Windows, expand the implicit globs in a list of patterns'''
256 '''On Windows, expand the implicit globs in a list of patterns'''
257 if os.name != 'nt':
257 if os.name != 'nt':
258 return list(pats)
258 return list(pats)
259 ret = []
259 ret = []
260 for p in pats:
260 for p in pats:
261 kind, name = patkind(p, None)
261 kind, name = patkind(p, None)
262 if kind is None:
262 if kind is None:
263 globbed = glob.glob(name)
263 globbed = glob.glob(name)
264 if globbed:
264 if globbed:
265 ret.extend(globbed)
265 ret.extend(globbed)
266 continue
266 continue
267 # if we couldn't expand the glob, just keep it around
267 # if we couldn't expand the glob, just keep it around
268 ret.append(p)
268 ret.append(p)
269 return ret
269 return ret
270
270
271 def patkind(name, dflt_pat='glob'):
271 def patkind(name, dflt_pat='glob'):
272 """Split a string into an optional pattern kind prefix and the
272 """Split a string into an optional pattern kind prefix and the
273 actual pattern."""
273 actual pattern."""
274 for prefix in 're', 'glob', 'path', 'relglob', 'relpath', 'relre':
274 for prefix in 're', 'glob', 'path', 'relglob', 'relpath', 'relre':
275 if name.startswith(prefix + ':'): return name.split(':', 1)
275 if name.startswith(prefix + ':'): return name.split(':', 1)
276 return dflt_pat, name
276 return dflt_pat, name
277
277
278 def globre(pat, head='^', tail='$'):
278 def globre(pat, head='^', tail='$'):
279 "convert a glob pattern into a regexp"
279 "convert a glob pattern into a regexp"
280 i, n = 0, len(pat)
280 i, n = 0, len(pat)
281 res = ''
281 res = ''
282 group = False
282 group = False
283 def peek(): return i < n and pat[i]
283 def peek(): return i < n and pat[i]
284 while i < n:
284 while i < n:
285 c = pat[i]
285 c = pat[i]
286 i = i+1
286 i = i+1
287 if c == '*':
287 if c == '*':
288 if peek() == '*':
288 if peek() == '*':
289 i += 1
289 i += 1
290 res += '.*'
290 res += '.*'
291 else:
291 else:
292 res += '[^/]*'
292 res += '[^/]*'
293 elif c == '?':
293 elif c == '?':
294 res += '.'
294 res += '.'
295 elif c == '[':
295 elif c == '[':
296 j = i
296 j = i
297 if j < n and pat[j] in '!]':
297 if j < n and pat[j] in '!]':
298 j += 1
298 j += 1
299 while j < n and pat[j] != ']':
299 while j < n and pat[j] != ']':
300 j += 1
300 j += 1
301 if j >= n:
301 if j >= n:
302 res += '\\['
302 res += '\\['
303 else:
303 else:
304 stuff = pat[i:j].replace('\\','\\\\')
304 stuff = pat[i:j].replace('\\','\\\\')
305 i = j + 1
305 i = j + 1
306 if stuff[0] == '!':
306 if stuff[0] == '!':
307 stuff = '^' + stuff[1:]
307 stuff = '^' + stuff[1:]
308 elif stuff[0] == '^':
308 elif stuff[0] == '^':
309 stuff = '\\' + stuff
309 stuff = '\\' + stuff
310 res = '%s[%s]' % (res, stuff)
310 res = '%s[%s]' % (res, stuff)
311 elif c == '{':
311 elif c == '{':
312 group = True
312 group = True
313 res += '(?:'
313 res += '(?:'
314 elif c == '}' and group:
314 elif c == '}' and group:
315 res += ')'
315 res += ')'
316 group = False
316 group = False
317 elif c == ',' and group:
317 elif c == ',' and group:
318 res += '|'
318 res += '|'
319 elif c == '\\':
319 elif c == '\\':
320 p = peek()
320 p = peek()
321 if p:
321 if p:
322 i += 1
322 i += 1
323 res += re.escape(p)
323 res += re.escape(p)
324 else:
324 else:
325 res += re.escape(c)
325 res += re.escape(c)
326 else:
326 else:
327 res += re.escape(c)
327 res += re.escape(c)
328 return head + res + tail
328 return head + res + tail
329
329
330 _globchars = {'[': 1, '{': 1, '*': 1, '?': 1}
330 _globchars = {'[': 1, '{': 1, '*': 1, '?': 1}
331
331
332 def pathto(root, n1, n2):
332 def pathto(root, n1, n2):
333 '''return the relative path from one place to another.
333 '''return the relative path from one place to another.
334 root should use os.sep to separate directories
334 root should use os.sep to separate directories
335 n1 should use os.sep to separate directories
335 n1 should use os.sep to separate directories
336 n2 should use "/" to separate directories
336 n2 should use "/" to separate directories
337 returns an os.sep-separated path.
337 returns an os.sep-separated path.
338
338
339 If n1 is a relative path, it's assumed it's
339 If n1 is a relative path, it's assumed it's
340 relative to root.
340 relative to root.
341 n2 should always be relative to root.
341 n2 should always be relative to root.
342 '''
342 '''
343 if not n1: return localpath(n2)
343 if not n1: return localpath(n2)
344 if os.path.isabs(n1):
344 if os.path.isabs(n1):
345 if os.path.splitdrive(root)[0] != os.path.splitdrive(n1)[0]:
345 if os.path.splitdrive(root)[0] != os.path.splitdrive(n1)[0]:
346 return os.path.join(root, localpath(n2))
346 return os.path.join(root, localpath(n2))
347 n2 = '/'.join((pconvert(root), n2))
347 n2 = '/'.join((pconvert(root), n2))
348 a, b = n1.split(os.sep), n2.split('/')
348 a, b = n1.split(os.sep), n2.split('/')
349 a.reverse()
349 a.reverse()
350 b.reverse()
350 b.reverse()
351 while a and b and a[-1] == b[-1]:
351 while a and b and a[-1] == b[-1]:
352 a.pop()
352 a.pop()
353 b.pop()
353 b.pop()
354 b.reverse()
354 b.reverse()
355 return os.sep.join((['..'] * len(a)) + b)
355 return os.sep.join((['..'] * len(a)) + b)
356
356
357 def canonpath(root, cwd, myname):
357 def canonpath(root, cwd, myname):
358 """return the canonical path of myname, given cwd and root"""
358 """return the canonical path of myname, given cwd and root"""
359 if root == os.sep:
359 if root == os.sep:
360 rootsep = os.sep
360 rootsep = os.sep
361 elif root.endswith(os.sep):
361 elif root.endswith(os.sep):
362 rootsep = root
362 rootsep = root
363 else:
363 else:
364 rootsep = root + os.sep
364 rootsep = root + os.sep
365 name = myname
365 name = myname
366 if not os.path.isabs(name):
366 if not os.path.isabs(name):
367 name = os.path.join(root, cwd, name)
367 name = os.path.join(root, cwd, name)
368 name = os.path.normpath(name)
368 name = os.path.normpath(name)
369 audit_path = path_auditor(root)
369 audit_path = path_auditor(root)
370 if name != rootsep and name.startswith(rootsep):
370 if name != rootsep and name.startswith(rootsep):
371 name = name[len(rootsep):]
371 name = name[len(rootsep):]
372 audit_path(name)
372 audit_path(name)
373 return pconvert(name)
373 return pconvert(name)
374 elif name == root:
374 elif name == root:
375 return ''
375 return ''
376 else:
376 else:
377 # Determine whether `name' is in the hierarchy at or beneath `root',
377 # Determine whether `name' is in the hierarchy at or beneath `root',
378 # by iterating name=dirname(name) until that causes no change (can't
378 # by iterating name=dirname(name) until that causes no change (can't
379 # check name == '/', because that doesn't work on windows). For each
379 # check name == '/', because that doesn't work on windows). For each
380 # `name', compare dev/inode numbers. If they match, the list `rel'
380 # `name', compare dev/inode numbers. If they match, the list `rel'
381 # holds the reversed list of components making up the relative file
381 # holds the reversed list of components making up the relative file
382 # name we want.
382 # name we want.
383 root_st = os.stat(root)
383 root_st = os.stat(root)
384 rel = []
384 rel = []
385 while True:
385 while True:
386 try:
386 try:
387 name_st = os.stat(name)
387 name_st = os.stat(name)
388 except OSError:
388 except OSError:
389 break
389 break
390 if samestat(name_st, root_st):
390 if samestat(name_st, root_st):
391 if not rel:
391 if not rel:
392 # name was actually the same as root (maybe a symlink)
392 # name was actually the same as root (maybe a symlink)
393 return ''
393 return ''
394 rel.reverse()
394 rel.reverse()
395 name = os.path.join(*rel)
395 name = os.path.join(*rel)
396 audit_path(name)
396 audit_path(name)
397 return pconvert(name)
397 return pconvert(name)
398 dirname, basename = os.path.split(name)
398 dirname, basename = os.path.split(name)
399 rel.append(basename)
399 rel.append(basename)
400 if dirname == name:
400 if dirname == name:
401 break
401 break
402 name = dirname
402 name = dirname
403
403
404 raise Abort('%s not under root' % myname)
404 raise Abort('%s not under root' % myname)
405
405
406 def matcher(canonroot, cwd='', names=[], inc=[], exc=[], src=None):
406 def matcher(canonroot, cwd='', names=[], inc=[], exc=[], src=None):
407 return _matcher(canonroot, cwd, names, inc, exc, 'glob', src)
407 return _matcher(canonroot, cwd, names, inc, exc, 'glob', src)
408
408
409 def cmdmatcher(canonroot, cwd='', names=[], inc=[], exc=[], src=None,
409 def cmdmatcher(canonroot, cwd='', names=[], inc=[], exc=[], src=None,
410 globbed=False, default=None):
410 globbed=False, default=None):
411 default = default or 'relpath'
411 default = default or 'relpath'
412 if default == 'relpath' and not globbed:
412 if default == 'relpath' and not globbed:
413 names = expand_glob(names)
413 names = expand_glob(names)
414 return _matcher(canonroot, cwd, names, inc, exc, default, src)
414 return _matcher(canonroot, cwd, names, inc, exc, default, src)
415
415
416 def _matcher(canonroot, cwd, names, inc, exc, dflt_pat, src):
416 def _matcher(canonroot, cwd, names, inc, exc, dflt_pat, src):
417 """build a function to match a set of file patterns
417 """build a function to match a set of file patterns
418
418
419 arguments:
419 arguments:
420 canonroot - the canonical root of the tree you're matching against
420 canonroot - the canonical root of the tree you're matching against
421 cwd - the current working directory, if relevant
421 cwd - the current working directory, if relevant
422 names - patterns to find
422 names - patterns to find
423 inc - patterns to include
423 inc - patterns to include
424 exc - patterns to exclude
424 exc - patterns to exclude
425 dflt_pat - if a pattern in names has no explicit type, assume this one
425 dflt_pat - if a pattern in names has no explicit type, assume this one
426 src - where these patterns came from (e.g. .hgignore)
426 src - where these patterns came from (e.g. .hgignore)
427
427
428 a pattern is one of:
428 a pattern is one of:
429 'glob:<glob>' - a glob relative to cwd
429 'glob:<glob>' - a glob relative to cwd
430 're:<regexp>' - a regular expression
430 're:<regexp>' - a regular expression
431 'path:<path>' - a path relative to canonroot
431 'path:<path>' - a path relative to canonroot
432 'relglob:<glob>' - an unrooted glob (*.c matches C files in all dirs)
432 'relglob:<glob>' - an unrooted glob (*.c matches C files in all dirs)
433 'relpath:<path>' - a path relative to cwd
433 'relpath:<path>' - a path relative to cwd
434 'relre:<regexp>' - a regexp that doesn't have to match the start of a name
434 'relre:<regexp>' - a regexp that doesn't have to match the start of a name
435 '<something>' - one of the cases above, selected by the dflt_pat argument
435 '<something>' - one of the cases above, selected by the dflt_pat argument
436
436
437 returns:
437 returns:
438 a 3-tuple containing
438 a 3-tuple containing
439 - list of roots (places where one should start a recursive walk of the fs);
439 - list of roots (places where one should start a recursive walk of the fs);
440 this often matches the explicit non-pattern names passed in, but also
440 this often matches the explicit non-pattern names passed in, but also
441 includes the initial part of glob: patterns that has no glob characters
441 includes the initial part of glob: patterns that has no glob characters
442 - a bool match(filename) function
442 - a bool match(filename) function
443 - a bool indicating if any patterns were passed in
443 - a bool indicating if any patterns were passed in
444 """
444 """
445
445
446 # a common case: no patterns at all
446 # a common case: no patterns at all
447 if not names and not inc and not exc:
447 if not names and not inc and not exc:
448 return [], always, False
448 return [], always, False
449
449
450 def contains_glob(name):
450 def contains_glob(name):
451 for c in name:
451 for c in name:
452 if c in _globchars: return True
452 if c in _globchars: return True
453 return False
453 return False
454
454
455 def regex(kind, name, tail):
455 def regex(kind, name, tail):
456 '''convert a pattern into a regular expression'''
456 '''convert a pattern into a regular expression'''
457 if not name:
457 if not name:
458 return ''
458 return ''
459 if kind == 're':
459 if kind == 're':
460 return name
460 return name
461 elif kind == 'path':
461 elif kind == 'path':
462 return '^' + re.escape(name) + '(?:/|$)'
462 return '^' + re.escape(name) + '(?:/|$)'
463 elif kind == 'relglob':
463 elif kind == 'relglob':
464 return globre(name, '(?:|.*/)', tail)
464 return globre(name, '(?:|.*/)', tail)
465 elif kind == 'relpath':
465 elif kind == 'relpath':
466 return re.escape(name) + '(?:/|$)'
466 return re.escape(name) + '(?:/|$)'
467 elif kind == 'relre':
467 elif kind == 'relre':
468 if name.startswith('^'):
468 if name.startswith('^'):
469 return name
469 return name
470 return '.*' + name
470 return '.*' + name
471 return globre(name, '', tail)
471 return globre(name, '', tail)
472
472
473 def matchfn(pats, tail):
473 def matchfn(pats, tail):
474 """build a matching function from a set of patterns"""
474 """build a matching function from a set of patterns"""
475 if not pats:
475 if not pats:
476 return
476 return
477 try:
477 try:
478 pat = '(?:%s)' % '|'.join([regex(k, p, tail) for (k, p) in pats])
478 pat = '(?:%s)' % '|'.join([regex(k, p, tail) for (k, p) in pats])
479 return re.compile(pat).match
479 return re.compile(pat).match
480 except OverflowError:
480 except OverflowError:
481 # We're using a Python with a tiny regex engine and we
481 # We're using a Python with a tiny regex engine and we
482 # made it explode, so we'll divide the pattern list in two
482 # made it explode, so we'll divide the pattern list in two
483 # until it works
483 # until it works
484 l = len(pats)
484 l = len(pats)
485 if l < 2:
485 if l < 2:
486 raise
486 raise
487 a, b = matchfn(pats[:l/2], tail), matchfn(pats[l/2:], tail)
487 a, b = matchfn(pats[:l/2], tail), matchfn(pats[l/2:], tail)
488 return lambda s: a(s) or b(s)
488 return lambda s: a(s) or b(s)
489 except re.error:
489 except re.error:
490 for k, p in pats:
490 for k, p in pats:
491 try:
491 try:
492 re.compile('(?:%s)' % regex(k, p, tail))
492 re.compile('(?:%s)' % regex(k, p, tail))
493 except re.error:
493 except re.error:
494 if src:
494 if src:
495 raise Abort("%s: invalid pattern (%s): %s" %
495 raise Abort("%s: invalid pattern (%s): %s" %
496 (src, k, p))
496 (src, k, p))
497 else:
497 else:
498 raise Abort("invalid pattern (%s): %s" % (k, p))
498 raise Abort("invalid pattern (%s): %s" % (k, p))
499 raise Abort("invalid pattern")
499 raise Abort("invalid pattern")
500
500
501 def globprefix(pat):
501 def globprefix(pat):
502 '''return the non-glob prefix of a path, e.g. foo/* -> foo'''
502 '''return the non-glob prefix of a path, e.g. foo/* -> foo'''
503 root = []
503 root = []
504 for p in pat.split('/'):
504 for p in pat.split('/'):
505 if contains_glob(p): break
505 if contains_glob(p): break
506 root.append(p)
506 root.append(p)
507 return '/'.join(root) or '.'
507 return '/'.join(root) or '.'
508
508
509 def normalizepats(names, default):
509 def normalizepats(names, default):
510 pats = []
510 pats = []
511 roots = []
511 roots = []
512 anypats = False
512 anypats = False
513 for kind, name in [patkind(p, default) for p in names]:
513 for kind, name in [patkind(p, default) for p in names]:
514 if kind in ('glob', 'relpath'):
514 if kind in ('glob', 'relpath'):
515 name = canonpath(canonroot, cwd, name)
515 name = canonpath(canonroot, cwd, name)
516 elif kind in ('relglob', 'path'):
516 elif kind in ('relglob', 'path'):
517 name = normpath(name)
517 name = normpath(name)
518
518
519 pats.append((kind, name))
519 pats.append((kind, name))
520
520
521 if kind in ('glob', 're', 'relglob', 'relre'):
521 if kind in ('glob', 're', 'relglob', 'relre'):
522 anypats = True
522 anypats = True
523
523
524 if kind == 'glob':
524 if kind == 'glob':
525 root = globprefix(name)
525 root = globprefix(name)
526 roots.append(root)
526 roots.append(root)
527 elif kind in ('relpath', 'path'):
527 elif kind in ('relpath', 'path'):
528 roots.append(name or '.')
528 roots.append(name or '.')
529 elif kind == 'relglob':
529 elif kind == 'relglob':
530 roots.append('.')
530 roots.append('.')
531 return roots, pats, anypats
531 return roots, pats, anypats
532
532
533 roots, pats, anypats = normalizepats(names, dflt_pat)
533 roots, pats, anypats = normalizepats(names, dflt_pat)
534
534
535 patmatch = matchfn(pats, '$') or always
535 patmatch = matchfn(pats, '$') or always
536 incmatch = always
536 incmatch = always
537 if inc:
537 if inc:
538 dummy, inckinds, dummy = normalizepats(inc, 'glob')
538 dummy, inckinds, dummy = normalizepats(inc, 'glob')
539 incmatch = matchfn(inckinds, '(?:/|$)')
539 incmatch = matchfn(inckinds, '(?:/|$)')
540 excmatch = lambda fn: False
540 excmatch = lambda fn: False
541 if exc:
541 if exc:
542 dummy, exckinds, dummy = normalizepats(exc, 'glob')
542 dummy, exckinds, dummy = normalizepats(exc, 'glob')
543 excmatch = matchfn(exckinds, '(?:/|$)')
543 excmatch = matchfn(exckinds, '(?:/|$)')
544
544
545 if not names and inc and not exc:
545 if not names and inc and not exc:
546 # common case: hgignore patterns
546 # common case: hgignore patterns
547 match = incmatch
547 match = incmatch
548 else:
548 else:
549 match = lambda fn: incmatch(fn) and not excmatch(fn) and patmatch(fn)
549 match = lambda fn: incmatch(fn) and not excmatch(fn) and patmatch(fn)
550
550
551 return (roots, match, (inc or exc or anypats) and True)
551 return (roots, match, (inc or exc or anypats) and True)
552
552
553 _hgexecutable = None
553 _hgexecutable = None
554
554
555 def hgexecutable():
555 def hgexecutable():
556 """return location of the 'hg' executable.
556 """return location of the 'hg' executable.
557
557
558 Defaults to $HG or 'hg' in the search path.
558 Defaults to $HG or 'hg' in the search path.
559 """
559 """
560 if _hgexecutable is None:
560 if _hgexecutable is None:
561 set_hgexecutable(os.environ.get('HG') or find_exe('hg', 'hg'))
561 set_hgexecutable(os.environ.get('HG') or find_exe('hg', 'hg'))
562 return _hgexecutable
562 return _hgexecutable
563
563
564 def set_hgexecutable(path):
564 def set_hgexecutable(path):
565 """set location of the 'hg' executable"""
565 """set location of the 'hg' executable"""
566 global _hgexecutable
566 global _hgexecutable
567 _hgexecutable = path
567 _hgexecutable = path
568
568
569 def system(cmd, environ={}, cwd=None, onerr=None, errprefix=None):
569 def system(cmd, environ={}, cwd=None, onerr=None, errprefix=None):
570 '''enhanced shell command execution.
570 '''enhanced shell command execution.
571 run with environment maybe modified, maybe in different dir.
571 run with environment maybe modified, maybe in different dir.
572
572
573 if command fails and onerr is None, return status. if ui object,
573 if command fails and onerr is None, return status. if ui object,
574 print error message and return status, else raise onerr object as
574 print error message and return status, else raise onerr object as
575 exception.'''
575 exception.'''
576 def py2shell(val):
576 def py2shell(val):
577 'convert python object into string that is useful to shell'
577 'convert python object into string that is useful to shell'
578 if val in (None, False):
578 if val in (None, False):
579 return '0'
579 return '0'
580 if val == True:
580 if val == True:
581 return '1'
581 return '1'
582 return str(val)
582 return str(val)
583 oldenv = {}
583 oldenv = {}
584 for k in environ:
584 for k in environ:
585 oldenv[k] = os.environ.get(k)
585 oldenv[k] = os.environ.get(k)
586 if cwd is not None:
586 if cwd is not None:
587 oldcwd = os.getcwd()
587 oldcwd = os.getcwd()
588 origcmd = cmd
588 origcmd = cmd
589 if os.name == 'nt':
589 if os.name == 'nt':
590 cmd = '"%s"' % cmd
590 cmd = '"%s"' % cmd
591 try:
591 try:
592 for k, v in environ.iteritems():
592 for k, v in environ.iteritems():
593 os.environ[k] = py2shell(v)
593 os.environ[k] = py2shell(v)
594 os.environ['HG'] = hgexecutable()
594 os.environ['HG'] = hgexecutable()
595 if cwd is not None and oldcwd != cwd:
595 if cwd is not None and oldcwd != cwd:
596 os.chdir(cwd)
596 os.chdir(cwd)
597 rc = os.system(cmd)
597 rc = os.system(cmd)
598 if sys.platform == 'OpenVMS' and rc & 1:
598 if sys.platform == 'OpenVMS' and rc & 1:
599 rc = 0
599 rc = 0
600 if rc and onerr:
600 if rc and onerr:
601 errmsg = '%s %s' % (os.path.basename(origcmd.split(None, 1)[0]),
601 errmsg = '%s %s' % (os.path.basename(origcmd.split(None, 1)[0]),
602 explain_exit(rc)[0])
602 explain_exit(rc)[0])
603 if errprefix:
603 if errprefix:
604 errmsg = '%s: %s' % (errprefix, errmsg)
604 errmsg = '%s: %s' % (errprefix, errmsg)
605 try:
605 try:
606 onerr.warn(errmsg + '\n')
606 onerr.warn(errmsg + '\n')
607 except AttributeError:
607 except AttributeError:
608 raise onerr(errmsg)
608 raise onerr(errmsg)
609 return rc
609 return rc
610 finally:
610 finally:
611 for k, v in oldenv.iteritems():
611 for k, v in oldenv.iteritems():
612 if v is None:
612 if v is None:
613 del os.environ[k]
613 del os.environ[k]
614 else:
614 else:
615 os.environ[k] = v
615 os.environ[k] = v
616 if cwd is not None and oldcwd != cwd:
616 if cwd is not None and oldcwd != cwd:
617 os.chdir(oldcwd)
617 os.chdir(oldcwd)
618
618
619 # os.path.lexists is not available on python2.3
619 # os.path.lexists is not available on python2.3
620 def lexists(filename):
620 def lexists(filename):
621 "test whether a file with this name exists. does not follow symlinks"
621 "test whether a file with this name exists. does not follow symlinks"
622 try:
622 try:
623 os.lstat(filename)
623 os.lstat(filename)
624 except:
624 except:
625 return False
625 return False
626 return True
626 return True
627
627
628 def rename(src, dst):
628 def rename(src, dst):
629 """forcibly rename a file"""
629 """forcibly rename a file"""
630 try:
630 try:
631 os.rename(src, dst)
631 os.rename(src, dst)
632 except OSError, err: # FIXME: check err (EEXIST ?)
632 except OSError, err: # FIXME: check err (EEXIST ?)
633 # on windows, rename to existing file is not allowed, so we
633 # on windows, rename to existing file is not allowed, so we
634 # must delete destination first. but if file is open, unlink
634 # must delete destination first. but if file is open, unlink
635 # schedules it for delete but does not delete it. rename
635 # schedules it for delete but does not delete it. rename
636 # happens immediately even for open files, so we create
636 # happens immediately even for open files, so we create
637 # temporary file, delete it, rename destination to that name,
637 # temporary file, delete it, rename destination to that name,
638 # then delete that. then rename is safe to do.
638 # then delete that. then rename is safe to do.
639 fd, temp = tempfile.mkstemp(dir=os.path.dirname(dst) or '.')
639 fd, temp = tempfile.mkstemp(dir=os.path.dirname(dst) or '.')
640 os.close(fd)
640 os.close(fd)
641 os.unlink(temp)
641 os.unlink(temp)
642 os.rename(dst, temp)
642 os.rename(dst, temp)
643 os.unlink(temp)
643 os.unlink(temp)
644 os.rename(src, dst)
644 os.rename(src, dst)
645
645
646 def unlink(f):
646 def unlink(f):
647 """unlink and remove the directory if it is empty"""
647 """unlink and remove the directory if it is empty"""
648 os.unlink(f)
648 os.unlink(f)
649 # try removing directories that might now be empty
649 # try removing directories that might now be empty
650 try:
650 try:
651 os.removedirs(os.path.dirname(f))
651 os.removedirs(os.path.dirname(f))
652 except OSError:
652 except OSError:
653 pass
653 pass
654
654
655 def copyfile(src, dest):
655 def copyfile(src, dest):
656 "copy a file, preserving mode"
656 "copy a file, preserving mode"
657 if os.path.islink(src):
657 if os.path.islink(src):
658 try:
658 try:
659 os.unlink(dest)
659 os.unlink(dest)
660 except:
660 except:
661 pass
661 pass
662 os.symlink(os.readlink(src), dest)
662 os.symlink(os.readlink(src), dest)
663 else:
663 else:
664 try:
664 try:
665 shutil.copyfile(src, dest)
665 shutil.copyfile(src, dest)
666 shutil.copymode(src, dest)
666 shutil.copymode(src, dest)
667 except shutil.Error, inst:
667 except shutil.Error, inst:
668 raise Abort(str(inst))
668 raise Abort(str(inst))
669
669
670 def copyfiles(src, dst, hardlink=None):
670 def copyfiles(src, dst, hardlink=None):
671 """Copy a directory tree using hardlinks if possible"""
671 """Copy a directory tree using hardlinks if possible"""
672
672
673 if hardlink is None:
673 if hardlink is None:
674 hardlink = (os.stat(src).st_dev ==
674 hardlink = (os.stat(src).st_dev ==
675 os.stat(os.path.dirname(dst)).st_dev)
675 os.stat(os.path.dirname(dst)).st_dev)
676
676
677 if os.path.isdir(src):
677 if os.path.isdir(src):
678 os.mkdir(dst)
678 os.mkdir(dst)
679 for name, kind in osutil.listdir(src):
679 for name, kind in osutil.listdir(src):
680 srcname = os.path.join(src, name)
680 srcname = os.path.join(src, name)
681 dstname = os.path.join(dst, name)
681 dstname = os.path.join(dst, name)
682 copyfiles(srcname, dstname, hardlink)
682 copyfiles(srcname, dstname, hardlink)
683 else:
683 else:
684 if hardlink:
684 if hardlink:
685 try:
685 try:
686 os_link(src, dst)
686 os_link(src, dst)
687 except (IOError, OSError):
687 except (IOError, OSError):
688 hardlink = False
688 hardlink = False
689 shutil.copy(src, dst)
689 shutil.copy(src, dst)
690 else:
690 else:
691 shutil.copy(src, dst)
691 shutil.copy(src, dst)
692
692
693 class path_auditor(object):
693 class path_auditor(object):
694 '''ensure that a filesystem path contains no banned components.
694 '''ensure that a filesystem path contains no banned components.
695 the following properties of a path are checked:
695 the following properties of a path are checked:
696
696
697 - under top-level .hg
697 - under top-level .hg
698 - starts at the root of a windows drive
698 - starts at the root of a windows drive
699 - contains ".."
699 - contains ".."
700 - traverses a symlink (e.g. a/symlink_here/b)
700 - traverses a symlink (e.g. a/symlink_here/b)
701 - inside a nested repository'''
701 - inside a nested repository'''
702
702
703 def __init__(self, root):
703 def __init__(self, root):
704 self.audited = set()
704 self.audited = set()
705 self.auditeddir = set()
705 self.auditeddir = set()
706 self.root = root
706 self.root = root
707
707
708 def __call__(self, path):
708 def __call__(self, path):
709 if path in self.audited:
709 if path in self.audited:
710 return
710 return
711 normpath = os.path.normcase(path)
711 normpath = os.path.normcase(path)
712 parts = normpath.split(os.sep)
712 parts = normpath.split(os.sep)
713 if (os.path.splitdrive(path)[0] or parts[0] in ('.hg', '')
713 if (os.path.splitdrive(path)[0] or parts[0] in ('.hg', '')
714 or os.pardir in parts):
714 or os.pardir in parts):
715 raise Abort(_("path contains illegal component: %s") % path)
715 raise Abort(_("path contains illegal component: %s") % path)
716 def check(prefix):
716 def check(prefix):
717 curpath = os.path.join(self.root, prefix)
717 curpath = os.path.join(self.root, prefix)
718 try:
718 try:
719 st = os.lstat(curpath)
719 st = os.lstat(curpath)
720 except OSError, err:
720 except OSError, err:
721 # EINVAL can be raised as invalid path syntax under win32.
721 # EINVAL can be raised as invalid path syntax under win32.
722 # They must be ignored for patterns can be checked too.
722 # They must be ignored for patterns can be checked too.
723 if err.errno not in (errno.ENOENT, errno.EINVAL):
723 if err.errno not in (errno.ENOENT, errno.EINVAL):
724 raise
724 raise
725 else:
725 else:
726 if stat.S_ISLNK(st.st_mode):
726 if stat.S_ISLNK(st.st_mode):
727 raise Abort(_('path %r traverses symbolic link %r') %
727 raise Abort(_('path %r traverses symbolic link %r') %
728 (path, prefix))
728 (path, prefix))
729 elif (stat.S_ISDIR(st.st_mode) and
729 elif (stat.S_ISDIR(st.st_mode) and
730 os.path.isdir(os.path.join(curpath, '.hg'))):
730 os.path.isdir(os.path.join(curpath, '.hg'))):
731 raise Abort(_('path %r is inside repo %r') %
731 raise Abort(_('path %r is inside repo %r') %
732 (path, prefix))
732 (path, prefix))
733
733
734 prefixes = []
734 prefixes = []
735 for c in strutil.rfindall(normpath, os.sep):
735 for c in strutil.rfindall(normpath, os.sep):
736 prefix = normpath[:c]
736 prefix = normpath[:c]
737 if prefix in self.auditeddir:
737 if prefix in self.auditeddir:
738 break
738 break
739 check(prefix)
739 check(prefix)
740 prefixes.append(prefix)
740 prefixes.append(prefix)
741
741
742 self.audited.add(path)
742 self.audited.add(path)
743 # only add prefixes to the cache after checking everything: we don't
743 # only add prefixes to the cache after checking everything: we don't
744 # want to add "foo/bar/baz" before checking if there's a "foo/.hg"
744 # want to add "foo/bar/baz" before checking if there's a "foo/.hg"
745 self.auditeddir.update(prefixes)
745 self.auditeddir.update(prefixes)
746
746
747 def _makelock_file(info, pathname):
747 def _makelock_file(info, pathname):
748 ld = os.open(pathname, os.O_CREAT | os.O_WRONLY | os.O_EXCL)
748 ld = os.open(pathname, os.O_CREAT | os.O_WRONLY | os.O_EXCL)
749 os.write(ld, info)
749 os.write(ld, info)
750 os.close(ld)
750 os.close(ld)
751
751
752 def _readlock_file(pathname):
752 def _readlock_file(pathname):
753 return posixfile(pathname).read()
753 return posixfile(pathname).read()
754
754
755 def nlinks(pathname):
755 def nlinks(pathname):
756 """Return number of hardlinks for the given file."""
756 """Return number of hardlinks for the given file."""
757 return os.lstat(pathname).st_nlink
757 return os.lstat(pathname).st_nlink
758
758
759 if hasattr(os, 'link'):
759 if hasattr(os, 'link'):
760 os_link = os.link
760 os_link = os.link
761 else:
761 else:
762 def os_link(src, dst):
762 def os_link(src, dst):
763 raise OSError(0, _("Hardlinks not supported"))
763 raise OSError(0, _("Hardlinks not supported"))
764
764
765 def fstat(fp):
765 def fstat(fp):
766 '''stat file object that may not have fileno method.'''
766 '''stat file object that may not have fileno method.'''
767 try:
767 try:
768 return os.fstat(fp.fileno())
768 return os.fstat(fp.fileno())
769 except AttributeError:
769 except AttributeError:
770 return os.stat(fp.name)
770 return os.stat(fp.name)
771
771
772 posixfile = file
772 posixfile = file
773
773
774 def is_win_9x():
774 def is_win_9x():
775 '''return true if run on windows 95, 98 or me.'''
775 '''return true if run on windows 95, 98 or me.'''
776 try:
776 try:
777 return sys.getwindowsversion()[3] == 1
777 return sys.getwindowsversion()[3] == 1
778 except AttributeError:
778 except AttributeError:
779 return os.name == 'nt' and 'command' in os.environ.get('comspec', '')
779 return os.name == 'nt' and 'command' in os.environ.get('comspec', '')
780
780
781 getuser_fallback = None
781 getuser_fallback = None
782
782
783 def getuser():
783 def getuser():
784 '''return name of current user'''
784 '''return name of current user'''
785 try:
785 try:
786 return getpass.getuser()
786 return getpass.getuser()
787 except ImportError:
787 except ImportError:
788 # import of pwd will fail on windows - try fallback
788 # import of pwd will fail on windows - try fallback
789 if getuser_fallback:
789 if getuser_fallback:
790 return getuser_fallback()
790 return getuser_fallback()
791 # raised if win32api not available
791 # raised if win32api not available
792 raise Abort(_('user name not available - set USERNAME '
792 raise Abort(_('user name not available - set USERNAME '
793 'environment variable'))
793 'environment variable'))
794
794
795 def username(uid=None):
795 def username(uid=None):
796 """Return the name of the user with the given uid.
796 """Return the name of the user with the given uid.
797
797
798 If uid is None, return the name of the current user."""
798 If uid is None, return the name of the current user."""
799 try:
799 try:
800 import pwd
800 import pwd
801 if uid is None:
801 if uid is None:
802 uid = os.getuid()
802 uid = os.getuid()
803 try:
803 try:
804 return pwd.getpwuid(uid)[0]
804 return pwd.getpwuid(uid)[0]
805 except KeyError:
805 except KeyError:
806 return str(uid)
806 return str(uid)
807 except ImportError:
807 except ImportError:
808 return None
808 return None
809
809
810 def groupname(gid=None):
810 def groupname(gid=None):
811 """Return the name of the group with the given gid.
811 """Return the name of the group with the given gid.
812
812
813 If gid is None, return the name of the current group."""
813 If gid is None, return the name of the current group."""
814 try:
814 try:
815 import grp
815 import grp
816 if gid is None:
816 if gid is None:
817 gid = os.getgid()
817 gid = os.getgid()
818 try:
818 try:
819 return grp.getgrgid(gid)[0]
819 return grp.getgrgid(gid)[0]
820 except KeyError:
820 except KeyError:
821 return str(gid)
821 return str(gid)
822 except ImportError:
822 except ImportError:
823 return None
823 return None
824
824
825 # File system features
825 # File system features
826
826
827 def checkfolding(path):
827 def checkfolding(path):
828 """
828 """
829 Check whether the given path is on a case-sensitive filesystem
829 Check whether the given path is on a case-sensitive filesystem
830
830
831 Requires a path (like /foo/.hg) ending with a foldable final
831 Requires a path (like /foo/.hg) ending with a foldable final
832 directory component.
832 directory component.
833 """
833 """
834 s1 = os.stat(path)
834 s1 = os.stat(path)
835 d, b = os.path.split(path)
835 d, b = os.path.split(path)
836 p2 = os.path.join(d, b.upper())
836 p2 = os.path.join(d, b.upper())
837 if path == p2:
837 if path == p2:
838 p2 = os.path.join(d, b.lower())
838 p2 = os.path.join(d, b.lower())
839 try:
839 try:
840 s2 = os.stat(p2)
840 s2 = os.stat(p2)
841 if s2 == s1:
841 if s2 == s1:
842 return False
842 return False
843 return True
843 return True
844 except:
844 except:
845 return True
845 return True
846
846
847 def checkexec(path):
847 def checkexec(path):
848 """
848 """
849 Check whether the given path is on a filesystem with UNIX-like exec flags
849 Check whether the given path is on a filesystem with UNIX-like exec flags
850
850
851 Requires a directory (like /foo/.hg)
851 Requires a directory (like /foo/.hg)
852 """
852 """
853 try:
853 try:
854 EXECFLAGS = stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH
854 EXECFLAGS = stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH
855 fh, fn = tempfile.mkstemp("", "", path)
855 fh, fn = tempfile.mkstemp("", "", path)
856 os.close(fh)
856 os.close(fh)
857 m = os.stat(fn).st_mode
857 m = os.stat(fn).st_mode
858 # VFAT on Linux can flip mode but it doesn't persist a FS remount.
858 # VFAT on Linux can flip mode but it doesn't persist a FS remount.
859 # frequently we can detect it if files are created with exec bit on.
859 # frequently we can detect it if files are created with exec bit on.
860 new_file_has_exec = m & EXECFLAGS
860 new_file_has_exec = m & EXECFLAGS
861 os.chmod(fn, m ^ EXECFLAGS)
861 os.chmod(fn, m ^ EXECFLAGS)
862 exec_flags_cannot_flip = (os.stat(fn).st_mode == m)
862 exec_flags_cannot_flip = (os.stat(fn).st_mode == m)
863 os.unlink(fn)
863 os.unlink(fn)
864 except (IOError,OSError):
864 except (IOError,OSError):
865 # we don't care, the user probably won't be able to commit anyway
865 # we don't care, the user probably won't be able to commit anyway
866 return False
866 return False
867 return not (new_file_has_exec or exec_flags_cannot_flip)
867 return not (new_file_has_exec or exec_flags_cannot_flip)
868
868
869 def execfunc(path, fallback):
869 def execfunc(path, fallback):
870 '''return an is_exec() function with default to fallback'''
870 '''return an is_exec() function with default to fallback'''
871 if checkexec(path):
871 if checkexec(path):
872 return lambda x: is_exec(os.path.join(path, x))
872 return lambda x: is_exec(os.path.join(path, x))
873 return fallback
873 return fallback
874
874
875 def checklink(path):
875 def checklink(path):
876 """check whether the given path is on a symlink-capable filesystem"""
876 """check whether the given path is on a symlink-capable filesystem"""
877 # mktemp is not racy because symlink creation will fail if the
877 # mktemp is not racy because symlink creation will fail if the
878 # file already exists
878 # file already exists
879 name = tempfile.mktemp(dir=path)
879 name = tempfile.mktemp(dir=path)
880 try:
880 try:
881 os.symlink(".", name)
881 os.symlink(".", name)
882 os.unlink(name)
882 os.unlink(name)
883 return True
883 return True
884 except (OSError, AttributeError):
884 except (OSError, AttributeError):
885 return False
885 return False
886
886
887 def linkfunc(path, fallback):
887 def linkfunc(path, fallback):
888 '''return an is_link() function with default to fallback'''
888 '''return an is_link() function with default to fallback'''
889 if checklink(path):
889 if checklink(path):
890 return lambda x: os.path.islink(os.path.join(path, x))
890 return lambda x: os.path.islink(os.path.join(path, x))
891 return fallback
891 return fallback
892
892
893 _umask = os.umask(0)
893 _umask = os.umask(0)
894 os.umask(_umask)
894 os.umask(_umask)
895
895
896 def needbinarypatch():
896 def needbinarypatch():
897 """return True if patches should be applied in binary mode by default."""
897 """return True if patches should be applied in binary mode by default."""
898 return os.name == 'nt'
898 return os.name == 'nt'
899
899
900 # Platform specific variants
900 # Platform specific variants
901 if os.name == 'nt':
901 if os.name == 'nt':
902 import msvcrt
902 import msvcrt
903 nulldev = 'NUL:'
903 nulldev = 'NUL:'
904
904
905 class winstdout:
905 class winstdout:
906 '''stdout on windows misbehaves if sent through a pipe'''
906 '''stdout on windows misbehaves if sent through a pipe'''
907
907
908 def __init__(self, fp):
908 def __init__(self, fp):
909 self.fp = fp
909 self.fp = fp
910
910
911 def __getattr__(self, key):
911 def __getattr__(self, key):
912 return getattr(self.fp, key)
912 return getattr(self.fp, key)
913
913
914 def close(self):
914 def close(self):
915 try:
915 try:
916 self.fp.close()
916 self.fp.close()
917 except: pass
917 except: pass
918
918
919 def write(self, s):
919 def write(self, s):
920 try:
920 try:
921 return self.fp.write(s)
921 return self.fp.write(s)
922 except IOError, inst:
922 except IOError, inst:
923 if inst.errno != 0: raise
923 if inst.errno != 0: raise
924 self.close()
924 self.close()
925 raise IOError(errno.EPIPE, 'Broken pipe')
925 raise IOError(errno.EPIPE, 'Broken pipe')
926
926
927 def flush(self):
927 def flush(self):
928 try:
928 try:
929 return self.fp.flush()
929 return self.fp.flush()
930 except IOError, inst:
930 except IOError, inst:
931 if inst.errno != errno.EINVAL: raise
931 if inst.errno != errno.EINVAL: raise
932 self.close()
932 self.close()
933 raise IOError(errno.EPIPE, 'Broken pipe')
933 raise IOError(errno.EPIPE, 'Broken pipe')
934
934
935 sys.stdout = winstdout(sys.stdout)
935 sys.stdout = winstdout(sys.stdout)
936
936
937 def system_rcpath():
937 def system_rcpath():
938 try:
938 try:
939 return system_rcpath_win32()
939 return system_rcpath_win32()
940 except:
940 except:
941 return [r'c:\mercurial\mercurial.ini']
941 return [r'c:\mercurial\mercurial.ini']
942
942
943 def user_rcpath():
943 def user_rcpath():
944 '''return os-specific hgrc search path to the user dir'''
944 '''return os-specific hgrc search path to the user dir'''
945 try:
945 try:
946 userrc = user_rcpath_win32()
946 userrc = user_rcpath_win32()
947 except:
947 except:
948 userrc = os.path.join(os.path.expanduser('~'), 'mercurial.ini')
948 userrc = os.path.join(os.path.expanduser('~'), 'mercurial.ini')
949 path = [userrc]
949 path = [userrc]
950 userprofile = os.environ.get('USERPROFILE')
950 userprofile = os.environ.get('USERPROFILE')
951 if userprofile:
951 if userprofile:
952 path.append(os.path.join(userprofile, 'mercurial.ini'))
952 path.append(os.path.join(userprofile, 'mercurial.ini'))
953 return path
953 return path
954
954
955 def parse_patch_output(output_line):
955 def parse_patch_output(output_line):
956 """parses the output produced by patch and returns the file name"""
956 """parses the output produced by patch and returns the file name"""
957 pf = output_line[14:]
957 pf = output_line[14:]
958 if pf[0] == '`':
958 if pf[0] == '`':
959 pf = pf[1:-1] # Remove the quotes
959 pf = pf[1:-1] # Remove the quotes
960 return pf
960 return pf
961
961
962 def testpid(pid):
962 def testpid(pid):
963 '''return False if pid dead, True if running or not known'''
963 '''return False if pid dead, True if running or not known'''
964 return True
964 return True
965
965
966 def set_exec(f, mode):
966 def set_exec(f, mode):
967 pass
967 pass
968
968
969 def set_link(f, mode):
969 def set_link(f, mode):
970 pass
970 pass
971
971
972 def set_binary(fd):
972 def set_binary(fd):
973 msvcrt.setmode(fd.fileno(), os.O_BINARY)
973 msvcrt.setmode(fd.fileno(), os.O_BINARY)
974
974
975 def pconvert(path):
975 def pconvert(path):
976 return path.replace("\\", "/")
976 return path.replace("\\", "/")
977
977
978 def localpath(path):
978 def localpath(path):
979 return path.replace('/', '\\')
979 return path.replace('/', '\\')
980
980
981 def normpath(path):
981 def normpath(path):
982 return pconvert(os.path.normpath(path))
982 return pconvert(os.path.normpath(path))
983
983
984 makelock = _makelock_file
984 makelock = _makelock_file
985 readlock = _readlock_file
985 readlock = _readlock_file
986
986
987 def samestat(s1, s2):
987 def samestat(s1, s2):
988 return False
988 return False
989
989
990 # A sequence of backslashes is special iff it precedes a double quote:
990 # A sequence of backslashes is special iff it precedes a double quote:
991 # - if there's an even number of backslashes, the double quote is not
991 # - if there's an even number of backslashes, the double quote is not
992 # quoted (i.e. it ends the quoted region)
992 # quoted (i.e. it ends the quoted region)
993 # - if there's an odd number of backslashes, the double quote is quoted
993 # - if there's an odd number of backslashes, the double quote is quoted
994 # - in both cases, every pair of backslashes is unquoted into a single
994 # - in both cases, every pair of backslashes is unquoted into a single
995 # backslash
995 # backslash
996 # (See http://msdn2.microsoft.com/en-us/library/a1y7w461.aspx )
996 # (See http://msdn2.microsoft.com/en-us/library/a1y7w461.aspx )
997 # So, to quote a string, we must surround it in double quotes, double
997 # So, to quote a string, we must surround it in double quotes, double
998 # the number of backslashes that preceed double quotes and add another
998 # the number of backslashes that preceed double quotes and add another
999 # backslash before every double quote (being careful with the double
999 # backslash before every double quote (being careful with the double
1000 # quote we've appended to the end)
1000 # quote we've appended to the end)
1001 _quotere = None
1001 _quotere = None
1002 def shellquote(s):
1002 def shellquote(s):
1003 global _quotere
1003 global _quotere
1004 if _quotere is None:
1004 if _quotere is None:
1005 _quotere = re.compile(r'(\\*)("|\\$)')
1005 _quotere = re.compile(r'(\\*)("|\\$)')
1006 return '"%s"' % _quotere.sub(r'\1\1\\\2', s)
1006 return '"%s"' % _quotere.sub(r'\1\1\\\2', s)
1007
1007
1008 def quotecommand(cmd):
1008 def quotecommand(cmd):
1009 """Build a command string suitable for os.popen* calls."""
1009 """Build a command string suitable for os.popen* calls."""
1010 # The extra quotes are needed because popen* runs the command
1010 # The extra quotes are needed because popen* runs the command
1011 # through the current COMSPEC. cmd.exe suppress enclosing quotes.
1011 # through the current COMSPEC. cmd.exe suppress enclosing quotes.
1012 return '"' + cmd + '"'
1012 return '"' + cmd + '"'
1013
1013
1014 def explain_exit(code):
1014 def explain_exit(code):
1015 return _("exited with status %d") % code, code
1015 return _("exited with status %d") % code, code
1016
1016
1017 # if you change this stub into a real check, please try to implement the
1017 # if you change this stub into a real check, please try to implement the
1018 # username and groupname functions above, too.
1018 # username and groupname functions above, too.
1019 def isowner(fp, st=None):
1019 def isowner(fp, st=None):
1020 return True
1020 return True
1021
1021
1022 def find_in_path(name, path, default=None):
1022 def find_in_path(name, path, default=None):
1023 '''find name in search path. path can be string (will be split
1023 '''find name in search path. path can be string (will be split
1024 with os.pathsep), or iterable thing that returns strings. if name
1024 with os.pathsep), or iterable thing that returns strings. if name
1025 found, return path to name. else return default. name is looked up
1025 found, return path to name. else return default. name is looked up
1026 using cmd.exe rules, using PATHEXT.'''
1026 using cmd.exe rules, using PATHEXT.'''
1027 if isinstance(path, str):
1027 if isinstance(path, str):
1028 path = path.split(os.pathsep)
1028 path = path.split(os.pathsep)
1029
1029
1030 pathext = os.environ.get('PATHEXT', '.COM;.EXE;.BAT;.CMD')
1030 pathext = os.environ.get('PATHEXT', '.COM;.EXE;.BAT;.CMD')
1031 pathext = pathext.lower().split(os.pathsep)
1031 pathext = pathext.lower().split(os.pathsep)
1032 isexec = os.path.splitext(name)[1].lower() in pathext
1032 isexec = os.path.splitext(name)[1].lower() in pathext
1033
1033
1034 for p in path:
1034 for p in path:
1035 p_name = os.path.join(p, name)
1035 p_name = os.path.join(p, name)
1036
1036
1037 if isexec and os.path.exists(p_name):
1037 if isexec and os.path.exists(p_name):
1038 return p_name
1038 return p_name
1039
1039
1040 for ext in pathext:
1040 for ext in pathext:
1041 p_name_ext = p_name + ext
1041 p_name_ext = p_name + ext
1042 if os.path.exists(p_name_ext):
1042 if os.path.exists(p_name_ext):
1043 return p_name_ext
1043 return p_name_ext
1044 return default
1044 return default
1045
1045
1046 def set_signal_handler():
1046 def set_signal_handler():
1047 try:
1047 try:
1048 set_signal_handler_win32()
1048 set_signal_handler_win32()
1049 except NameError:
1049 except NameError:
1050 pass
1050 pass
1051
1051
1052 try:
1052 try:
1053 # override functions with win32 versions if possible
1053 # override functions with win32 versions if possible
1054 from util_win32 import *
1054 from util_win32 import *
1055 if not is_win_9x():
1055 if not is_win_9x():
1056 posixfile = posixfile_nt
1056 posixfile = posixfile_nt
1057 except ImportError:
1057 except ImportError:
1058 pass
1058 pass
1059
1059
1060 else:
1060 else:
1061 nulldev = '/dev/null'
1061 nulldev = '/dev/null'
1062
1062
1063 def rcfiles(path):
1063 def rcfiles(path):
1064 rcs = [os.path.join(path, 'hgrc')]
1064 rcs = [os.path.join(path, 'hgrc')]
1065 rcdir = os.path.join(path, 'hgrc.d')
1065 rcdir = os.path.join(path, 'hgrc.d')
1066 try:
1066 try:
1067 rcs.extend([os.path.join(rcdir, f)
1067 rcs.extend([os.path.join(rcdir, f)
1068 for f, kind in osutil.listdir(rcdir)
1068 for f, kind in osutil.listdir(rcdir)
1069 if f.endswith(".rc")])
1069 if f.endswith(".rc")])
1070 except OSError:
1070 except OSError:
1071 pass
1071 pass
1072 return rcs
1072 return rcs
1073
1073
1074 def system_rcpath():
1074 def system_rcpath():
1075 path = []
1075 path = []
1076 # old mod_python does not set sys.argv
1076 # old mod_python does not set sys.argv
1077 if len(getattr(sys, 'argv', [])) > 0:
1077 if len(getattr(sys, 'argv', [])) > 0:
1078 path.extend(rcfiles(os.path.dirname(sys.argv[0]) +
1078 path.extend(rcfiles(os.path.dirname(sys.argv[0]) +
1079 '/../etc/mercurial'))
1079 '/../etc/mercurial'))
1080 path.extend(rcfiles('/etc/mercurial'))
1080 path.extend(rcfiles('/etc/mercurial'))
1081 return path
1081 return path
1082
1082
1083 def user_rcpath():
1083 def user_rcpath():
1084 return [os.path.expanduser('~/.hgrc')]
1084 return [os.path.expanduser('~/.hgrc')]
1085
1085
1086 def parse_patch_output(output_line):
1086 def parse_patch_output(output_line):
1087 """parses the output produced by patch and returns the file name"""
1087 """parses the output produced by patch and returns the file name"""
1088 pf = output_line[14:]
1088 pf = output_line[14:]
1089 if os.sys.platform == 'OpenVMS':
1089 if os.sys.platform == 'OpenVMS':
1090 if pf[0] == '`':
1090 if pf[0] == '`':
1091 pf = pf[1:-1] # Remove the quotes
1091 pf = pf[1:-1] # Remove the quotes
1092 else:
1092 else:
1093 if pf.startswith("'") and pf.endswith("'") and " " in pf:
1093 if pf.startswith("'") and pf.endswith("'") and " " in pf:
1094 pf = pf[1:-1] # Remove the quotes
1094 pf = pf[1:-1] # Remove the quotes
1095 return pf
1095 return pf
1096
1096
1097 def is_exec(f):
1097 def is_exec(f):
1098 """check whether a file is executable"""
1098 """check whether a file is executable"""
1099 return (os.lstat(f).st_mode & 0100 != 0)
1099 return (os.lstat(f).st_mode & 0100 != 0)
1100
1100
1101 def set_exec(f, mode):
1101 def set_exec(f, mode):
1102 s = os.lstat(f).st_mode
1102 s = os.lstat(f).st_mode
1103 if stat.S_ISLNK(s) or (s & 0100 != 0) == mode:
1103 if stat.S_ISLNK(s) or (s & 0100 != 0) == mode:
1104 return
1104 return
1105 if mode:
1105 if mode:
1106 # Turn on +x for every +r bit when making a file executable
1106 # Turn on +x for every +r bit when making a file executable
1107 # and obey umask.
1107 # and obey umask.
1108 os.chmod(f, s | (s & 0444) >> 2 & ~_umask)
1108 os.chmod(f, s | (s & 0444) >> 2 & ~_umask)
1109 else:
1109 else:
1110 os.chmod(f, s & 0666)
1110 os.chmod(f, s & 0666)
1111
1111
1112 def set_link(f, mode):
1112 def set_link(f, mode):
1113 """make a file a symbolic link/regular file
1113 """make a file a symbolic link/regular file
1114
1114
1115 if a file is changed to a link, its contents become the link data
1115 if a file is changed to a link, its contents become the link data
1116 if a link is changed to a file, its link data become its contents
1116 if a link is changed to a file, its link data become its contents
1117 """
1117 """
1118
1118
1119 m = os.path.islink(f)
1119 m = os.path.islink(f)
1120 if m == bool(mode):
1120 if m == bool(mode):
1121 return
1121 return
1122
1122
1123 if mode: # switch file to link
1123 if mode: # switch file to link
1124 data = file(f).read()
1124 data = file(f).read()
1125 os.unlink(f)
1125 os.unlink(f)
1126 os.symlink(data, f)
1126 os.symlink(data, f)
1127 else:
1127 else:
1128 data = os.readlink(f)
1128 data = os.readlink(f)
1129 os.unlink(f)
1129 os.unlink(f)
1130 file(f, "w").write(data)
1130 file(f, "w").write(data)
1131
1131
1132 def set_binary(fd):
1132 def set_binary(fd):
1133 pass
1133 pass
1134
1134
1135 def pconvert(path):
1135 def pconvert(path):
1136 return path
1136 return path
1137
1137
1138 def localpath(path):
1138 def localpath(path):
1139 return path
1139 return path
1140
1140
1141 normpath = os.path.normpath
1141 normpath = os.path.normpath
1142 samestat = os.path.samestat
1142 samestat = os.path.samestat
1143
1143
1144 def makelock(info, pathname):
1144 def makelock(info, pathname):
1145 try:
1145 try:
1146 os.symlink(info, pathname)
1146 os.symlink(info, pathname)
1147 except OSError, why:
1147 except OSError, why:
1148 if why.errno == errno.EEXIST:
1148 if why.errno == errno.EEXIST:
1149 raise
1149 raise
1150 else:
1150 else:
1151 _makelock_file(info, pathname)
1151 _makelock_file(info, pathname)
1152
1152
1153 def readlock(pathname):
1153 def readlock(pathname):
1154 try:
1154 try:
1155 return os.readlink(pathname)
1155 return os.readlink(pathname)
1156 except OSError, why:
1156 except OSError, why:
1157 if why.errno in (errno.EINVAL, errno.ENOSYS):
1157 if why.errno in (errno.EINVAL, errno.ENOSYS):
1158 return _readlock_file(pathname)
1158 return _readlock_file(pathname)
1159 else:
1159 else:
1160 raise
1160 raise
1161
1161
1162 def shellquote(s):
1162 def shellquote(s):
1163 if os.sys.platform == 'OpenVMS':
1163 if os.sys.platform == 'OpenVMS':
1164 return '"%s"' % s
1164 return '"%s"' % s
1165 else:
1165 else:
1166 return "'%s'" % s.replace("'", "'\\''")
1166 return "'%s'" % s.replace("'", "'\\''")
1167
1167
1168 def quotecommand(cmd):
1168 def quotecommand(cmd):
1169 return cmd
1169 return cmd
1170
1170
1171 def testpid(pid):
1171 def testpid(pid):
1172 '''return False if pid dead, True if running or not sure'''
1172 '''return False if pid dead, True if running or not sure'''
1173 if os.sys.platform == 'OpenVMS':
1173 if os.sys.platform == 'OpenVMS':
1174 return True
1174 return True
1175 try:
1175 try:
1176 os.kill(pid, 0)
1176 os.kill(pid, 0)
1177 return True
1177 return True
1178 except OSError, inst:
1178 except OSError, inst:
1179 return inst.errno != errno.ESRCH
1179 return inst.errno != errno.ESRCH
1180
1180
1181 def explain_exit(code):
1181 def explain_exit(code):
1182 """return a 2-tuple (desc, code) describing a process's status"""
1182 """return a 2-tuple (desc, code) describing a process's status"""
1183 if os.WIFEXITED(code):
1183 if os.WIFEXITED(code):
1184 val = os.WEXITSTATUS(code)
1184 val = os.WEXITSTATUS(code)
1185 return _("exited with status %d") % val, val
1185 return _("exited with status %d") % val, val
1186 elif os.WIFSIGNALED(code):
1186 elif os.WIFSIGNALED(code):
1187 val = os.WTERMSIG(code)
1187 val = os.WTERMSIG(code)
1188 return _("killed by signal %d") % val, val
1188 return _("killed by signal %d") % val, val
1189 elif os.WIFSTOPPED(code):
1189 elif os.WIFSTOPPED(code):
1190 val = os.WSTOPSIG(code)
1190 val = os.WSTOPSIG(code)
1191 return _("stopped by signal %d") % val, val
1191 return _("stopped by signal %d") % val, val
1192 raise ValueError(_("invalid exit code"))
1192 raise ValueError(_("invalid exit code"))
1193
1193
1194 def isowner(fp, st=None):
1194 def isowner(fp, st=None):
1195 """Return True if the file object f belongs to the current user.
1195 """Return True if the file object f belongs to the current user.
1196
1196
1197 The return value of a util.fstat(f) may be passed as the st argument.
1197 The return value of a util.fstat(f) may be passed as the st argument.
1198 """
1198 """
1199 if st is None:
1199 if st is None:
1200 st = fstat(fp)
1200 st = fstat(fp)
1201 return st.st_uid == os.getuid()
1201 return st.st_uid == os.getuid()
1202
1202
1203 def find_in_path(name, path, default=None):
1203 def find_in_path(name, path, default=None):
1204 '''find name in search path. path can be string (will be split
1204 '''find name in search path. path can be string (will be split
1205 with os.pathsep), or iterable thing that returns strings. if name
1205 with os.pathsep), or iterable thing that returns strings. if name
1206 found, return path to name. else return default.'''
1206 found, return path to name. else return default.'''
1207 if isinstance(path, str):
1207 if isinstance(path, str):
1208 path = path.split(os.pathsep)
1208 path = path.split(os.pathsep)
1209 for p in path:
1209 for p in path:
1210 p_name = os.path.join(p, name)
1210 p_name = os.path.join(p, name)
1211 if os.path.exists(p_name):
1211 if os.path.exists(p_name):
1212 return p_name
1212 return p_name
1213 return default
1213 return default
1214
1214
1215 def set_signal_handler():
1215 def set_signal_handler():
1216 pass
1216 pass
1217
1217
1218 def find_exe(name, default=None):
1218 def find_exe(name, default=None):
1219 '''find path of an executable.
1219 '''find path of an executable.
1220 if name contains a path component, return it as is. otherwise,
1220 if name contains a path component, return it as is. otherwise,
1221 use normal executable search path.'''
1221 use normal executable search path.'''
1222
1222
1223 if os.sep in name or sys.platform == 'OpenVMS':
1223 if os.sep in name or sys.platform == 'OpenVMS':
1224 # don't check the executable bit. if the file isn't
1224 # don't check the executable bit. if the file isn't
1225 # executable, whoever tries to actually run it will give a
1225 # executable, whoever tries to actually run it will give a
1226 # much more useful error message.
1226 # much more useful error message.
1227 return name
1227 return name
1228 return find_in_path(name, os.environ.get('PATH', ''), default=default)
1228 return find_in_path(name, os.environ.get('PATH', ''), default=default)
1229
1229
1230 def _buildencodefun():
1230 def _buildencodefun():
1231 e = '_'
1231 e = '_'
1232 win_reserved = [ord(x) for x in '\\:*?"<>|']
1232 win_reserved = [ord(x) for x in '\\:*?"<>|']
1233 cmap = dict([ (chr(x), chr(x)) for x in xrange(127) ])
1233 cmap = dict([ (chr(x), chr(x)) for x in xrange(127) ])
1234 for x in (range(32) + range(126, 256) + win_reserved):
1234 for x in (range(32) + range(126, 256) + win_reserved):
1235 cmap[chr(x)] = "~%02x" % x
1235 cmap[chr(x)] = "~%02x" % x
1236 for x in range(ord("A"), ord("Z")+1) + [ord(e)]:
1236 for x in range(ord("A"), ord("Z")+1) + [ord(e)]:
1237 cmap[chr(x)] = e + chr(x).lower()
1237 cmap[chr(x)] = e + chr(x).lower()
1238 dmap = {}
1238 dmap = {}
1239 for k, v in cmap.iteritems():
1239 for k, v in cmap.iteritems():
1240 dmap[v] = k
1240 dmap[v] = k
1241 def decode(s):
1241 def decode(s):
1242 i = 0
1242 i = 0
1243 while i < len(s):
1243 while i < len(s):
1244 for l in xrange(1, 4):
1244 for l in xrange(1, 4):
1245 try:
1245 try:
1246 yield dmap[s[i:i+l]]
1246 yield dmap[s[i:i+l]]
1247 i += l
1247 i += l
1248 break
1248 break
1249 except KeyError:
1249 except KeyError:
1250 pass
1250 pass
1251 else:
1251 else:
1252 raise KeyError
1252 raise KeyError
1253 return (lambda s: "".join([cmap[c] for c in s]),
1253 return (lambda s: "".join([cmap[c] for c in s]),
1254 lambda s: "".join(list(decode(s))))
1254 lambda s: "".join(list(decode(s))))
1255
1255
1256 encodefilename, decodefilename = _buildencodefun()
1256 encodefilename, decodefilename = _buildencodefun()
1257
1257
1258 def encodedopener(openerfn, fn):
1258 def encodedopener(openerfn, fn):
1259 def o(path, *args, **kw):
1259 def o(path, *args, **kw):
1260 return openerfn(fn(path), *args, **kw)
1260 return openerfn(fn(path), *args, **kw)
1261 return o
1261 return o
1262
1262
1263 def mktempcopy(name, emptyok=False):
1263 def mktempcopy(name, emptyok=False):
1264 """Create a temporary file with the same contents from name
1264 """Create a temporary file with the same contents from name
1265
1265
1266 The permission bits are copied from the original file.
1266 The permission bits are copied from the original file.
1267
1267
1268 If the temporary file is going to be truncated immediately, you
1268 If the temporary file is going to be truncated immediately, you
1269 can use emptyok=True as an optimization.
1269 can use emptyok=True as an optimization.
1270
1270
1271 Returns the name of the temporary file.
1271 Returns the name of the temporary file.
1272 """
1272 """
1273 d, fn = os.path.split(name)
1273 d, fn = os.path.split(name)
1274 fd, temp = tempfile.mkstemp(prefix='.%s-' % fn, dir=d)
1274 fd, temp = tempfile.mkstemp(prefix='.%s-' % fn, dir=d)
1275 os.close(fd)
1275 os.close(fd)
1276 # Temporary files are created with mode 0600, which is usually not
1276 # Temporary files are created with mode 0600, which is usually not
1277 # what we want. If the original file already exists, just copy
1277 # what we want. If the original file already exists, just copy
1278 # its mode. Otherwise, manually obey umask.
1278 # its mode. Otherwise, manually obey umask.
1279 try:
1279 try:
1280 st_mode = os.lstat(name).st_mode
1280 st_mode = os.lstat(name).st_mode
1281 except OSError, inst:
1281 except OSError, inst:
1282 if inst.errno != errno.ENOENT:
1282 if inst.errno != errno.ENOENT:
1283 raise
1283 raise
1284 st_mode = 0666 & ~_umask
1284 st_mode = 0666 & ~_umask
1285 os.chmod(temp, st_mode)
1285 os.chmod(temp, st_mode)
1286 if emptyok:
1286 if emptyok:
1287 return temp
1287 return temp
1288 try:
1288 try:
1289 try:
1289 try:
1290 ifp = posixfile(name, "rb")
1290 ifp = posixfile(name, "rb")
1291 except IOError, inst:
1291 except IOError, inst:
1292 if inst.errno == errno.ENOENT:
1292 if inst.errno == errno.ENOENT:
1293 return temp
1293 return temp
1294 if not getattr(inst, 'filename', None):
1294 if not getattr(inst, 'filename', None):
1295 inst.filename = name
1295 inst.filename = name
1296 raise
1296 raise
1297 ofp = posixfile(temp, "wb")
1297 ofp = posixfile(temp, "wb")
1298 for chunk in filechunkiter(ifp):
1298 for chunk in filechunkiter(ifp):
1299 ofp.write(chunk)
1299 ofp.write(chunk)
1300 ifp.close()
1300 ifp.close()
1301 ofp.close()
1301 ofp.close()
1302 except:
1302 except:
1303 try: os.unlink(temp)
1303 try: os.unlink(temp)
1304 except: pass
1304 except: pass
1305 raise
1305 raise
1306 return temp
1306 return temp
1307
1307
1308 class atomictempfile(posixfile):
1308 class atomictempfile(posixfile):
1309 """file-like object that atomically updates a file
1309 """file-like object that atomically updates a file
1310
1310
1311 All writes will be redirected to a temporary copy of the original
1311 All writes will be redirected to a temporary copy of the original
1312 file. When rename is called, the copy is renamed to the original
1312 file. When rename is called, the copy is renamed to the original
1313 name, making the changes visible.
1313 name, making the changes visible.
1314 """
1314 """
1315 def __init__(self, name, mode):
1315 def __init__(self, name, mode):
1316 self.__name = name
1316 self.__name = name
1317 self.temp = mktempcopy(name, emptyok=('w' in mode))
1317 self.temp = mktempcopy(name, emptyok=('w' in mode))
1318 posixfile.__init__(self, self.temp, mode)
1318 posixfile.__init__(self, self.temp, mode)
1319
1319
1320 def rename(self):
1320 def rename(self):
1321 if not self.closed:
1321 if not self.closed:
1322 posixfile.close(self)
1322 posixfile.close(self)
1323 rename(self.temp, localpath(self.__name))
1323 rename(self.temp, localpath(self.__name))
1324
1324
1325 def __del__(self):
1325 def __del__(self):
1326 if not self.closed:
1326 if not self.closed:
1327 try:
1327 try:
1328 os.unlink(self.temp)
1328 os.unlink(self.temp)
1329 except: pass
1329 except: pass
1330 posixfile.close(self)
1330 posixfile.close(self)
1331
1331
1332 class opener(object):
1332 class opener(object):
1333 """Open files relative to a base directory
1333 """Open files relative to a base directory
1334
1334
1335 This class is used to hide the details of COW semantics and
1335 This class is used to hide the details of COW semantics and
1336 remote file access from higher level code.
1336 remote file access from higher level code.
1337 """
1337 """
1338 def __init__(self, base, audit=True):
1338 def __init__(self, base, audit=True):
1339 self.base = base
1339 self.base = base
1340 if audit:
1340 if audit:
1341 self.audit_path = path_auditor(base)
1341 self.audit_path = path_auditor(base)
1342 else:
1342 else:
1343 self.audit_path = always
1343 self.audit_path = always
1344
1344
1345 def __getattr__(self, name):
1345 def __getattr__(self, name):
1346 if name == '_can_symlink':
1346 if name == '_can_symlink':
1347 self._can_symlink = checklink(self.base)
1347 self._can_symlink = checklink(self.base)
1348 return self._can_symlink
1348 return self._can_symlink
1349 raise AttributeError(name)
1349 raise AttributeError(name)
1350
1350
1351 def __call__(self, path, mode="r", text=False, atomictemp=False):
1351 def __call__(self, path, mode="r", text=False, atomictemp=False):
1352 self.audit_path(path)
1352 self.audit_path(path)
1353 f = os.path.join(self.base, path)
1353 f = os.path.join(self.base, path)
1354
1354
1355 if not text and "b" not in mode:
1355 if not text and "b" not in mode:
1356 mode += "b" # for that other OS
1356 mode += "b" # for that other OS
1357
1357
1358 if mode[0] != "r":
1358 if mode[0] != "r":
1359 try:
1359 try:
1360 nlink = nlinks(f)
1360 nlink = nlinks(f)
1361 except OSError:
1361 except OSError:
1362 nlink = 0
1362 nlink = 0
1363 d = os.path.dirname(f)
1363 d = os.path.dirname(f)
1364 if not os.path.isdir(d):
1364 if not os.path.isdir(d):
1365 os.makedirs(d)
1365 os.makedirs(d)
1366 if atomictemp:
1366 if atomictemp:
1367 return atomictempfile(f, mode)
1367 return atomictempfile(f, mode)
1368 if nlink > 1:
1368 if nlink > 1:
1369 rename(mktempcopy(f), f)
1369 rename(mktempcopy(f), f)
1370 return posixfile(f, mode)
1370 return posixfile(f, mode)
1371
1371
1372 def symlink(self, src, dst):
1372 def symlink(self, src, dst):
1373 self.audit_path(dst)
1373 self.audit_path(dst)
1374 linkname = os.path.join(self.base, dst)
1374 linkname = os.path.join(self.base, dst)
1375 try:
1375 try:
1376 os.unlink(linkname)
1376 os.unlink(linkname)
1377 except OSError:
1377 except OSError:
1378 pass
1378 pass
1379
1379
1380 dirname = os.path.dirname(linkname)
1380 dirname = os.path.dirname(linkname)
1381 if not os.path.exists(dirname):
1381 if not os.path.exists(dirname):
1382 os.makedirs(dirname)
1382 os.makedirs(dirname)
1383
1383
1384 if self._can_symlink:
1384 if self._can_symlink:
1385 try:
1385 try:
1386 os.symlink(src, linkname)
1386 os.symlink(src, linkname)
1387 except OSError, err:
1387 except OSError, err:
1388 raise OSError(err.errno, _('could not symlink to %r: %s') %
1388 raise OSError(err.errno, _('could not symlink to %r: %s') %
1389 (src, err.strerror), linkname)
1389 (src, err.strerror), linkname)
1390 else:
1390 else:
1391 f = self(dst, "w")
1391 f = self(dst, "w")
1392 f.write(src)
1392 f.write(src)
1393 f.close()
1393 f.close()
1394
1394
1395 class chunkbuffer(object):
1395 class chunkbuffer(object):
1396 """Allow arbitrary sized chunks of data to be efficiently read from an
1396 """Allow arbitrary sized chunks of data to be efficiently read from an
1397 iterator over chunks of arbitrary size."""
1397 iterator over chunks of arbitrary size."""
1398
1398
1399 def __init__(self, in_iter, targetsize = 2**16):
1399 def __init__(self, in_iter):
1400 """in_iter is the iterator that's iterating over the input chunks.
1400 """in_iter is the iterator that's iterating over the input chunks.
1401 targetsize is how big a buffer to try to maintain."""
1401 targetsize is how big a buffer to try to maintain."""
1402 self.in_iter = iter(in_iter)
1402 self.in_iter = iter(in_iter)
1403 self.buf = ''
1403 self.buf = ''
1404 self.targetsize = int(targetsize)
1404 self.targetsize = 2**16
1405 if self.targetsize <= 0:
1406 raise ValueError(_("targetsize must be greater than 0, was %d") %
1407 targetsize)
1408 self.iterempty = False
1405 self.iterempty = False
1409
1406
1410 def fillbuf(self):
1411 """Ignore target size; read every chunk from iterator until empty."""
1412 if not self.iterempty:
1413 collector = cStringIO.StringIO()
1414 collector.write(self.buf)
1415 for ch in self.in_iter:
1416 collector.write(ch)
1417 self.buf = collector.getvalue()
1418 self.iterempty = True
1419
1420 def read(self, l):
1407 def read(self, l):
1421 """Read L bytes of data from the iterator of chunks of data.
1408 """Read L bytes of data from the iterator of chunks of data.
1422 Returns less than L bytes if the iterator runs dry."""
1409 Returns less than L bytes if the iterator runs dry."""
1423 if l > len(self.buf) and not self.iterempty:
1410 if l > len(self.buf) and not self.iterempty:
1424 # Clamp to a multiple of self.targetsize
1411 # Clamp to a multiple of self.targetsize
1425 targetsize = self.targetsize * ((l // self.targetsize) + 1)
1412 targetsize = self.targetsize * ((l // self.targetsize) + 1)
1426 collector = cStringIO.StringIO()
1413 collector = cStringIO.StringIO()
1427 collector.write(self.buf)
1414 collector.write(self.buf)
1428 collected = len(self.buf)
1415 collected = len(self.buf)
1429 for chunk in self.in_iter:
1416 for chunk in self.in_iter:
1430 collector.write(chunk)
1417 collector.write(chunk)
1431 collected += len(chunk)
1418 collected += len(chunk)
1432 if collected >= targetsize:
1419 if collected >= targetsize:
1433 break
1420 break
1434 if collected < targetsize:
1421 if collected < targetsize:
1435 self.iterempty = True
1422 self.iterempty = True
1436 self.buf = collector.getvalue()
1423 self.buf = collector.getvalue()
1437 s, self.buf = self.buf[:l], buffer(self.buf, l)
1424 s, self.buf = self.buf[:l], buffer(self.buf, l)
1438 return s
1425 return s
1439
1426
1440 def filechunkiter(f, size=65536, limit=None):
1427 def filechunkiter(f, size=65536, limit=None):
1441 """Create a generator that produces the data in the file size
1428 """Create a generator that produces the data in the file size
1442 (default 65536) bytes at a time, up to optional limit (default is
1429 (default 65536) bytes at a time, up to optional limit (default is
1443 to read all data). Chunks may be less than size bytes if the
1430 to read all data). Chunks may be less than size bytes if the
1444 chunk is the last chunk in the file, or the file is a socket or
1431 chunk is the last chunk in the file, or the file is a socket or
1445 some other type of file that sometimes reads less data than is
1432 some other type of file that sometimes reads less data than is
1446 requested."""
1433 requested."""
1447 assert size >= 0
1434 assert size >= 0
1448 assert limit is None or limit >= 0
1435 assert limit is None or limit >= 0
1449 while True:
1436 while True:
1450 if limit is None: nbytes = size
1437 if limit is None: nbytes = size
1451 else: nbytes = min(limit, size)
1438 else: nbytes = min(limit, size)
1452 s = nbytes and f.read(nbytes)
1439 s = nbytes and f.read(nbytes)
1453 if not s: break
1440 if not s: break
1454 if limit: limit -= len(s)
1441 if limit: limit -= len(s)
1455 yield s
1442 yield s
1456
1443
1457 def makedate():
1444 def makedate():
1458 lt = time.localtime()
1445 lt = time.localtime()
1459 if lt[8] == 1 and time.daylight:
1446 if lt[8] == 1 and time.daylight:
1460 tz = time.altzone
1447 tz = time.altzone
1461 else:
1448 else:
1462 tz = time.timezone
1449 tz = time.timezone
1463 return time.mktime(lt), tz
1450 return time.mktime(lt), tz
1464
1451
1465 def datestr(date=None, format='%a %b %d %H:%M:%S %Y', timezone=True, timezone_format=" %+03d%02d"):
1452 def datestr(date=None, format='%a %b %d %H:%M:%S %Y', timezone=True, timezone_format=" %+03d%02d"):
1466 """represent a (unixtime, offset) tuple as a localized time.
1453 """represent a (unixtime, offset) tuple as a localized time.
1467 unixtime is seconds since the epoch, and offset is the time zone's
1454 unixtime is seconds since the epoch, and offset is the time zone's
1468 number of seconds away from UTC. if timezone is false, do not
1455 number of seconds away from UTC. if timezone is false, do not
1469 append time zone to string."""
1456 append time zone to string."""
1470 t, tz = date or makedate()
1457 t, tz = date or makedate()
1471 s = time.strftime(format, time.gmtime(float(t) - tz))
1458 s = time.strftime(format, time.gmtime(float(t) - tz))
1472 if timezone:
1459 if timezone:
1473 s += timezone_format % (-tz / 3600, ((-tz % 3600) / 60))
1460 s += timezone_format % (-tz / 3600, ((-tz % 3600) / 60))
1474 return s
1461 return s
1475
1462
1476 def strdate(string, format, defaults=[]):
1463 def strdate(string, format, defaults=[]):
1477 """parse a localized time string and return a (unixtime, offset) tuple.
1464 """parse a localized time string and return a (unixtime, offset) tuple.
1478 if the string cannot be parsed, ValueError is raised."""
1465 if the string cannot be parsed, ValueError is raised."""
1479 def timezone(string):
1466 def timezone(string):
1480 tz = string.split()[-1]
1467 tz = string.split()[-1]
1481 if tz[0] in "+-" and len(tz) == 5 and tz[1:].isdigit():
1468 if tz[0] in "+-" and len(tz) == 5 and tz[1:].isdigit():
1482 tz = int(tz)
1469 tz = int(tz)
1483 offset = - 3600 * (tz / 100) - 60 * (tz % 100)
1470 offset = - 3600 * (tz / 100) - 60 * (tz % 100)
1484 return offset
1471 return offset
1485 if tz == "GMT" or tz == "UTC":
1472 if tz == "GMT" or tz == "UTC":
1486 return 0
1473 return 0
1487 return None
1474 return None
1488
1475
1489 # NOTE: unixtime = localunixtime + offset
1476 # NOTE: unixtime = localunixtime + offset
1490 offset, date = timezone(string), string
1477 offset, date = timezone(string), string
1491 if offset != None:
1478 if offset != None:
1492 date = " ".join(string.split()[:-1])
1479 date = " ".join(string.split()[:-1])
1493
1480
1494 # add missing elements from defaults
1481 # add missing elements from defaults
1495 for part in defaults:
1482 for part in defaults:
1496 found = [True for p in part if ("%"+p) in format]
1483 found = [True for p in part if ("%"+p) in format]
1497 if not found:
1484 if not found:
1498 date += "@" + defaults[part]
1485 date += "@" + defaults[part]
1499 format += "@%" + part[0]
1486 format += "@%" + part[0]
1500
1487
1501 timetuple = time.strptime(date, format)
1488 timetuple = time.strptime(date, format)
1502 localunixtime = int(calendar.timegm(timetuple))
1489 localunixtime = int(calendar.timegm(timetuple))
1503 if offset is None:
1490 if offset is None:
1504 # local timezone
1491 # local timezone
1505 unixtime = int(time.mktime(timetuple))
1492 unixtime = int(time.mktime(timetuple))
1506 offset = unixtime - localunixtime
1493 offset = unixtime - localunixtime
1507 else:
1494 else:
1508 unixtime = localunixtime + offset
1495 unixtime = localunixtime + offset
1509 return unixtime, offset
1496 return unixtime, offset
1510
1497
1511 def parsedate(string, formats=None, defaults=None):
1498 def parsedate(string, formats=None, defaults=None):
1512 """parse a localized time string and return a (unixtime, offset) tuple.
1499 """parse a localized time string and return a (unixtime, offset) tuple.
1513 The date may be a "unixtime offset" string or in one of the specified
1500 The date may be a "unixtime offset" string or in one of the specified
1514 formats."""
1501 formats."""
1515 if not string:
1502 if not string:
1516 return 0, 0
1503 return 0, 0
1517 if not formats:
1504 if not formats:
1518 formats = defaultdateformats
1505 formats = defaultdateformats
1519 string = string.strip()
1506 string = string.strip()
1520 try:
1507 try:
1521 when, offset = map(int, string.split(' '))
1508 when, offset = map(int, string.split(' '))
1522 except ValueError:
1509 except ValueError:
1523 # fill out defaults
1510 # fill out defaults
1524 if not defaults:
1511 if not defaults:
1525 defaults = {}
1512 defaults = {}
1526 now = makedate()
1513 now = makedate()
1527 for part in "d mb yY HI M S".split():
1514 for part in "d mb yY HI M S".split():
1528 if part not in defaults:
1515 if part not in defaults:
1529 if part[0] in "HMS":
1516 if part[0] in "HMS":
1530 defaults[part] = "00"
1517 defaults[part] = "00"
1531 elif part[0] in "dm":
1518 elif part[0] in "dm":
1532 defaults[part] = "1"
1519 defaults[part] = "1"
1533 else:
1520 else:
1534 defaults[part] = datestr(now, "%" + part[0], False)
1521 defaults[part] = datestr(now, "%" + part[0], False)
1535
1522
1536 for format in formats:
1523 for format in formats:
1537 try:
1524 try:
1538 when, offset = strdate(string, format, defaults)
1525 when, offset = strdate(string, format, defaults)
1539 except ValueError:
1526 except ValueError:
1540 pass
1527 pass
1541 else:
1528 else:
1542 break
1529 break
1543 else:
1530 else:
1544 raise Abort(_('invalid date: %r ') % string)
1531 raise Abort(_('invalid date: %r ') % string)
1545 # validate explicit (probably user-specified) date and
1532 # validate explicit (probably user-specified) date and
1546 # time zone offset. values must fit in signed 32 bits for
1533 # time zone offset. values must fit in signed 32 bits for
1547 # current 32-bit linux runtimes. timezones go from UTC-12
1534 # current 32-bit linux runtimes. timezones go from UTC-12
1548 # to UTC+14
1535 # to UTC+14
1549 if abs(when) > 0x7fffffff:
1536 if abs(when) > 0x7fffffff:
1550 raise Abort(_('date exceeds 32 bits: %d') % when)
1537 raise Abort(_('date exceeds 32 bits: %d') % when)
1551 if offset < -50400 or offset > 43200:
1538 if offset < -50400 or offset > 43200:
1552 raise Abort(_('impossible time zone offset: %d') % offset)
1539 raise Abort(_('impossible time zone offset: %d') % offset)
1553 return when, offset
1540 return when, offset
1554
1541
1555 def matchdate(date):
1542 def matchdate(date):
1556 """Return a function that matches a given date match specifier
1543 """Return a function that matches a given date match specifier
1557
1544
1558 Formats include:
1545 Formats include:
1559
1546
1560 '{date}' match a given date to the accuracy provided
1547 '{date}' match a given date to the accuracy provided
1561
1548
1562 '<{date}' on or before a given date
1549 '<{date}' on or before a given date
1563
1550
1564 '>{date}' on or after a given date
1551 '>{date}' on or after a given date
1565
1552
1566 """
1553 """
1567
1554
1568 def lower(date):
1555 def lower(date):
1569 return parsedate(date, extendeddateformats)[0]
1556 return parsedate(date, extendeddateformats)[0]
1570
1557
1571 def upper(date):
1558 def upper(date):
1572 d = dict(mb="12", HI="23", M="59", S="59")
1559 d = dict(mb="12", HI="23", M="59", S="59")
1573 for days in "31 30 29".split():
1560 for days in "31 30 29".split():
1574 try:
1561 try:
1575 d["d"] = days
1562 d["d"] = days
1576 return parsedate(date, extendeddateformats, d)[0]
1563 return parsedate(date, extendeddateformats, d)[0]
1577 except:
1564 except:
1578 pass
1565 pass
1579 d["d"] = "28"
1566 d["d"] = "28"
1580 return parsedate(date, extendeddateformats, d)[0]
1567 return parsedate(date, extendeddateformats, d)[0]
1581
1568
1582 if date[0] == "<":
1569 if date[0] == "<":
1583 when = upper(date[1:])
1570 when = upper(date[1:])
1584 return lambda x: x <= when
1571 return lambda x: x <= when
1585 elif date[0] == ">":
1572 elif date[0] == ">":
1586 when = lower(date[1:])
1573 when = lower(date[1:])
1587 return lambda x: x >= when
1574 return lambda x: x >= when
1588 elif date[0] == "-":
1575 elif date[0] == "-":
1589 try:
1576 try:
1590 days = int(date[1:])
1577 days = int(date[1:])
1591 except ValueError:
1578 except ValueError:
1592 raise Abort(_("invalid day spec: %s") % date[1:])
1579 raise Abort(_("invalid day spec: %s") % date[1:])
1593 when = makedate()[0] - days * 3600 * 24
1580 when = makedate()[0] - days * 3600 * 24
1594 return lambda x: x >= when
1581 return lambda x: x >= when
1595 elif " to " in date:
1582 elif " to " in date:
1596 a, b = date.split(" to ")
1583 a, b = date.split(" to ")
1597 start, stop = lower(a), upper(b)
1584 start, stop = lower(a), upper(b)
1598 return lambda x: x >= start and x <= stop
1585 return lambda x: x >= start and x <= stop
1599 else:
1586 else:
1600 start, stop = lower(date), upper(date)
1587 start, stop = lower(date), upper(date)
1601 return lambda x: x >= start and x <= stop
1588 return lambda x: x >= start and x <= stop
1602
1589
1603 def shortuser(user):
1590 def shortuser(user):
1604 """Return a short representation of a user name or email address."""
1591 """Return a short representation of a user name or email address."""
1605 f = user.find('@')
1592 f = user.find('@')
1606 if f >= 0:
1593 if f >= 0:
1607 user = user[:f]
1594 user = user[:f]
1608 f = user.find('<')
1595 f = user.find('<')
1609 if f >= 0:
1596 if f >= 0:
1610 user = user[f+1:]
1597 user = user[f+1:]
1611 f = user.find(' ')
1598 f = user.find(' ')
1612 if f >= 0:
1599 if f >= 0:
1613 user = user[:f]
1600 user = user[:f]
1614 f = user.find('.')
1601 f = user.find('.')
1615 if f >= 0:
1602 if f >= 0:
1616 user = user[:f]
1603 user = user[:f]
1617 return user
1604 return user
1618
1605
1619 def ellipsis(text, maxlength=400):
1606 def ellipsis(text, maxlength=400):
1620 """Trim string to at most maxlength (default: 400) characters."""
1607 """Trim string to at most maxlength (default: 400) characters."""
1621 if len(text) <= maxlength:
1608 if len(text) <= maxlength:
1622 return text
1609 return text
1623 else:
1610 else:
1624 return "%s..." % (text[:maxlength-3])
1611 return "%s..." % (text[:maxlength-3])
1625
1612
1626 def walkrepos(path):
1613 def walkrepos(path):
1627 '''yield every hg repository under path, recursively.'''
1614 '''yield every hg repository under path, recursively.'''
1628 def errhandler(err):
1615 def errhandler(err):
1629 if err.filename == path:
1616 if err.filename == path:
1630 raise err
1617 raise err
1631
1618
1632 for root, dirs, files in os.walk(path, onerror=errhandler):
1619 for root, dirs, files in os.walk(path, onerror=errhandler):
1633 for d in dirs:
1620 for d in dirs:
1634 if d == '.hg':
1621 if d == '.hg':
1635 yield root
1622 yield root
1636 dirs[:] = []
1623 dirs[:] = []
1637 break
1624 break
1638
1625
1639 _rcpath = None
1626 _rcpath = None
1640
1627
1641 def os_rcpath():
1628 def os_rcpath():
1642 '''return default os-specific hgrc search path'''
1629 '''return default os-specific hgrc search path'''
1643 path = system_rcpath()
1630 path = system_rcpath()
1644 path.extend(user_rcpath())
1631 path.extend(user_rcpath())
1645 path = [os.path.normpath(f) for f in path]
1632 path = [os.path.normpath(f) for f in path]
1646 return path
1633 return path
1647
1634
1648 def rcpath():
1635 def rcpath():
1649 '''return hgrc search path. if env var HGRCPATH is set, use it.
1636 '''return hgrc search path. if env var HGRCPATH is set, use it.
1650 for each item in path, if directory, use files ending in .rc,
1637 for each item in path, if directory, use files ending in .rc,
1651 else use item.
1638 else use item.
1652 make HGRCPATH empty to only look in .hg/hgrc of current repo.
1639 make HGRCPATH empty to only look in .hg/hgrc of current repo.
1653 if no HGRCPATH, use default os-specific path.'''
1640 if no HGRCPATH, use default os-specific path.'''
1654 global _rcpath
1641 global _rcpath
1655 if _rcpath is None:
1642 if _rcpath is None:
1656 if 'HGRCPATH' in os.environ:
1643 if 'HGRCPATH' in os.environ:
1657 _rcpath = []
1644 _rcpath = []
1658 for p in os.environ['HGRCPATH'].split(os.pathsep):
1645 for p in os.environ['HGRCPATH'].split(os.pathsep):
1659 if not p: continue
1646 if not p: continue
1660 if os.path.isdir(p):
1647 if os.path.isdir(p):
1661 for f, kind in osutil.listdir(p):
1648 for f, kind in osutil.listdir(p):
1662 if f.endswith('.rc'):
1649 if f.endswith('.rc'):
1663 _rcpath.append(os.path.join(p, f))
1650 _rcpath.append(os.path.join(p, f))
1664 else:
1651 else:
1665 _rcpath.append(p)
1652 _rcpath.append(p)
1666 else:
1653 else:
1667 _rcpath = os_rcpath()
1654 _rcpath = os_rcpath()
1668 return _rcpath
1655 return _rcpath
1669
1656
1670 def bytecount(nbytes):
1657 def bytecount(nbytes):
1671 '''return byte count formatted as readable string, with units'''
1658 '''return byte count formatted as readable string, with units'''
1672
1659
1673 units = (
1660 units = (
1674 (100, 1<<30, _('%.0f GB')),
1661 (100, 1<<30, _('%.0f GB')),
1675 (10, 1<<30, _('%.1f GB')),
1662 (10, 1<<30, _('%.1f GB')),
1676 (1, 1<<30, _('%.2f GB')),
1663 (1, 1<<30, _('%.2f GB')),
1677 (100, 1<<20, _('%.0f MB')),
1664 (100, 1<<20, _('%.0f MB')),
1678 (10, 1<<20, _('%.1f MB')),
1665 (10, 1<<20, _('%.1f MB')),
1679 (1, 1<<20, _('%.2f MB')),
1666 (1, 1<<20, _('%.2f MB')),
1680 (100, 1<<10, _('%.0f KB')),
1667 (100, 1<<10, _('%.0f KB')),
1681 (10, 1<<10, _('%.1f KB')),
1668 (10, 1<<10, _('%.1f KB')),
1682 (1, 1<<10, _('%.2f KB')),
1669 (1, 1<<10, _('%.2f KB')),
1683 (1, 1, _('%.0f bytes')),
1670 (1, 1, _('%.0f bytes')),
1684 )
1671 )
1685
1672
1686 for multiplier, divisor, format in units:
1673 for multiplier, divisor, format in units:
1687 if nbytes >= divisor * multiplier:
1674 if nbytes >= divisor * multiplier:
1688 return format % (nbytes / float(divisor))
1675 return format % (nbytes / float(divisor))
1689 return units[-1][2] % nbytes
1676 return units[-1][2] % nbytes
1690
1677
1691 def drop_scheme(scheme, path):
1678 def drop_scheme(scheme, path):
1692 sc = scheme + ':'
1679 sc = scheme + ':'
1693 if path.startswith(sc):
1680 if path.startswith(sc):
1694 path = path[len(sc):]
1681 path = path[len(sc):]
1695 if path.startswith('//'):
1682 if path.startswith('//'):
1696 path = path[2:]
1683 path = path[2:]
1697 return path
1684 return path
1698
1685
1699 def uirepr(s):
1686 def uirepr(s):
1700 # Avoid double backslash in Windows path repr()
1687 # Avoid double backslash in Windows path repr()
1701 return repr(s).replace('\\\\', '\\')
1688 return repr(s).replace('\\\\', '\\')
General Comments 0
You need to be logged in to leave comments. Login now