##// END OF EJS Templates
util: delegate seek and tell methods of atomictempfile
Bryan O'Sullivan -
r17237:e7312853 stable
parent child Browse files
Show More
@@ -1,1799 +1,1801
1 1 # util.py - Mercurial utility functions and platform specfic implementations
2 2 #
3 3 # Copyright 2005 K. Thananchayan <thananck@yahoo.com>
4 4 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
5 5 # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
6 6 #
7 7 # This software may be used and distributed according to the terms of the
8 8 # GNU General Public License version 2 or any later version.
9 9
10 10 """Mercurial utility functions and platform specfic implementations.
11 11
12 12 This contains helper routines that are independent of the SCM core and
13 13 hide platform-specific details from the core.
14 14 """
15 15
16 16 from i18n import _
17 17 import error, osutil, encoding, collections
18 18 import errno, re, shutil, sys, tempfile, traceback
19 19 import os, time, datetime, calendar, textwrap, signal
20 20 import imp, socket, urllib
21 21
22 22 if os.name == 'nt':
23 23 import windows as platform
24 24 else:
25 25 import posix as platform
26 26
27 27 cachestat = platform.cachestat
28 28 checkexec = platform.checkexec
29 29 checklink = platform.checklink
30 30 copymode = platform.copymode
31 31 executablepath = platform.executablepath
32 32 expandglobs = platform.expandglobs
33 33 explainexit = platform.explainexit
34 34 findexe = platform.findexe
35 35 gethgcmd = platform.gethgcmd
36 36 getuser = platform.getuser
37 37 groupmembers = platform.groupmembers
38 38 groupname = platform.groupname
39 39 hidewindow = platform.hidewindow
40 40 isexec = platform.isexec
41 41 isowner = platform.isowner
42 42 localpath = platform.localpath
43 43 lookupreg = platform.lookupreg
44 44 makedir = platform.makedir
45 45 nlinks = platform.nlinks
46 46 normpath = platform.normpath
47 47 normcase = platform.normcase
48 48 nulldev = platform.nulldev
49 49 openhardlinks = platform.openhardlinks
50 50 oslink = platform.oslink
51 51 parsepatchoutput = platform.parsepatchoutput
52 52 pconvert = platform.pconvert
53 53 popen = platform.popen
54 54 posixfile = platform.posixfile
55 55 quotecommand = platform.quotecommand
56 56 realpath = platform.realpath
57 57 rename = platform.rename
58 58 samedevice = platform.samedevice
59 59 samefile = platform.samefile
60 60 samestat = platform.samestat
61 61 setbinary = platform.setbinary
62 62 setflags = platform.setflags
63 63 setsignalhandler = platform.setsignalhandler
64 64 shellquote = platform.shellquote
65 65 spawndetached = platform.spawndetached
66 66 sshargs = platform.sshargs
67 67 statfiles = platform.statfiles
68 68 termwidth = platform.termwidth
69 69 testpid = platform.testpid
70 70 umask = platform.umask
71 71 unlink = platform.unlink
72 72 unlinkpath = platform.unlinkpath
73 73 username = platform.username
74 74
75 75 # Python compatibility
76 76
77 77 _notset = object()
78 78
79 79 def safehasattr(thing, attr):
80 80 return getattr(thing, attr, _notset) is not _notset
81 81
82 82 def sha1(s=''):
83 83 '''
84 84 Low-overhead wrapper around Python's SHA support
85 85
86 86 >>> f = _fastsha1
87 87 >>> a = sha1()
88 88 >>> a = f()
89 89 >>> a.hexdigest()
90 90 'da39a3ee5e6b4b0d3255bfef95601890afd80709'
91 91 '''
92 92
93 93 return _fastsha1(s)
94 94
95 95 def _fastsha1(s=''):
96 96 # This function will import sha1 from hashlib or sha (whichever is
97 97 # available) and overwrite itself with it on the first call.
98 98 # Subsequent calls will go directly to the imported function.
99 99 if sys.version_info >= (2, 5):
100 100 from hashlib import sha1 as _sha1
101 101 else:
102 102 from sha import sha as _sha1
103 103 global _fastsha1, sha1
104 104 _fastsha1 = sha1 = _sha1
105 105 return _sha1(s)
106 106
107 107 try:
108 108 buffer = buffer
109 109 except NameError:
110 110 if sys.version_info[0] < 3:
111 111 def buffer(sliceable, offset=0):
112 112 return sliceable[offset:]
113 113 else:
114 114 def buffer(sliceable, offset=0):
115 115 return memoryview(sliceable)[offset:]
116 116
117 117 import subprocess
118 118 closefds = os.name == 'posix'
119 119
120 120 def popen2(cmd, env=None, newlines=False):
121 121 # Setting bufsize to -1 lets the system decide the buffer size.
122 122 # The default for bufsize is 0, meaning unbuffered. This leads to
123 123 # poor performance on Mac OS X: http://bugs.python.org/issue4194
124 124 p = subprocess.Popen(cmd, shell=True, bufsize=-1,
125 125 close_fds=closefds,
126 126 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
127 127 universal_newlines=newlines,
128 128 env=env)
129 129 return p.stdin, p.stdout
130 130
131 131 def popen3(cmd, env=None, newlines=False):
132 132 p = subprocess.Popen(cmd, shell=True, bufsize=-1,
133 133 close_fds=closefds,
134 134 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
135 135 stderr=subprocess.PIPE,
136 136 universal_newlines=newlines,
137 137 env=env)
138 138 return p.stdin, p.stdout, p.stderr
139 139
140 140 def version():
141 141 """Return version information if available."""
142 142 try:
143 143 import __version__
144 144 return __version__.version
145 145 except ImportError:
146 146 return 'unknown'
147 147
148 148 # used by parsedate
149 149 defaultdateformats = (
150 150 '%Y-%m-%d %H:%M:%S',
151 151 '%Y-%m-%d %I:%M:%S%p',
152 152 '%Y-%m-%d %H:%M',
153 153 '%Y-%m-%d %I:%M%p',
154 154 '%Y-%m-%d',
155 155 '%m-%d',
156 156 '%m/%d',
157 157 '%m/%d/%y',
158 158 '%m/%d/%Y',
159 159 '%a %b %d %H:%M:%S %Y',
160 160 '%a %b %d %I:%M:%S%p %Y',
161 161 '%a, %d %b %Y %H:%M:%S', # GNU coreutils "/bin/date --rfc-2822"
162 162 '%b %d %H:%M:%S %Y',
163 163 '%b %d %I:%M:%S%p %Y',
164 164 '%b %d %H:%M:%S',
165 165 '%b %d %I:%M:%S%p',
166 166 '%b %d %H:%M',
167 167 '%b %d %I:%M%p',
168 168 '%b %d %Y',
169 169 '%b %d',
170 170 '%H:%M:%S',
171 171 '%I:%M:%S%p',
172 172 '%H:%M',
173 173 '%I:%M%p',
174 174 )
175 175
176 176 extendeddateformats = defaultdateformats + (
177 177 "%Y",
178 178 "%Y-%m",
179 179 "%b",
180 180 "%b %Y",
181 181 )
182 182
183 183 def cachefunc(func):
184 184 '''cache the result of function calls'''
185 185 # XXX doesn't handle keywords args
186 186 cache = {}
187 187 if func.func_code.co_argcount == 1:
188 188 # we gain a small amount of time because
189 189 # we don't need to pack/unpack the list
190 190 def f(arg):
191 191 if arg not in cache:
192 192 cache[arg] = func(arg)
193 193 return cache[arg]
194 194 else:
195 195 def f(*args):
196 196 if args not in cache:
197 197 cache[args] = func(*args)
198 198 return cache[args]
199 199
200 200 return f
201 201
202 202 try:
203 203 collections.deque.remove
204 204 deque = collections.deque
205 205 except AttributeError:
206 206 # python 2.4 lacks deque.remove
207 207 class deque(collections.deque):
208 208 def remove(self, val):
209 209 for i, v in enumerate(self):
210 210 if v == val:
211 211 del self[i]
212 212 break
213 213
214 214 def lrucachefunc(func):
215 215 '''cache most recent results of function calls'''
216 216 cache = {}
217 217 order = deque()
218 218 if func.func_code.co_argcount == 1:
219 219 def f(arg):
220 220 if arg not in cache:
221 221 if len(cache) > 20:
222 222 del cache[order.popleft()]
223 223 cache[arg] = func(arg)
224 224 else:
225 225 order.remove(arg)
226 226 order.append(arg)
227 227 return cache[arg]
228 228 else:
229 229 def f(*args):
230 230 if args not in cache:
231 231 if len(cache) > 20:
232 232 del cache[order.popleft()]
233 233 cache[args] = func(*args)
234 234 else:
235 235 order.remove(args)
236 236 order.append(args)
237 237 return cache[args]
238 238
239 239 return f
240 240
241 241 class propertycache(object):
242 242 def __init__(self, func):
243 243 self.func = func
244 244 self.name = func.__name__
245 245 def __get__(self, obj, type=None):
246 246 result = self.func(obj)
247 247 setattr(obj, self.name, result)
248 248 return result
249 249
250 250 def pipefilter(s, cmd):
251 251 '''filter string S through command CMD, returning its output'''
252 252 p = subprocess.Popen(cmd, shell=True, close_fds=closefds,
253 253 stdin=subprocess.PIPE, stdout=subprocess.PIPE)
254 254 pout, perr = p.communicate(s)
255 255 return pout
256 256
257 257 def tempfilter(s, cmd):
258 258 '''filter string S through a pair of temporary files with CMD.
259 259 CMD is used as a template to create the real command to be run,
260 260 with the strings INFILE and OUTFILE replaced by the real names of
261 261 the temporary files generated.'''
262 262 inname, outname = None, None
263 263 try:
264 264 infd, inname = tempfile.mkstemp(prefix='hg-filter-in-')
265 265 fp = os.fdopen(infd, 'wb')
266 266 fp.write(s)
267 267 fp.close()
268 268 outfd, outname = tempfile.mkstemp(prefix='hg-filter-out-')
269 269 os.close(outfd)
270 270 cmd = cmd.replace('INFILE', inname)
271 271 cmd = cmd.replace('OUTFILE', outname)
272 272 code = os.system(cmd)
273 273 if sys.platform == 'OpenVMS' and code & 1:
274 274 code = 0
275 275 if code:
276 276 raise Abort(_("command '%s' failed: %s") %
277 277 (cmd, explainexit(code)))
278 278 fp = open(outname, 'rb')
279 279 r = fp.read()
280 280 fp.close()
281 281 return r
282 282 finally:
283 283 try:
284 284 if inname:
285 285 os.unlink(inname)
286 286 except OSError:
287 287 pass
288 288 try:
289 289 if outname:
290 290 os.unlink(outname)
291 291 except OSError:
292 292 pass
293 293
294 294 filtertable = {
295 295 'tempfile:': tempfilter,
296 296 'pipe:': pipefilter,
297 297 }
298 298
299 299 def filter(s, cmd):
300 300 "filter a string through a command that transforms its input to its output"
301 301 for name, fn in filtertable.iteritems():
302 302 if cmd.startswith(name):
303 303 return fn(s, cmd[len(name):].lstrip())
304 304 return pipefilter(s, cmd)
305 305
306 306 def binary(s):
307 307 """return true if a string is binary data"""
308 308 return bool(s and '\0' in s)
309 309
310 310 def increasingchunks(source, min=1024, max=65536):
311 311 '''return no less than min bytes per chunk while data remains,
312 312 doubling min after each chunk until it reaches max'''
313 313 def log2(x):
314 314 if not x:
315 315 return 0
316 316 i = 0
317 317 while x:
318 318 x >>= 1
319 319 i += 1
320 320 return i - 1
321 321
322 322 buf = []
323 323 blen = 0
324 324 for chunk in source:
325 325 buf.append(chunk)
326 326 blen += len(chunk)
327 327 if blen >= min:
328 328 if min < max:
329 329 min = min << 1
330 330 nmin = 1 << log2(blen)
331 331 if nmin > min:
332 332 min = nmin
333 333 if min > max:
334 334 min = max
335 335 yield ''.join(buf)
336 336 blen = 0
337 337 buf = []
338 338 if buf:
339 339 yield ''.join(buf)
340 340
341 341 Abort = error.Abort
342 342
343 343 def always(fn):
344 344 return True
345 345
346 346 def never(fn):
347 347 return False
348 348
349 349 def pathto(root, n1, n2):
350 350 '''return the relative path from one place to another.
351 351 root should use os.sep to separate directories
352 352 n1 should use os.sep to separate directories
353 353 n2 should use "/" to separate directories
354 354 returns an os.sep-separated path.
355 355
356 356 If n1 is a relative path, it's assumed it's
357 357 relative to root.
358 358 n2 should always be relative to root.
359 359 '''
360 360 if not n1:
361 361 return localpath(n2)
362 362 if os.path.isabs(n1):
363 363 if os.path.splitdrive(root)[0] != os.path.splitdrive(n1)[0]:
364 364 return os.path.join(root, localpath(n2))
365 365 n2 = '/'.join((pconvert(root), n2))
366 366 a, b = splitpath(n1), n2.split('/')
367 367 a.reverse()
368 368 b.reverse()
369 369 while a and b and a[-1] == b[-1]:
370 370 a.pop()
371 371 b.pop()
372 372 b.reverse()
373 373 return os.sep.join((['..'] * len(a)) + b) or '.'
374 374
375 375 _hgexecutable = None
376 376
377 377 def mainfrozen():
378 378 """return True if we are a frozen executable.
379 379
380 380 The code supports py2exe (most common, Windows only) and tools/freeze
381 381 (portable, not much used).
382 382 """
383 383 return (safehasattr(sys, "frozen") or # new py2exe
384 384 safehasattr(sys, "importers") or # old py2exe
385 385 imp.is_frozen("__main__")) # tools/freeze
386 386
387 387 def hgexecutable():
388 388 """return location of the 'hg' executable.
389 389
390 390 Defaults to $HG or 'hg' in the search path.
391 391 """
392 392 if _hgexecutable is None:
393 393 hg = os.environ.get('HG')
394 394 mainmod = sys.modules['__main__']
395 395 if hg:
396 396 _sethgexecutable(hg)
397 397 elif mainfrozen():
398 398 _sethgexecutable(sys.executable)
399 399 elif os.path.basename(getattr(mainmod, '__file__', '')) == 'hg':
400 400 _sethgexecutable(mainmod.__file__)
401 401 else:
402 402 exe = findexe('hg') or os.path.basename(sys.argv[0])
403 403 _sethgexecutable(exe)
404 404 return _hgexecutable
405 405
406 406 def _sethgexecutable(path):
407 407 """set location of the 'hg' executable"""
408 408 global _hgexecutable
409 409 _hgexecutable = path
410 410
411 411 def system(cmd, environ={}, cwd=None, onerr=None, errprefix=None, out=None):
412 412 '''enhanced shell command execution.
413 413 run with environment maybe modified, maybe in different dir.
414 414
415 415 if command fails and onerr is None, return status. if ui object,
416 416 print error message and return status, else raise onerr object as
417 417 exception.
418 418
419 419 if out is specified, it is assumed to be a file-like object that has a
420 420 write() method. stdout and stderr will be redirected to out.'''
421 421 try:
422 422 sys.stdout.flush()
423 423 except Exception:
424 424 pass
425 425 def py2shell(val):
426 426 'convert python object into string that is useful to shell'
427 427 if val is None or val is False:
428 428 return '0'
429 429 if val is True:
430 430 return '1'
431 431 return str(val)
432 432 origcmd = cmd
433 433 cmd = quotecommand(cmd)
434 434 if sys.platform == 'plan9':
435 435 # subprocess kludge to work around issues in half-baked Python
436 436 # ports, notably bichued/python:
437 437 if not cwd is None:
438 438 os.chdir(cwd)
439 439 rc = os.system(cmd)
440 440 else:
441 441 env = dict(os.environ)
442 442 env.update((k, py2shell(v)) for k, v in environ.iteritems())
443 443 env['HG'] = hgexecutable()
444 444 if out is None or out == sys.__stdout__:
445 445 rc = subprocess.call(cmd, shell=True, close_fds=closefds,
446 446 env=env, cwd=cwd)
447 447 else:
448 448 proc = subprocess.Popen(cmd, shell=True, close_fds=closefds,
449 449 env=env, cwd=cwd, stdout=subprocess.PIPE,
450 450 stderr=subprocess.STDOUT)
451 451 for line in proc.stdout:
452 452 out.write(line)
453 453 proc.wait()
454 454 rc = proc.returncode
455 455 if sys.platform == 'OpenVMS' and rc & 1:
456 456 rc = 0
457 457 if rc and onerr:
458 458 errmsg = '%s %s' % (os.path.basename(origcmd.split(None, 1)[0]),
459 459 explainexit(rc)[0])
460 460 if errprefix:
461 461 errmsg = '%s: %s' % (errprefix, errmsg)
462 462 try:
463 463 onerr.warn(errmsg + '\n')
464 464 except AttributeError:
465 465 raise onerr(errmsg)
466 466 return rc
467 467
468 468 def checksignature(func):
469 469 '''wrap a function with code to check for calling errors'''
470 470 def check(*args, **kwargs):
471 471 try:
472 472 return func(*args, **kwargs)
473 473 except TypeError:
474 474 if len(traceback.extract_tb(sys.exc_info()[2])) == 1:
475 475 raise error.SignatureError
476 476 raise
477 477
478 478 return check
479 479
480 480 def copyfile(src, dest):
481 481 "copy a file, preserving mode and atime/mtime"
482 482 if os.path.islink(src):
483 483 try:
484 484 os.unlink(dest)
485 485 except OSError:
486 486 pass
487 487 os.symlink(os.readlink(src), dest)
488 488 else:
489 489 try:
490 490 shutil.copyfile(src, dest)
491 491 shutil.copymode(src, dest)
492 492 except shutil.Error, inst:
493 493 raise Abort(str(inst))
494 494
495 495 def copyfiles(src, dst, hardlink=None):
496 496 """Copy a directory tree using hardlinks if possible"""
497 497
498 498 if hardlink is None:
499 499 hardlink = (os.stat(src).st_dev ==
500 500 os.stat(os.path.dirname(dst)).st_dev)
501 501
502 502 num = 0
503 503 if os.path.isdir(src):
504 504 os.mkdir(dst)
505 505 for name, kind in osutil.listdir(src):
506 506 srcname = os.path.join(src, name)
507 507 dstname = os.path.join(dst, name)
508 508 hardlink, n = copyfiles(srcname, dstname, hardlink)
509 509 num += n
510 510 else:
511 511 if hardlink:
512 512 try:
513 513 oslink(src, dst)
514 514 except (IOError, OSError):
515 515 hardlink = False
516 516 shutil.copy(src, dst)
517 517 else:
518 518 shutil.copy(src, dst)
519 519 num += 1
520 520
521 521 return hardlink, num
522 522
523 523 _winreservednames = '''con prn aux nul
524 524 com1 com2 com3 com4 com5 com6 com7 com8 com9
525 525 lpt1 lpt2 lpt3 lpt4 lpt5 lpt6 lpt7 lpt8 lpt9'''.split()
526 526 _winreservedchars = ':*?"<>|'
527 527 def checkwinfilename(path):
528 528 '''Check that the base-relative path is a valid filename on Windows.
529 529 Returns None if the path is ok, or a UI string describing the problem.
530 530
531 531 >>> checkwinfilename("just/a/normal/path")
532 532 >>> checkwinfilename("foo/bar/con.xml")
533 533 "filename contains 'con', which is reserved on Windows"
534 534 >>> checkwinfilename("foo/con.xml/bar")
535 535 "filename contains 'con', which is reserved on Windows"
536 536 >>> checkwinfilename("foo/bar/xml.con")
537 537 >>> checkwinfilename("foo/bar/AUX/bla.txt")
538 538 "filename contains 'AUX', which is reserved on Windows"
539 539 >>> checkwinfilename("foo/bar/bla:.txt")
540 540 "filename contains ':', which is reserved on Windows"
541 541 >>> checkwinfilename("foo/bar/b\07la.txt")
542 542 "filename contains '\\\\x07', which is invalid on Windows"
543 543 >>> checkwinfilename("foo/bar/bla ")
544 544 "filename ends with ' ', which is not allowed on Windows"
545 545 >>> checkwinfilename("../bar")
546 546 '''
547 547 for n in path.replace('\\', '/').split('/'):
548 548 if not n:
549 549 continue
550 550 for c in n:
551 551 if c in _winreservedchars:
552 552 return _("filename contains '%s', which is reserved "
553 553 "on Windows") % c
554 554 if ord(c) <= 31:
555 555 return _("filename contains %r, which is invalid "
556 556 "on Windows") % c
557 557 base = n.split('.')[0]
558 558 if base and base.lower() in _winreservednames:
559 559 return _("filename contains '%s', which is reserved "
560 560 "on Windows") % base
561 561 t = n[-1]
562 562 if t in '. ' and n not in '..':
563 563 return _("filename ends with '%s', which is not allowed "
564 564 "on Windows") % t
565 565
566 566 if os.name == 'nt':
567 567 checkosfilename = checkwinfilename
568 568 else:
569 569 checkosfilename = platform.checkosfilename
570 570
571 571 def makelock(info, pathname):
572 572 try:
573 573 return os.symlink(info, pathname)
574 574 except OSError, why:
575 575 if why.errno == errno.EEXIST:
576 576 raise
577 577 except AttributeError: # no symlink in os
578 578 pass
579 579
580 580 ld = os.open(pathname, os.O_CREAT | os.O_WRONLY | os.O_EXCL)
581 581 os.write(ld, info)
582 582 os.close(ld)
583 583
584 584 def readlock(pathname):
585 585 try:
586 586 return os.readlink(pathname)
587 587 except OSError, why:
588 588 if why.errno not in (errno.EINVAL, errno.ENOSYS):
589 589 raise
590 590 except AttributeError: # no symlink in os
591 591 pass
592 592 fp = posixfile(pathname)
593 593 r = fp.read()
594 594 fp.close()
595 595 return r
596 596
597 597 def fstat(fp):
598 598 '''stat file object that may not have fileno method.'''
599 599 try:
600 600 return os.fstat(fp.fileno())
601 601 except AttributeError:
602 602 return os.stat(fp.name)
603 603
604 604 # File system features
605 605
606 606 def checkcase(path):
607 607 """
608 608 Check whether the given path is on a case-sensitive filesystem
609 609
610 610 Requires a path (like /foo/.hg) ending with a foldable final
611 611 directory component.
612 612 """
613 613 s1 = os.stat(path)
614 614 d, b = os.path.split(path)
615 615 b2 = b.upper()
616 616 if b == b2:
617 617 b2 = b.lower()
618 618 if b == b2:
619 619 return True # no evidence against case sensitivity
620 620 p2 = os.path.join(d, b2)
621 621 try:
622 622 s2 = os.stat(p2)
623 623 if s2 == s1:
624 624 return False
625 625 return True
626 626 except OSError:
627 627 return True
628 628
629 629 try:
630 630 import re2
631 631 _re2 = None
632 632 except ImportError:
633 633 _re2 = False
634 634
635 635 def compilere(pat):
636 636 '''Compile a regular expression, using re2 if possible
637 637
638 638 For best performance, use only re2-compatible regexp features.'''
639 639 global _re2
640 640 if _re2 is None:
641 641 try:
642 642 re2.compile
643 643 _re2 = True
644 644 except ImportError:
645 645 _re2 = False
646 646 if _re2:
647 647 try:
648 648 return re2.compile(pat)
649 649 except re2.error:
650 650 pass
651 651 return re.compile(pat)
652 652
653 653 _fspathcache = {}
654 654 def fspath(name, root):
655 655 '''Get name in the case stored in the filesystem
656 656
657 657 The name should be relative to root, and be normcase-ed for efficiency.
658 658
659 659 Note that this function is unnecessary, and should not be
660 660 called, for case-sensitive filesystems (simply because it's expensive).
661 661
662 662 The root should be normcase-ed, too.
663 663 '''
664 664 def find(p, contents):
665 665 for n in contents:
666 666 if normcase(n) == p:
667 667 return n
668 668 return None
669 669
670 670 seps = os.sep
671 671 if os.altsep:
672 672 seps = seps + os.altsep
673 673 # Protect backslashes. This gets silly very quickly.
674 674 seps.replace('\\','\\\\')
675 675 pattern = re.compile(r'([^%s]+)|([%s]+)' % (seps, seps))
676 676 dir = os.path.normpath(root)
677 677 result = []
678 678 for part, sep in pattern.findall(name):
679 679 if sep:
680 680 result.append(sep)
681 681 continue
682 682
683 683 if dir not in _fspathcache:
684 684 _fspathcache[dir] = os.listdir(dir)
685 685 contents = _fspathcache[dir]
686 686
687 687 found = find(part, contents)
688 688 if not found:
689 689 # retry "once per directory" per "dirstate.walk" which
690 690 # may take place for each patches of "hg qpush", for example
691 691 contents = os.listdir(dir)
692 692 _fspathcache[dir] = contents
693 693 found = find(part, contents)
694 694
695 695 result.append(found or part)
696 696 dir = os.path.join(dir, part)
697 697
698 698 return ''.join(result)
699 699
700 700 def checknlink(testfile):
701 701 '''check whether hardlink count reporting works properly'''
702 702
703 703 # testfile may be open, so we need a separate file for checking to
704 704 # work around issue2543 (or testfile may get lost on Samba shares)
705 705 f1 = testfile + ".hgtmp1"
706 706 if os.path.lexists(f1):
707 707 return False
708 708 try:
709 709 posixfile(f1, 'w').close()
710 710 except IOError:
711 711 return False
712 712
713 713 f2 = testfile + ".hgtmp2"
714 714 fd = None
715 715 try:
716 716 try:
717 717 oslink(f1, f2)
718 718 except OSError:
719 719 return False
720 720
721 721 # nlinks() may behave differently for files on Windows shares if
722 722 # the file is open.
723 723 fd = posixfile(f2)
724 724 return nlinks(f2) > 1
725 725 finally:
726 726 if fd is not None:
727 727 fd.close()
728 728 for f in (f1, f2):
729 729 try:
730 730 os.unlink(f)
731 731 except OSError:
732 732 pass
733 733
734 734 return False
735 735
736 736 def endswithsep(path):
737 737 '''Check path ends with os.sep or os.altsep.'''
738 738 return path.endswith(os.sep) or os.altsep and path.endswith(os.altsep)
739 739
740 740 def splitpath(path):
741 741 '''Split path by os.sep.
742 742 Note that this function does not use os.altsep because this is
743 743 an alternative of simple "xxx.split(os.sep)".
744 744 It is recommended to use os.path.normpath() before using this
745 745 function if need.'''
746 746 return path.split(os.sep)
747 747
748 748 def gui():
749 749 '''Are we running in a GUI?'''
750 750 if sys.platform == 'darwin':
751 751 if 'SSH_CONNECTION' in os.environ:
752 752 # handle SSH access to a box where the user is logged in
753 753 return False
754 754 elif getattr(osutil, 'isgui', None):
755 755 # check if a CoreGraphics session is available
756 756 return osutil.isgui()
757 757 else:
758 758 # pure build; use a safe default
759 759 return True
760 760 else:
761 761 return os.name == "nt" or os.environ.get("DISPLAY")
762 762
763 763 def mktempcopy(name, emptyok=False, createmode=None):
764 764 """Create a temporary file with the same contents from name
765 765
766 766 The permission bits are copied from the original file.
767 767
768 768 If the temporary file is going to be truncated immediately, you
769 769 can use emptyok=True as an optimization.
770 770
771 771 Returns the name of the temporary file.
772 772 """
773 773 d, fn = os.path.split(name)
774 774 fd, temp = tempfile.mkstemp(prefix='.%s-' % fn, dir=d)
775 775 os.close(fd)
776 776 # Temporary files are created with mode 0600, which is usually not
777 777 # what we want. If the original file already exists, just copy
778 778 # its mode. Otherwise, manually obey umask.
779 779 copymode(name, temp, createmode)
780 780 if emptyok:
781 781 return temp
782 782 try:
783 783 try:
784 784 ifp = posixfile(name, "rb")
785 785 except IOError, inst:
786 786 if inst.errno == errno.ENOENT:
787 787 return temp
788 788 if not getattr(inst, 'filename', None):
789 789 inst.filename = name
790 790 raise
791 791 ofp = posixfile(temp, "wb")
792 792 for chunk in filechunkiter(ifp):
793 793 ofp.write(chunk)
794 794 ifp.close()
795 795 ofp.close()
796 796 except: # re-raises
797 797 try: os.unlink(temp)
798 798 except OSError: pass
799 799 raise
800 800 return temp
801 801
802 802 class atomictempfile(object):
803 803 '''writeable file object that atomically updates a file
804 804
805 805 All writes will go to a temporary copy of the original file. Call
806 806 close() when you are done writing, and atomictempfile will rename
807 807 the temporary copy to the original name, making the changes
808 808 visible. If the object is destroyed without being closed, all your
809 809 writes are discarded.
810 810 '''
811 811 def __init__(self, name, mode='w+b', createmode=None):
812 812 self.__name = name # permanent name
813 813 self._tempname = mktempcopy(name, emptyok=('w' in mode),
814 814 createmode=createmode)
815 815 self._fp = posixfile(self._tempname, mode)
816 816
817 817 # delegated methods
818 818 self.write = self._fp.write
819 self.seek = self._fp.seek
820 self.tell = self._fp.tell
819 821 self.fileno = self._fp.fileno
820 822
821 823 def close(self):
822 824 if not self._fp.closed:
823 825 self._fp.close()
824 826 rename(self._tempname, localpath(self.__name))
825 827
826 828 def discard(self):
827 829 if not self._fp.closed:
828 830 try:
829 831 os.unlink(self._tempname)
830 832 except OSError:
831 833 pass
832 834 self._fp.close()
833 835
834 836 def __del__(self):
835 837 if safehasattr(self, '_fp'): # constructor actually did something
836 838 self.discard()
837 839
838 840 def makedirs(name, mode=None):
839 841 """recursive directory creation with parent mode inheritance"""
840 842 try:
841 843 os.mkdir(name)
842 844 except OSError, err:
843 845 if err.errno == errno.EEXIST:
844 846 return
845 847 if err.errno != errno.ENOENT or not name:
846 848 raise
847 849 parent = os.path.dirname(os.path.abspath(name))
848 850 if parent == name:
849 851 raise
850 852 makedirs(parent, mode)
851 853 os.mkdir(name)
852 854 if mode is not None:
853 855 os.chmod(name, mode)
854 856
855 857 def readfile(path):
856 858 fp = open(path, 'rb')
857 859 try:
858 860 return fp.read()
859 861 finally:
860 862 fp.close()
861 863
862 864 def writefile(path, text):
863 865 fp = open(path, 'wb')
864 866 try:
865 867 fp.write(text)
866 868 finally:
867 869 fp.close()
868 870
869 871 def appendfile(path, text):
870 872 fp = open(path, 'ab')
871 873 try:
872 874 fp.write(text)
873 875 finally:
874 876 fp.close()
875 877
876 878 class chunkbuffer(object):
877 879 """Allow arbitrary sized chunks of data to be efficiently read from an
878 880 iterator over chunks of arbitrary size."""
879 881
880 882 def __init__(self, in_iter):
881 883 """in_iter is the iterator that's iterating over the input chunks.
882 884 targetsize is how big a buffer to try to maintain."""
883 885 def splitbig(chunks):
884 886 for chunk in chunks:
885 887 if len(chunk) > 2**20:
886 888 pos = 0
887 889 while pos < len(chunk):
888 890 end = pos + 2 ** 18
889 891 yield chunk[pos:end]
890 892 pos = end
891 893 else:
892 894 yield chunk
893 895 self.iter = splitbig(in_iter)
894 896 self._queue = deque()
895 897
896 898 def read(self, l):
897 899 """Read L bytes of data from the iterator of chunks of data.
898 900 Returns less than L bytes if the iterator runs dry."""
899 901 left = l
900 902 buf = ''
901 903 queue = self._queue
902 904 while left > 0:
903 905 # refill the queue
904 906 if not queue:
905 907 target = 2**18
906 908 for chunk in self.iter:
907 909 queue.append(chunk)
908 910 target -= len(chunk)
909 911 if target <= 0:
910 912 break
911 913 if not queue:
912 914 break
913 915
914 916 chunk = queue.popleft()
915 917 left -= len(chunk)
916 918 if left < 0:
917 919 queue.appendleft(chunk[left:])
918 920 buf += chunk[:left]
919 921 else:
920 922 buf += chunk
921 923
922 924 return buf
923 925
924 926 def filechunkiter(f, size=65536, limit=None):
925 927 """Create a generator that produces the data in the file size
926 928 (default 65536) bytes at a time, up to optional limit (default is
927 929 to read all data). Chunks may be less than size bytes if the
928 930 chunk is the last chunk in the file, or the file is a socket or
929 931 some other type of file that sometimes reads less data than is
930 932 requested."""
931 933 assert size >= 0
932 934 assert limit is None or limit >= 0
933 935 while True:
934 936 if limit is None:
935 937 nbytes = size
936 938 else:
937 939 nbytes = min(limit, size)
938 940 s = nbytes and f.read(nbytes)
939 941 if not s:
940 942 break
941 943 if limit:
942 944 limit -= len(s)
943 945 yield s
944 946
945 947 def makedate():
946 948 ct = time.time()
947 949 if ct < 0:
948 950 hint = _("check your clock")
949 951 raise Abort(_("negative timestamp: %d") % ct, hint=hint)
950 952 delta = (datetime.datetime.utcfromtimestamp(ct) -
951 953 datetime.datetime.fromtimestamp(ct))
952 954 tz = delta.days * 86400 + delta.seconds
953 955 return ct, tz
954 956
955 957 def datestr(date=None, format='%a %b %d %H:%M:%S %Y %1%2'):
956 958 """represent a (unixtime, offset) tuple as a localized time.
957 959 unixtime is seconds since the epoch, and offset is the time zone's
958 960 number of seconds away from UTC. if timezone is false, do not
959 961 append time zone to string."""
960 962 t, tz = date or makedate()
961 963 if t < 0:
962 964 t = 0 # time.gmtime(lt) fails on Windows for lt < -43200
963 965 tz = 0
964 966 if "%1" in format or "%2" in format:
965 967 sign = (tz > 0) and "-" or "+"
966 968 minutes = abs(tz) // 60
967 969 format = format.replace("%1", "%c%02d" % (sign, minutes // 60))
968 970 format = format.replace("%2", "%02d" % (minutes % 60))
969 971 try:
970 972 t = time.gmtime(float(t) - tz)
971 973 except ValueError:
972 974 # time was out of range
973 975 t = time.gmtime(sys.maxint)
974 976 s = time.strftime(format, t)
975 977 return s
976 978
977 979 def shortdate(date=None):
978 980 """turn (timestamp, tzoff) tuple into iso 8631 date."""
979 981 return datestr(date, format='%Y-%m-%d')
980 982
981 983 def strdate(string, format, defaults=[]):
982 984 """parse a localized time string and return a (unixtime, offset) tuple.
983 985 if the string cannot be parsed, ValueError is raised."""
984 986 def timezone(string):
985 987 tz = string.split()[-1]
986 988 if tz[0] in "+-" and len(tz) == 5 and tz[1:].isdigit():
987 989 sign = (tz[0] == "+") and 1 or -1
988 990 hours = int(tz[1:3])
989 991 minutes = int(tz[3:5])
990 992 return -sign * (hours * 60 + minutes) * 60
991 993 if tz == "GMT" or tz == "UTC":
992 994 return 0
993 995 return None
994 996
995 997 # NOTE: unixtime = localunixtime + offset
996 998 offset, date = timezone(string), string
997 999 if offset is not None:
998 1000 date = " ".join(string.split()[:-1])
999 1001
1000 1002 # add missing elements from defaults
1001 1003 usenow = False # default to using biased defaults
1002 1004 for part in ("S", "M", "HI", "d", "mb", "yY"): # decreasing specificity
1003 1005 found = [True for p in part if ("%"+p) in format]
1004 1006 if not found:
1005 1007 date += "@" + defaults[part][usenow]
1006 1008 format += "@%" + part[0]
1007 1009 else:
1008 1010 # We've found a specific time element, less specific time
1009 1011 # elements are relative to today
1010 1012 usenow = True
1011 1013
1012 1014 timetuple = time.strptime(date, format)
1013 1015 localunixtime = int(calendar.timegm(timetuple))
1014 1016 if offset is None:
1015 1017 # local timezone
1016 1018 unixtime = int(time.mktime(timetuple))
1017 1019 offset = unixtime - localunixtime
1018 1020 else:
1019 1021 unixtime = localunixtime + offset
1020 1022 return unixtime, offset
1021 1023
1022 1024 def parsedate(date, formats=None, bias={}):
1023 1025 """parse a localized date/time and return a (unixtime, offset) tuple.
1024 1026
1025 1027 The date may be a "unixtime offset" string or in one of the specified
1026 1028 formats. If the date already is a (unixtime, offset) tuple, it is returned.
1027 1029 """
1028 1030 if not date:
1029 1031 return 0, 0
1030 1032 if isinstance(date, tuple) and len(date) == 2:
1031 1033 return date
1032 1034 if not formats:
1033 1035 formats = defaultdateformats
1034 1036 date = date.strip()
1035 1037 try:
1036 1038 when, offset = map(int, date.split(' '))
1037 1039 except ValueError:
1038 1040 # fill out defaults
1039 1041 now = makedate()
1040 1042 defaults = {}
1041 1043 for part in ("d", "mb", "yY", "HI", "M", "S"):
1042 1044 # this piece is for rounding the specific end of unknowns
1043 1045 b = bias.get(part)
1044 1046 if b is None:
1045 1047 if part[0] in "HMS":
1046 1048 b = "00"
1047 1049 else:
1048 1050 b = "0"
1049 1051
1050 1052 # this piece is for matching the generic end to today's date
1051 1053 n = datestr(now, "%" + part[0])
1052 1054
1053 1055 defaults[part] = (b, n)
1054 1056
1055 1057 for format in formats:
1056 1058 try:
1057 1059 when, offset = strdate(date, format, defaults)
1058 1060 except (ValueError, OverflowError):
1059 1061 pass
1060 1062 else:
1061 1063 break
1062 1064 else:
1063 1065 raise Abort(_('invalid date: %r') % date)
1064 1066 # validate explicit (probably user-specified) date and
1065 1067 # time zone offset. values must fit in signed 32 bits for
1066 1068 # current 32-bit linux runtimes. timezones go from UTC-12
1067 1069 # to UTC+14
1068 1070 if abs(when) > 0x7fffffff:
1069 1071 raise Abort(_('date exceeds 32 bits: %d') % when)
1070 1072 if when < 0:
1071 1073 raise Abort(_('negative date value: %d') % when)
1072 1074 if offset < -50400 or offset > 43200:
1073 1075 raise Abort(_('impossible time zone offset: %d') % offset)
1074 1076 return when, offset
1075 1077
1076 1078 def matchdate(date):
1077 1079 """Return a function that matches a given date match specifier
1078 1080
1079 1081 Formats include:
1080 1082
1081 1083 '{date}' match a given date to the accuracy provided
1082 1084
1083 1085 '<{date}' on or before a given date
1084 1086
1085 1087 '>{date}' on or after a given date
1086 1088
1087 1089 >>> p1 = parsedate("10:29:59")
1088 1090 >>> p2 = parsedate("10:30:00")
1089 1091 >>> p3 = parsedate("10:30:59")
1090 1092 >>> p4 = parsedate("10:31:00")
1091 1093 >>> p5 = parsedate("Sep 15 10:30:00 1999")
1092 1094 >>> f = matchdate("10:30")
1093 1095 >>> f(p1[0])
1094 1096 False
1095 1097 >>> f(p2[0])
1096 1098 True
1097 1099 >>> f(p3[0])
1098 1100 True
1099 1101 >>> f(p4[0])
1100 1102 False
1101 1103 >>> f(p5[0])
1102 1104 False
1103 1105 """
1104 1106
1105 1107 def lower(date):
1106 1108 d = dict(mb="1", d="1")
1107 1109 return parsedate(date, extendeddateformats, d)[0]
1108 1110
1109 1111 def upper(date):
1110 1112 d = dict(mb="12", HI="23", M="59", S="59")
1111 1113 for days in ("31", "30", "29"):
1112 1114 try:
1113 1115 d["d"] = days
1114 1116 return parsedate(date, extendeddateformats, d)[0]
1115 1117 except Abort:
1116 1118 pass
1117 1119 d["d"] = "28"
1118 1120 return parsedate(date, extendeddateformats, d)[0]
1119 1121
1120 1122 date = date.strip()
1121 1123
1122 1124 if not date:
1123 1125 raise Abort(_("dates cannot consist entirely of whitespace"))
1124 1126 elif date[0] == "<":
1125 1127 if not date[1:]:
1126 1128 raise Abort(_("invalid day spec, use '<DATE'"))
1127 1129 when = upper(date[1:])
1128 1130 return lambda x: x <= when
1129 1131 elif date[0] == ">":
1130 1132 if not date[1:]:
1131 1133 raise Abort(_("invalid day spec, use '>DATE'"))
1132 1134 when = lower(date[1:])
1133 1135 return lambda x: x >= when
1134 1136 elif date[0] == "-":
1135 1137 try:
1136 1138 days = int(date[1:])
1137 1139 except ValueError:
1138 1140 raise Abort(_("invalid day spec: %s") % date[1:])
1139 1141 if days < 0:
1140 1142 raise Abort(_("%s must be nonnegative (see 'hg help dates')")
1141 1143 % date[1:])
1142 1144 when = makedate()[0] - days * 3600 * 24
1143 1145 return lambda x: x >= when
1144 1146 elif " to " in date:
1145 1147 a, b = date.split(" to ")
1146 1148 start, stop = lower(a), upper(b)
1147 1149 return lambda x: x >= start and x <= stop
1148 1150 else:
1149 1151 start, stop = lower(date), upper(date)
1150 1152 return lambda x: x >= start and x <= stop
1151 1153
1152 1154 def shortuser(user):
1153 1155 """Return a short representation of a user name or email address."""
1154 1156 f = user.find('@')
1155 1157 if f >= 0:
1156 1158 user = user[:f]
1157 1159 f = user.find('<')
1158 1160 if f >= 0:
1159 1161 user = user[f + 1:]
1160 1162 f = user.find(' ')
1161 1163 if f >= 0:
1162 1164 user = user[:f]
1163 1165 f = user.find('.')
1164 1166 if f >= 0:
1165 1167 user = user[:f]
1166 1168 return user
1167 1169
1168 1170 def emailuser(user):
1169 1171 """Return the user portion of an email address."""
1170 1172 f = user.find('@')
1171 1173 if f >= 0:
1172 1174 user = user[:f]
1173 1175 f = user.find('<')
1174 1176 if f >= 0:
1175 1177 user = user[f + 1:]
1176 1178 return user
1177 1179
1178 1180 def email(author):
1179 1181 '''get email of author.'''
1180 1182 r = author.find('>')
1181 1183 if r == -1:
1182 1184 r = None
1183 1185 return author[author.find('<') + 1:r]
1184 1186
1185 1187 def _ellipsis(text, maxlength):
1186 1188 if len(text) <= maxlength:
1187 1189 return text, False
1188 1190 else:
1189 1191 return "%s..." % (text[:maxlength - 3]), True
1190 1192
1191 1193 def ellipsis(text, maxlength=400):
1192 1194 """Trim string to at most maxlength (default: 400) characters."""
1193 1195 try:
1194 1196 # use unicode not to split at intermediate multi-byte sequence
1195 1197 utext, truncated = _ellipsis(text.decode(encoding.encoding),
1196 1198 maxlength)
1197 1199 if not truncated:
1198 1200 return text
1199 1201 return utext.encode(encoding.encoding)
1200 1202 except (UnicodeDecodeError, UnicodeEncodeError):
1201 1203 return _ellipsis(text, maxlength)[0]
1202 1204
1203 1205 _byteunits = (
1204 1206 (100, 1 << 30, _('%.0f GB')),
1205 1207 (10, 1 << 30, _('%.1f GB')),
1206 1208 (1, 1 << 30, _('%.2f GB')),
1207 1209 (100, 1 << 20, _('%.0f MB')),
1208 1210 (10, 1 << 20, _('%.1f MB')),
1209 1211 (1, 1 << 20, _('%.2f MB')),
1210 1212 (100, 1 << 10, _('%.0f KB')),
1211 1213 (10, 1 << 10, _('%.1f KB')),
1212 1214 (1, 1 << 10, _('%.2f KB')),
1213 1215 (1, 1, _('%.0f bytes')),
1214 1216 )
1215 1217
1216 1218 def bytecount(nbytes):
1217 1219 '''return byte count formatted as readable string, with units'''
1218 1220
1219 1221 for multiplier, divisor, format in _byteunits:
1220 1222 if nbytes >= divisor * multiplier:
1221 1223 return format % (nbytes / float(divisor))
1222 1224 return _byteunits[-1][2] % nbytes
1223 1225
1224 1226 def uirepr(s):
1225 1227 # Avoid double backslash in Windows path repr()
1226 1228 return repr(s).replace('\\\\', '\\')
1227 1229
1228 1230 # delay import of textwrap
1229 1231 def MBTextWrapper(**kwargs):
1230 1232 class tw(textwrap.TextWrapper):
1231 1233 """
1232 1234 Extend TextWrapper for width-awareness.
1233 1235
1234 1236 Neither number of 'bytes' in any encoding nor 'characters' is
1235 1237 appropriate to calculate terminal columns for specified string.
1236 1238
1237 1239 Original TextWrapper implementation uses built-in 'len()' directly,
1238 1240 so overriding is needed to use width information of each characters.
1239 1241
1240 1242 In addition, characters classified into 'ambiguous' width are
1241 1243 treated as wide in east asian area, but as narrow in other.
1242 1244
1243 1245 This requires use decision to determine width of such characters.
1244 1246 """
1245 1247 def __init__(self, **kwargs):
1246 1248 textwrap.TextWrapper.__init__(self, **kwargs)
1247 1249
1248 1250 # for compatibility between 2.4 and 2.6
1249 1251 if getattr(self, 'drop_whitespace', None) is None:
1250 1252 self.drop_whitespace = kwargs.get('drop_whitespace', True)
1251 1253
1252 1254 def _cutdown(self, ucstr, space_left):
1253 1255 l = 0
1254 1256 colwidth = encoding.ucolwidth
1255 1257 for i in xrange(len(ucstr)):
1256 1258 l += colwidth(ucstr[i])
1257 1259 if space_left < l:
1258 1260 return (ucstr[:i], ucstr[i:])
1259 1261 return ucstr, ''
1260 1262
1261 1263 # overriding of base class
1262 1264 def _handle_long_word(self, reversed_chunks, cur_line, cur_len, width):
1263 1265 space_left = max(width - cur_len, 1)
1264 1266
1265 1267 if self.break_long_words:
1266 1268 cut, res = self._cutdown(reversed_chunks[-1], space_left)
1267 1269 cur_line.append(cut)
1268 1270 reversed_chunks[-1] = res
1269 1271 elif not cur_line:
1270 1272 cur_line.append(reversed_chunks.pop())
1271 1273
1272 1274 # this overriding code is imported from TextWrapper of python 2.6
1273 1275 # to calculate columns of string by 'encoding.ucolwidth()'
1274 1276 def _wrap_chunks(self, chunks):
1275 1277 colwidth = encoding.ucolwidth
1276 1278
1277 1279 lines = []
1278 1280 if self.width <= 0:
1279 1281 raise ValueError("invalid width %r (must be > 0)" % self.width)
1280 1282
1281 1283 # Arrange in reverse order so items can be efficiently popped
1282 1284 # from a stack of chucks.
1283 1285 chunks.reverse()
1284 1286
1285 1287 while chunks:
1286 1288
1287 1289 # Start the list of chunks that will make up the current line.
1288 1290 # cur_len is just the length of all the chunks in cur_line.
1289 1291 cur_line = []
1290 1292 cur_len = 0
1291 1293
1292 1294 # Figure out which static string will prefix this line.
1293 1295 if lines:
1294 1296 indent = self.subsequent_indent
1295 1297 else:
1296 1298 indent = self.initial_indent
1297 1299
1298 1300 # Maximum width for this line.
1299 1301 width = self.width - len(indent)
1300 1302
1301 1303 # First chunk on line is whitespace -- drop it, unless this
1302 1304 # is the very beginning of the text (ie. no lines started yet).
1303 1305 if self.drop_whitespace and chunks[-1].strip() == '' and lines:
1304 1306 del chunks[-1]
1305 1307
1306 1308 while chunks:
1307 1309 l = colwidth(chunks[-1])
1308 1310
1309 1311 # Can at least squeeze this chunk onto the current line.
1310 1312 if cur_len + l <= width:
1311 1313 cur_line.append(chunks.pop())
1312 1314 cur_len += l
1313 1315
1314 1316 # Nope, this line is full.
1315 1317 else:
1316 1318 break
1317 1319
1318 1320 # The current line is full, and the next chunk is too big to
1319 1321 # fit on *any* line (not just this one).
1320 1322 if chunks and colwidth(chunks[-1]) > width:
1321 1323 self._handle_long_word(chunks, cur_line, cur_len, width)
1322 1324
1323 1325 # If the last chunk on this line is all whitespace, drop it.
1324 1326 if (self.drop_whitespace and
1325 1327 cur_line and cur_line[-1].strip() == ''):
1326 1328 del cur_line[-1]
1327 1329
1328 1330 # Convert current line back to a string and store it in list
1329 1331 # of all lines (return value).
1330 1332 if cur_line:
1331 1333 lines.append(indent + ''.join(cur_line))
1332 1334
1333 1335 return lines
1334 1336
1335 1337 global MBTextWrapper
1336 1338 MBTextWrapper = tw
1337 1339 return tw(**kwargs)
1338 1340
1339 1341 def wrap(line, width, initindent='', hangindent=''):
1340 1342 maxindent = max(len(hangindent), len(initindent))
1341 1343 if width <= maxindent:
1342 1344 # adjust for weird terminal size
1343 1345 width = max(78, maxindent + 1)
1344 1346 line = line.decode(encoding.encoding, encoding.encodingmode)
1345 1347 initindent = initindent.decode(encoding.encoding, encoding.encodingmode)
1346 1348 hangindent = hangindent.decode(encoding.encoding, encoding.encodingmode)
1347 1349 wrapper = MBTextWrapper(width=width,
1348 1350 initial_indent=initindent,
1349 1351 subsequent_indent=hangindent)
1350 1352 return wrapper.fill(line).encode(encoding.encoding)
1351 1353
1352 1354 def iterlines(iterator):
1353 1355 for chunk in iterator:
1354 1356 for line in chunk.splitlines():
1355 1357 yield line
1356 1358
1357 1359 def expandpath(path):
1358 1360 return os.path.expanduser(os.path.expandvars(path))
1359 1361
1360 1362 def hgcmd():
1361 1363 """Return the command used to execute current hg
1362 1364
1363 1365 This is different from hgexecutable() because on Windows we want
1364 1366 to avoid things opening new shell windows like batch files, so we
1365 1367 get either the python call or current executable.
1366 1368 """
1367 1369 if mainfrozen():
1368 1370 return [sys.executable]
1369 1371 return gethgcmd()
1370 1372
1371 1373 def rundetached(args, condfn):
1372 1374 """Execute the argument list in a detached process.
1373 1375
1374 1376 condfn is a callable which is called repeatedly and should return
1375 1377 True once the child process is known to have started successfully.
1376 1378 At this point, the child process PID is returned. If the child
1377 1379 process fails to start or finishes before condfn() evaluates to
1378 1380 True, return -1.
1379 1381 """
1380 1382 # Windows case is easier because the child process is either
1381 1383 # successfully starting and validating the condition or exiting
1382 1384 # on failure. We just poll on its PID. On Unix, if the child
1383 1385 # process fails to start, it will be left in a zombie state until
1384 1386 # the parent wait on it, which we cannot do since we expect a long
1385 1387 # running process on success. Instead we listen for SIGCHLD telling
1386 1388 # us our child process terminated.
1387 1389 terminated = set()
1388 1390 def handler(signum, frame):
1389 1391 terminated.add(os.wait())
1390 1392 prevhandler = None
1391 1393 SIGCHLD = getattr(signal, 'SIGCHLD', None)
1392 1394 if SIGCHLD is not None:
1393 1395 prevhandler = signal.signal(SIGCHLD, handler)
1394 1396 try:
1395 1397 pid = spawndetached(args)
1396 1398 while not condfn():
1397 1399 if ((pid in terminated or not testpid(pid))
1398 1400 and not condfn()):
1399 1401 return -1
1400 1402 time.sleep(0.1)
1401 1403 return pid
1402 1404 finally:
1403 1405 if prevhandler is not None:
1404 1406 signal.signal(signal.SIGCHLD, prevhandler)
1405 1407
1406 1408 try:
1407 1409 any, all = any, all
1408 1410 except NameError:
1409 1411 def any(iterable):
1410 1412 for i in iterable:
1411 1413 if i:
1412 1414 return True
1413 1415 return False
1414 1416
1415 1417 def all(iterable):
1416 1418 for i in iterable:
1417 1419 if not i:
1418 1420 return False
1419 1421 return True
1420 1422
1421 1423 def interpolate(prefix, mapping, s, fn=None, escape_prefix=False):
1422 1424 """Return the result of interpolating items in the mapping into string s.
1423 1425
1424 1426 prefix is a single character string, or a two character string with
1425 1427 a backslash as the first character if the prefix needs to be escaped in
1426 1428 a regular expression.
1427 1429
1428 1430 fn is an optional function that will be applied to the replacement text
1429 1431 just before replacement.
1430 1432
1431 1433 escape_prefix is an optional flag that allows using doubled prefix for
1432 1434 its escaping.
1433 1435 """
1434 1436 fn = fn or (lambda s: s)
1435 1437 patterns = '|'.join(mapping.keys())
1436 1438 if escape_prefix:
1437 1439 patterns += '|' + prefix
1438 1440 if len(prefix) > 1:
1439 1441 prefix_char = prefix[1:]
1440 1442 else:
1441 1443 prefix_char = prefix
1442 1444 mapping[prefix_char] = prefix_char
1443 1445 r = re.compile(r'%s(%s)' % (prefix, patterns))
1444 1446 return r.sub(lambda x: fn(mapping[x.group()[1:]]), s)
1445 1447
1446 1448 def getport(port):
1447 1449 """Return the port for a given network service.
1448 1450
1449 1451 If port is an integer, it's returned as is. If it's a string, it's
1450 1452 looked up using socket.getservbyname(). If there's no matching
1451 1453 service, util.Abort is raised.
1452 1454 """
1453 1455 try:
1454 1456 return int(port)
1455 1457 except ValueError:
1456 1458 pass
1457 1459
1458 1460 try:
1459 1461 return socket.getservbyname(port)
1460 1462 except socket.error:
1461 1463 raise Abort(_("no port number associated with service '%s'") % port)
1462 1464
1463 1465 _booleans = {'1': True, 'yes': True, 'true': True, 'on': True, 'always': True,
1464 1466 '0': False, 'no': False, 'false': False, 'off': False,
1465 1467 'never': False}
1466 1468
1467 1469 def parsebool(s):
1468 1470 """Parse s into a boolean.
1469 1471
1470 1472 If s is not a valid boolean, returns None.
1471 1473 """
1472 1474 return _booleans.get(s.lower(), None)
1473 1475
1474 1476 _hexdig = '0123456789ABCDEFabcdef'
1475 1477 _hextochr = dict((a + b, chr(int(a + b, 16)))
1476 1478 for a in _hexdig for b in _hexdig)
1477 1479
1478 1480 def _urlunquote(s):
1479 1481 """unquote('abc%20def') -> 'abc def'."""
1480 1482 res = s.split('%')
1481 1483 # fastpath
1482 1484 if len(res) == 1:
1483 1485 return s
1484 1486 s = res[0]
1485 1487 for item in res[1:]:
1486 1488 try:
1487 1489 s += _hextochr[item[:2]] + item[2:]
1488 1490 except KeyError:
1489 1491 s += '%' + item
1490 1492 except UnicodeDecodeError:
1491 1493 s += unichr(int(item[:2], 16)) + item[2:]
1492 1494 return s
1493 1495
1494 1496 class url(object):
1495 1497 r"""Reliable URL parser.
1496 1498
1497 1499 This parses URLs and provides attributes for the following
1498 1500 components:
1499 1501
1500 1502 <scheme>://<user>:<passwd>@<host>:<port>/<path>?<query>#<fragment>
1501 1503
1502 1504 Missing components are set to None. The only exception is
1503 1505 fragment, which is set to '' if present but empty.
1504 1506
1505 1507 If parsefragment is False, fragment is included in query. If
1506 1508 parsequery is False, query is included in path. If both are
1507 1509 False, both fragment and query are included in path.
1508 1510
1509 1511 See http://www.ietf.org/rfc/rfc2396.txt for more information.
1510 1512
1511 1513 Note that for backward compatibility reasons, bundle URLs do not
1512 1514 take host names. That means 'bundle://../' has a path of '../'.
1513 1515
1514 1516 Examples:
1515 1517
1516 1518 >>> url('http://www.ietf.org/rfc/rfc2396.txt')
1517 1519 <url scheme: 'http', host: 'www.ietf.org', path: 'rfc/rfc2396.txt'>
1518 1520 >>> url('ssh://[::1]:2200//home/joe/repo')
1519 1521 <url scheme: 'ssh', host: '[::1]', port: '2200', path: '/home/joe/repo'>
1520 1522 >>> url('file:///home/joe/repo')
1521 1523 <url scheme: 'file', path: '/home/joe/repo'>
1522 1524 >>> url('file:///c:/temp/foo/')
1523 1525 <url scheme: 'file', path: 'c:/temp/foo/'>
1524 1526 >>> url('bundle:foo')
1525 1527 <url scheme: 'bundle', path: 'foo'>
1526 1528 >>> url('bundle://../foo')
1527 1529 <url scheme: 'bundle', path: '../foo'>
1528 1530 >>> url(r'c:\foo\bar')
1529 1531 <url path: 'c:\\foo\\bar'>
1530 1532 >>> url(r'\\blah\blah\blah')
1531 1533 <url path: '\\\\blah\\blah\\blah'>
1532 1534 >>> url(r'\\blah\blah\blah#baz')
1533 1535 <url path: '\\\\blah\\blah\\blah', fragment: 'baz'>
1534 1536
1535 1537 Authentication credentials:
1536 1538
1537 1539 >>> url('ssh://joe:xyz@x/repo')
1538 1540 <url scheme: 'ssh', user: 'joe', passwd: 'xyz', host: 'x', path: 'repo'>
1539 1541 >>> url('ssh://joe@x/repo')
1540 1542 <url scheme: 'ssh', user: 'joe', host: 'x', path: 'repo'>
1541 1543
1542 1544 Query strings and fragments:
1543 1545
1544 1546 >>> url('http://host/a?b#c')
1545 1547 <url scheme: 'http', host: 'host', path: 'a', query: 'b', fragment: 'c'>
1546 1548 >>> url('http://host/a?b#c', parsequery=False, parsefragment=False)
1547 1549 <url scheme: 'http', host: 'host', path: 'a?b#c'>
1548 1550 """
1549 1551
1550 1552 _safechars = "!~*'()+"
1551 1553 _safepchars = "/!~*'()+:"
1552 1554 _matchscheme = re.compile(r'^[a-zA-Z0-9+.\-]+:').match
1553 1555
1554 1556 def __init__(self, path, parsequery=True, parsefragment=True):
1555 1557 # We slowly chomp away at path until we have only the path left
1556 1558 self.scheme = self.user = self.passwd = self.host = None
1557 1559 self.port = self.path = self.query = self.fragment = None
1558 1560 self._localpath = True
1559 1561 self._hostport = ''
1560 1562 self._origpath = path
1561 1563
1562 1564 if parsefragment and '#' in path:
1563 1565 path, self.fragment = path.split('#', 1)
1564 1566 if not path:
1565 1567 path = None
1566 1568
1567 1569 # special case for Windows drive letters and UNC paths
1568 1570 if hasdriveletter(path) or path.startswith(r'\\'):
1569 1571 self.path = path
1570 1572 return
1571 1573
1572 1574 # For compatibility reasons, we can't handle bundle paths as
1573 1575 # normal URLS
1574 1576 if path.startswith('bundle:'):
1575 1577 self.scheme = 'bundle'
1576 1578 path = path[7:]
1577 1579 if path.startswith('//'):
1578 1580 path = path[2:]
1579 1581 self.path = path
1580 1582 return
1581 1583
1582 1584 if self._matchscheme(path):
1583 1585 parts = path.split(':', 1)
1584 1586 if parts[0]:
1585 1587 self.scheme, path = parts
1586 1588 self._localpath = False
1587 1589
1588 1590 if not path:
1589 1591 path = None
1590 1592 if self._localpath:
1591 1593 self.path = ''
1592 1594 return
1593 1595 else:
1594 1596 if self._localpath:
1595 1597 self.path = path
1596 1598 return
1597 1599
1598 1600 if parsequery and '?' in path:
1599 1601 path, self.query = path.split('?', 1)
1600 1602 if not path:
1601 1603 path = None
1602 1604 if not self.query:
1603 1605 self.query = None
1604 1606
1605 1607 # // is required to specify a host/authority
1606 1608 if path and path.startswith('//'):
1607 1609 parts = path[2:].split('/', 1)
1608 1610 if len(parts) > 1:
1609 1611 self.host, path = parts
1610 1612 path = path
1611 1613 else:
1612 1614 self.host = parts[0]
1613 1615 path = None
1614 1616 if not self.host:
1615 1617 self.host = None
1616 1618 # path of file:///d is /d
1617 1619 # path of file:///d:/ is d:/, not /d:/
1618 1620 if path and not hasdriveletter(path):
1619 1621 path = '/' + path
1620 1622
1621 1623 if self.host and '@' in self.host:
1622 1624 self.user, self.host = self.host.rsplit('@', 1)
1623 1625 if ':' in self.user:
1624 1626 self.user, self.passwd = self.user.split(':', 1)
1625 1627 if not self.host:
1626 1628 self.host = None
1627 1629
1628 1630 # Don't split on colons in IPv6 addresses without ports
1629 1631 if (self.host and ':' in self.host and
1630 1632 not (self.host.startswith('[') and self.host.endswith(']'))):
1631 1633 self._hostport = self.host
1632 1634 self.host, self.port = self.host.rsplit(':', 1)
1633 1635 if not self.host:
1634 1636 self.host = None
1635 1637
1636 1638 if (self.host and self.scheme == 'file' and
1637 1639 self.host not in ('localhost', '127.0.0.1', '[::1]')):
1638 1640 raise Abort(_('file:// URLs can only refer to localhost'))
1639 1641
1640 1642 self.path = path
1641 1643
1642 1644 # leave the query string escaped
1643 1645 for a in ('user', 'passwd', 'host', 'port',
1644 1646 'path', 'fragment'):
1645 1647 v = getattr(self, a)
1646 1648 if v is not None:
1647 1649 setattr(self, a, _urlunquote(v))
1648 1650
1649 1651 def __repr__(self):
1650 1652 attrs = []
1651 1653 for a in ('scheme', 'user', 'passwd', 'host', 'port', 'path',
1652 1654 'query', 'fragment'):
1653 1655 v = getattr(self, a)
1654 1656 if v is not None:
1655 1657 attrs.append('%s: %r' % (a, v))
1656 1658 return '<url %s>' % ', '.join(attrs)
1657 1659
1658 1660 def __str__(self):
1659 1661 r"""Join the URL's components back into a URL string.
1660 1662
1661 1663 Examples:
1662 1664
1663 1665 >>> str(url('http://user:pw@host:80/c:/bob?fo:oo#ba:ar'))
1664 1666 'http://user:pw@host:80/c:/bob?fo:oo#ba:ar'
1665 1667 >>> str(url('http://user:pw@host:80/?foo=bar&baz=42'))
1666 1668 'http://user:pw@host:80/?foo=bar&baz=42'
1667 1669 >>> str(url('http://user:pw@host:80/?foo=bar%3dbaz'))
1668 1670 'http://user:pw@host:80/?foo=bar%3dbaz'
1669 1671 >>> str(url('ssh://user:pw@[::1]:2200//home/joe#'))
1670 1672 'ssh://user:pw@[::1]:2200//home/joe#'
1671 1673 >>> str(url('http://localhost:80//'))
1672 1674 'http://localhost:80//'
1673 1675 >>> str(url('http://localhost:80/'))
1674 1676 'http://localhost:80/'
1675 1677 >>> str(url('http://localhost:80'))
1676 1678 'http://localhost:80/'
1677 1679 >>> str(url('bundle:foo'))
1678 1680 'bundle:foo'
1679 1681 >>> str(url('bundle://../foo'))
1680 1682 'bundle:../foo'
1681 1683 >>> str(url('path'))
1682 1684 'path'
1683 1685 >>> str(url('file:///tmp/foo/bar'))
1684 1686 'file:///tmp/foo/bar'
1685 1687 >>> str(url('file:///c:/tmp/foo/bar'))
1686 1688 'file:///c:/tmp/foo/bar'
1687 1689 >>> print url(r'bundle:foo\bar')
1688 1690 bundle:foo\bar
1689 1691 """
1690 1692 if self._localpath:
1691 1693 s = self.path
1692 1694 if self.scheme == 'bundle':
1693 1695 s = 'bundle:' + s
1694 1696 if self.fragment:
1695 1697 s += '#' + self.fragment
1696 1698 return s
1697 1699
1698 1700 s = self.scheme + ':'
1699 1701 if self.user or self.passwd or self.host:
1700 1702 s += '//'
1701 1703 elif self.scheme and (not self.path or self.path.startswith('/')
1702 1704 or hasdriveletter(self.path)):
1703 1705 s += '//'
1704 1706 if hasdriveletter(self.path):
1705 1707 s += '/'
1706 1708 if self.user:
1707 1709 s += urllib.quote(self.user, safe=self._safechars)
1708 1710 if self.passwd:
1709 1711 s += ':' + urllib.quote(self.passwd, safe=self._safechars)
1710 1712 if self.user or self.passwd:
1711 1713 s += '@'
1712 1714 if self.host:
1713 1715 if not (self.host.startswith('[') and self.host.endswith(']')):
1714 1716 s += urllib.quote(self.host)
1715 1717 else:
1716 1718 s += self.host
1717 1719 if self.port:
1718 1720 s += ':' + urllib.quote(self.port)
1719 1721 if self.host:
1720 1722 s += '/'
1721 1723 if self.path:
1722 1724 # TODO: similar to the query string, we should not unescape the
1723 1725 # path when we store it, the path might contain '%2f' = '/',
1724 1726 # which we should *not* escape.
1725 1727 s += urllib.quote(self.path, safe=self._safepchars)
1726 1728 if self.query:
1727 1729 # we store the query in escaped form.
1728 1730 s += '?' + self.query
1729 1731 if self.fragment is not None:
1730 1732 s += '#' + urllib.quote(self.fragment, safe=self._safepchars)
1731 1733 return s
1732 1734
1733 1735 def authinfo(self):
1734 1736 user, passwd = self.user, self.passwd
1735 1737 try:
1736 1738 self.user, self.passwd = None, None
1737 1739 s = str(self)
1738 1740 finally:
1739 1741 self.user, self.passwd = user, passwd
1740 1742 if not self.user:
1741 1743 return (s, None)
1742 1744 # authinfo[1] is passed to urllib2 password manager, and its
1743 1745 # URIs must not contain credentials. The host is passed in the
1744 1746 # URIs list because Python < 2.4.3 uses only that to search for
1745 1747 # a password.
1746 1748 return (s, (None, (s, self.host),
1747 1749 self.user, self.passwd or ''))
1748 1750
1749 1751 def isabs(self):
1750 1752 if self.scheme and self.scheme != 'file':
1751 1753 return True # remote URL
1752 1754 if hasdriveletter(self.path):
1753 1755 return True # absolute for our purposes - can't be joined()
1754 1756 if self.path.startswith(r'\\'):
1755 1757 return True # Windows UNC path
1756 1758 if self.path.startswith('/'):
1757 1759 return True # POSIX-style
1758 1760 return False
1759 1761
1760 1762 def localpath(self):
1761 1763 if self.scheme == 'file' or self.scheme == 'bundle':
1762 1764 path = self.path or '/'
1763 1765 # For Windows, we need to promote hosts containing drive
1764 1766 # letters to paths with drive letters.
1765 1767 if hasdriveletter(self._hostport):
1766 1768 path = self._hostport + '/' + self.path
1767 1769 elif (self.host is not None and self.path
1768 1770 and not hasdriveletter(path)):
1769 1771 path = '/' + path
1770 1772 return path
1771 1773 return self._origpath
1772 1774
1773 1775 def hasscheme(path):
1774 1776 return bool(url(path).scheme)
1775 1777
1776 1778 def hasdriveletter(path):
1777 1779 return path and path[1:2] == ':' and path[0:1].isalpha()
1778 1780
1779 1781 def urllocalpath(path):
1780 1782 return url(path, parsequery=False, parsefragment=False).localpath()
1781 1783
1782 1784 def hidepassword(u):
1783 1785 '''hide user credential in a url string'''
1784 1786 u = url(u)
1785 1787 if u.passwd:
1786 1788 u.passwd = '***'
1787 1789 return str(u)
1788 1790
1789 1791 def removeauth(u):
1790 1792 '''remove all authentication information from a url string'''
1791 1793 u = url(u)
1792 1794 u.user = u.passwd = None
1793 1795 return str(u)
1794 1796
1795 1797 def isatty(fd):
1796 1798 try:
1797 1799 return fd.isatty()
1798 1800 except AttributeError:
1799 1801 return False
General Comments 0
You need to be logged in to leave comments. Login now