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