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