##// END OF EJS Templates
dispatch: avoid double backslashes in error message...
David Golub -
r14914:41c3a71c stable
parent child Browse files
Show More
@@ -1,729 +1,729 b''
1 # dispatch.py - command dispatching for mercurial
1 # dispatch.py - command dispatching for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from i18n import _
8 from i18n import _
9 import os, sys, atexit, signal, pdb, socket, errno, shlex, time, traceback, re
9 import os, sys, atexit, signal, pdb, socket, errno, shlex, time, traceback, re
10 import util, commands, hg, fancyopts, extensions, hook, error
10 import util, commands, hg, fancyopts, extensions, hook, error
11 import cmdutil, encoding
11 import cmdutil, encoding
12 import ui as uimod
12 import ui as uimod
13
13
14 class request(object):
14 class request(object):
15 def __init__(self, args, ui=None, repo=None, fin=None, fout=None, ferr=None):
15 def __init__(self, args, ui=None, repo=None, fin=None, fout=None, ferr=None):
16 self.args = args
16 self.args = args
17 self.ui = ui
17 self.ui = ui
18 self.repo = repo
18 self.repo = repo
19
19
20 # input/output/error streams
20 # input/output/error streams
21 self.fin = fin
21 self.fin = fin
22 self.fout = fout
22 self.fout = fout
23 self.ferr = ferr
23 self.ferr = ferr
24
24
25 def run():
25 def run():
26 "run the command in sys.argv"
26 "run the command in sys.argv"
27 sys.exit(dispatch(request(sys.argv[1:])))
27 sys.exit(dispatch(request(sys.argv[1:])))
28
28
29 def dispatch(req):
29 def dispatch(req):
30 "run the command specified in req.args"
30 "run the command specified in req.args"
31 if req.ferr:
31 if req.ferr:
32 ferr = req.ferr
32 ferr = req.ferr
33 elif req.ui:
33 elif req.ui:
34 ferr = req.ui.ferr
34 ferr = req.ui.ferr
35 else:
35 else:
36 ferr = sys.stderr
36 ferr = sys.stderr
37
37
38 try:
38 try:
39 if not req.ui:
39 if not req.ui:
40 req.ui = uimod.ui()
40 req.ui = uimod.ui()
41 if '--traceback' in req.args:
41 if '--traceback' in req.args:
42 req.ui.setconfig('ui', 'traceback', 'on')
42 req.ui.setconfig('ui', 'traceback', 'on')
43
43
44 # set ui streams from the request
44 # set ui streams from the request
45 if req.fin:
45 if req.fin:
46 req.ui.fin = req.fin
46 req.ui.fin = req.fin
47 if req.fout:
47 if req.fout:
48 req.ui.fout = req.fout
48 req.ui.fout = req.fout
49 if req.ferr:
49 if req.ferr:
50 req.ui.ferr = req.ferr
50 req.ui.ferr = req.ferr
51 except util.Abort, inst:
51 except util.Abort, inst:
52 ferr.write(_("abort: %s\n") % inst)
52 ferr.write(_("abort: %s\n") % inst)
53 if inst.hint:
53 if inst.hint:
54 ferr.write(_("(%s)\n") % inst.hint)
54 ferr.write(_("(%s)\n") % inst.hint)
55 return -1
55 return -1
56 except error.ParseError, inst:
56 except error.ParseError, inst:
57 if len(inst.args) > 1:
57 if len(inst.args) > 1:
58 ferr.write(_("hg: parse error at %s: %s\n") %
58 ferr.write(_("hg: parse error at %s: %s\n") %
59 (inst.args[1], inst.args[0]))
59 (inst.args[1], inst.args[0]))
60 else:
60 else:
61 ferr.write(_("hg: parse error: %s\n") % inst.args[0])
61 ferr.write(_("hg: parse error: %s\n") % inst.args[0])
62 return -1
62 return -1
63
63
64 return _runcatch(req)
64 return _runcatch(req)
65
65
66 def _runcatch(req):
66 def _runcatch(req):
67 def catchterm(*args):
67 def catchterm(*args):
68 raise error.SignalInterrupt
68 raise error.SignalInterrupt
69
69
70 ui = req.ui
70 ui = req.ui
71 try:
71 try:
72 for name in 'SIGBREAK', 'SIGHUP', 'SIGTERM':
72 for name in 'SIGBREAK', 'SIGHUP', 'SIGTERM':
73 num = getattr(signal, name, None)
73 num = getattr(signal, name, None)
74 if num:
74 if num:
75 signal.signal(num, catchterm)
75 signal.signal(num, catchterm)
76 except ValueError:
76 except ValueError:
77 pass # happens if called in a thread
77 pass # happens if called in a thread
78
78
79 try:
79 try:
80 try:
80 try:
81 # enter the debugger before command execution
81 # enter the debugger before command execution
82 if '--debugger' in req.args:
82 if '--debugger' in req.args:
83 ui.warn(_("entering debugger - "
83 ui.warn(_("entering debugger - "
84 "type c to continue starting hg or h for help\n"))
84 "type c to continue starting hg or h for help\n"))
85 pdb.set_trace()
85 pdb.set_trace()
86 try:
86 try:
87 return _dispatch(req)
87 return _dispatch(req)
88 finally:
88 finally:
89 ui.flush()
89 ui.flush()
90 except:
90 except:
91 # enter the debugger when we hit an exception
91 # enter the debugger when we hit an exception
92 if '--debugger' in req.args:
92 if '--debugger' in req.args:
93 traceback.print_exc()
93 traceback.print_exc()
94 pdb.post_mortem(sys.exc_info()[2])
94 pdb.post_mortem(sys.exc_info()[2])
95 ui.traceback()
95 ui.traceback()
96 raise
96 raise
97
97
98 # Global exception handling, alphabetically
98 # Global exception handling, alphabetically
99 # Mercurial-specific first, followed by built-in and library exceptions
99 # Mercurial-specific first, followed by built-in and library exceptions
100 except error.AmbiguousCommand, inst:
100 except error.AmbiguousCommand, inst:
101 ui.warn(_("hg: command '%s' is ambiguous:\n %s\n") %
101 ui.warn(_("hg: command '%s' is ambiguous:\n %s\n") %
102 (inst.args[0], " ".join(inst.args[1])))
102 (inst.args[0], " ".join(inst.args[1])))
103 except error.ParseError, inst:
103 except error.ParseError, inst:
104 if len(inst.args) > 1:
104 if len(inst.args) > 1:
105 ui.warn(_("hg: parse error at %s: %s\n") %
105 ui.warn(_("hg: parse error at %s: %s\n") %
106 (inst.args[1], inst.args[0]))
106 (inst.args[1], inst.args[0]))
107 else:
107 else:
108 ui.warn(_("hg: parse error: %s\n") % inst.args[0])
108 ui.warn(_("hg: parse error: %s\n") % inst.args[0])
109 return -1
109 return -1
110 except error.LockHeld, inst:
110 except error.LockHeld, inst:
111 if inst.errno == errno.ETIMEDOUT:
111 if inst.errno == errno.ETIMEDOUT:
112 reason = _('timed out waiting for lock held by %s') % inst.locker
112 reason = _('timed out waiting for lock held by %s') % inst.locker
113 else:
113 else:
114 reason = _('lock held by %s') % inst.locker
114 reason = _('lock held by %s') % inst.locker
115 ui.warn(_("abort: %s: %s\n") % (inst.desc or inst.filename, reason))
115 ui.warn(_("abort: %s: %s\n") % (inst.desc or inst.filename, reason))
116 except error.LockUnavailable, inst:
116 except error.LockUnavailable, inst:
117 ui.warn(_("abort: could not lock %s: %s\n") %
117 ui.warn(_("abort: could not lock %s: %s\n") %
118 (inst.desc or inst.filename, inst.strerror))
118 (inst.desc or inst.filename, inst.strerror))
119 except error.CommandError, inst:
119 except error.CommandError, inst:
120 if inst.args[0]:
120 if inst.args[0]:
121 ui.warn(_("hg %s: %s\n") % (inst.args[0], inst.args[1]))
121 ui.warn(_("hg %s: %s\n") % (inst.args[0], inst.args[1]))
122 commands.help_(ui, inst.args[0], full=False, command=True)
122 commands.help_(ui, inst.args[0], full=False, command=True)
123 else:
123 else:
124 ui.warn(_("hg: %s\n") % inst.args[1])
124 ui.warn(_("hg: %s\n") % inst.args[1])
125 commands.help_(ui, 'shortlist')
125 commands.help_(ui, 'shortlist')
126 except error.RepoError, inst:
126 except error.RepoError, inst:
127 ui.warn(_("abort: %s!\n") % inst)
127 ui.warn(_("abort: %s!\n") % inst)
128 if inst.hint:
128 if inst.hint:
129 ui.warn(_("(%s)\n") % inst.hint)
129 ui.warn(_("(%s)\n") % inst.hint)
130 except error.ResponseError, inst:
130 except error.ResponseError, inst:
131 ui.warn(_("abort: %s") % inst.args[0])
131 ui.warn(_("abort: %s") % inst.args[0])
132 if not isinstance(inst.args[1], basestring):
132 if not isinstance(inst.args[1], basestring):
133 ui.warn(" %r\n" % (inst.args[1],))
133 ui.warn(" %r\n" % (inst.args[1],))
134 elif not inst.args[1]:
134 elif not inst.args[1]:
135 ui.warn(_(" empty string\n"))
135 ui.warn(_(" empty string\n"))
136 else:
136 else:
137 ui.warn("\n%r\n" % util.ellipsis(inst.args[1]))
137 ui.warn("\n%r\n" % util.ellipsis(inst.args[1]))
138 except error.RevlogError, inst:
138 except error.RevlogError, inst:
139 ui.warn(_("abort: %s!\n") % inst)
139 ui.warn(_("abort: %s!\n") % inst)
140 except error.SignalInterrupt:
140 except error.SignalInterrupt:
141 ui.warn(_("killed!\n"))
141 ui.warn(_("killed!\n"))
142 except error.UnknownCommand, inst:
142 except error.UnknownCommand, inst:
143 ui.warn(_("hg: unknown command '%s'\n") % inst.args[0])
143 ui.warn(_("hg: unknown command '%s'\n") % inst.args[0])
144 try:
144 try:
145 # check if the command is in a disabled extension
145 # check if the command is in a disabled extension
146 # (but don't check for extensions themselves)
146 # (but don't check for extensions themselves)
147 commands.help_(ui, inst.args[0], unknowncmd=True)
147 commands.help_(ui, inst.args[0], unknowncmd=True)
148 except error.UnknownCommand:
148 except error.UnknownCommand:
149 commands.help_(ui, 'shortlist')
149 commands.help_(ui, 'shortlist')
150 except util.Abort, inst:
150 except util.Abort, inst:
151 ui.warn(_("abort: %s\n") % inst)
151 ui.warn(_("abort: %s\n") % inst)
152 if inst.hint:
152 if inst.hint:
153 ui.warn(_("(%s)\n") % inst.hint)
153 ui.warn(_("(%s)\n") % inst.hint)
154 except ImportError, inst:
154 except ImportError, inst:
155 ui.warn(_("abort: %s!\n") % inst)
155 ui.warn(_("abort: %s!\n") % inst)
156 m = str(inst).split()[-1]
156 m = str(inst).split()[-1]
157 if m in "mpatch bdiff".split():
157 if m in "mpatch bdiff".split():
158 ui.warn(_("(did you forget to compile extensions?)\n"))
158 ui.warn(_("(did you forget to compile extensions?)\n"))
159 elif m in "zlib".split():
159 elif m in "zlib".split():
160 ui.warn(_("(is your Python install correct?)\n"))
160 ui.warn(_("(is your Python install correct?)\n"))
161 except IOError, inst:
161 except IOError, inst:
162 if hasattr(inst, "code"):
162 if hasattr(inst, "code"):
163 ui.warn(_("abort: %s\n") % inst)
163 ui.warn(_("abort: %s\n") % inst)
164 elif hasattr(inst, "reason"):
164 elif hasattr(inst, "reason"):
165 try: # usually it is in the form (errno, strerror)
165 try: # usually it is in the form (errno, strerror)
166 reason = inst.reason.args[1]
166 reason = inst.reason.args[1]
167 except (AttributeError, IndexError):
167 except (AttributeError, IndexError):
168 # it might be anything, for example a string
168 # it might be anything, for example a string
169 reason = inst.reason
169 reason = inst.reason
170 ui.warn(_("abort: error: %s\n") % reason)
170 ui.warn(_("abort: error: %s\n") % reason)
171 elif hasattr(inst, "args") and inst.args[0] == errno.EPIPE:
171 elif hasattr(inst, "args") and inst.args[0] == errno.EPIPE:
172 if ui.debugflag:
172 if ui.debugflag:
173 ui.warn(_("broken pipe\n"))
173 ui.warn(_("broken pipe\n"))
174 elif getattr(inst, "strerror", None):
174 elif getattr(inst, "strerror", None):
175 if getattr(inst, "filename", None):
175 if getattr(inst, "filename", None):
176 ui.warn(_("abort: %s: %s\n") % (inst.strerror, inst.filename))
176 ui.warn(_("abort: %s: %s\n") % (inst.strerror, inst.filename))
177 else:
177 else:
178 ui.warn(_("abort: %s\n") % inst.strerror)
178 ui.warn(_("abort: %s\n") % inst.strerror)
179 else:
179 else:
180 raise
180 raise
181 except OSError, inst:
181 except OSError, inst:
182 if getattr(inst, "filename", None):
182 if getattr(inst, "filename", None):
183 ui.warn(_("abort: %s: %s\n") % (inst.strerror, inst.filename))
183 ui.warn(_("abort: %s: %s\n") % (inst.strerror, inst.filename))
184 else:
184 else:
185 ui.warn(_("abort: %s\n") % inst.strerror)
185 ui.warn(_("abort: %s\n") % inst.strerror)
186 except KeyboardInterrupt:
186 except KeyboardInterrupt:
187 try:
187 try:
188 ui.warn(_("interrupted!\n"))
188 ui.warn(_("interrupted!\n"))
189 except IOError, inst:
189 except IOError, inst:
190 if inst.errno == errno.EPIPE:
190 if inst.errno == errno.EPIPE:
191 if ui.debugflag:
191 if ui.debugflag:
192 ui.warn(_("\nbroken pipe\n"))
192 ui.warn(_("\nbroken pipe\n"))
193 else:
193 else:
194 raise
194 raise
195 except MemoryError:
195 except MemoryError:
196 ui.warn(_("abort: out of memory\n"))
196 ui.warn(_("abort: out of memory\n"))
197 except SystemExit, inst:
197 except SystemExit, inst:
198 # Commands shouldn't sys.exit directly, but give a return code.
198 # Commands shouldn't sys.exit directly, but give a return code.
199 # Just in case catch this and and pass exit code to caller.
199 # Just in case catch this and and pass exit code to caller.
200 return inst.code
200 return inst.code
201 except socket.error, inst:
201 except socket.error, inst:
202 ui.warn(_("abort: %s\n") % inst.args[-1])
202 ui.warn(_("abort: %s\n") % inst.args[-1])
203 except:
203 except:
204 ui.warn(_("** unknown exception encountered,"
204 ui.warn(_("** unknown exception encountered,"
205 " please report by visiting\n"))
205 " please report by visiting\n"))
206 ui.warn(_("** http://mercurial.selenic.com/wiki/BugTracker\n"))
206 ui.warn(_("** http://mercurial.selenic.com/wiki/BugTracker\n"))
207 ui.warn(_("** Python %s\n") % sys.version.replace('\n', ''))
207 ui.warn(_("** Python %s\n") % sys.version.replace('\n', ''))
208 ui.warn(_("** Mercurial Distributed SCM (version %s)\n")
208 ui.warn(_("** Mercurial Distributed SCM (version %s)\n")
209 % util.version())
209 % util.version())
210 ui.warn(_("** Extensions loaded: %s\n")
210 ui.warn(_("** Extensions loaded: %s\n")
211 % ", ".join([x[0] for x in extensions.extensions()]))
211 % ", ".join([x[0] for x in extensions.extensions()]))
212 raise
212 raise
213
213
214 return -1
214 return -1
215
215
216 def aliasargs(fn, givenargs):
216 def aliasargs(fn, givenargs):
217 args = getattr(fn, 'args', [])
217 args = getattr(fn, 'args', [])
218 if args and givenargs:
218 if args and givenargs:
219 cmd = ' '.join(map(util.shellquote, args))
219 cmd = ' '.join(map(util.shellquote, args))
220
220
221 nums = []
221 nums = []
222 def replacer(m):
222 def replacer(m):
223 num = int(m.group(1)) - 1
223 num = int(m.group(1)) - 1
224 nums.append(num)
224 nums.append(num)
225 return givenargs[num]
225 return givenargs[num]
226 cmd = re.sub(r'\$(\d+|\$)', replacer, cmd)
226 cmd = re.sub(r'\$(\d+|\$)', replacer, cmd)
227 givenargs = [x for i, x in enumerate(givenargs)
227 givenargs = [x for i, x in enumerate(givenargs)
228 if i not in nums]
228 if i not in nums]
229 args = shlex.split(cmd)
229 args = shlex.split(cmd)
230 return args + givenargs
230 return args + givenargs
231
231
232 class cmdalias(object):
232 class cmdalias(object):
233 def __init__(self, name, definition, cmdtable):
233 def __init__(self, name, definition, cmdtable):
234 self.name = self.cmd = name
234 self.name = self.cmd = name
235 self.cmdname = ''
235 self.cmdname = ''
236 self.definition = definition
236 self.definition = definition
237 self.args = []
237 self.args = []
238 self.opts = []
238 self.opts = []
239 self.help = ''
239 self.help = ''
240 self.norepo = True
240 self.norepo = True
241 self.badalias = False
241 self.badalias = False
242
242
243 try:
243 try:
244 aliases, entry = cmdutil.findcmd(self.name, cmdtable)
244 aliases, entry = cmdutil.findcmd(self.name, cmdtable)
245 for alias, e in cmdtable.iteritems():
245 for alias, e in cmdtable.iteritems():
246 if e is entry:
246 if e is entry:
247 self.cmd = alias
247 self.cmd = alias
248 break
248 break
249 self.shadows = True
249 self.shadows = True
250 except error.UnknownCommand:
250 except error.UnknownCommand:
251 self.shadows = False
251 self.shadows = False
252
252
253 if not self.definition:
253 if not self.definition:
254 def fn(ui, *args):
254 def fn(ui, *args):
255 ui.warn(_("no definition for alias '%s'\n") % self.name)
255 ui.warn(_("no definition for alias '%s'\n") % self.name)
256 return 1
256 return 1
257 self.fn = fn
257 self.fn = fn
258 self.badalias = True
258 self.badalias = True
259
259
260 return
260 return
261
261
262 if self.definition.startswith('!'):
262 if self.definition.startswith('!'):
263 self.shell = True
263 self.shell = True
264 def fn(ui, *args):
264 def fn(ui, *args):
265 env = {'HG_ARGS': ' '.join((self.name,) + args)}
265 env = {'HG_ARGS': ' '.join((self.name,) + args)}
266 def _checkvar(m):
266 def _checkvar(m):
267 if m.groups()[0] == '$':
267 if m.groups()[0] == '$':
268 return m.group()
268 return m.group()
269 elif int(m.groups()[0]) <= len(args):
269 elif int(m.groups()[0]) <= len(args):
270 return m.group()
270 return m.group()
271 else:
271 else:
272 ui.debug("No argument found for substitution "
272 ui.debug("No argument found for substitution "
273 "of %i variable in alias '%s' definition."
273 "of %i variable in alias '%s' definition."
274 % (int(m.groups()[0]), self.name))
274 % (int(m.groups()[0]), self.name))
275 return ''
275 return ''
276 cmd = re.sub(r'\$(\d+|\$)', _checkvar, self.definition[1:])
276 cmd = re.sub(r'\$(\d+|\$)', _checkvar, self.definition[1:])
277 replace = dict((str(i + 1), arg) for i, arg in enumerate(args))
277 replace = dict((str(i + 1), arg) for i, arg in enumerate(args))
278 replace['0'] = self.name
278 replace['0'] = self.name
279 replace['@'] = ' '.join(args)
279 replace['@'] = ' '.join(args)
280 cmd = util.interpolate(r'\$', replace, cmd, escape_prefix=True)
280 cmd = util.interpolate(r'\$', replace, cmd, escape_prefix=True)
281 return util.system(cmd, environ=env, out=ui.fout)
281 return util.system(cmd, environ=env, out=ui.fout)
282 self.fn = fn
282 self.fn = fn
283 return
283 return
284
284
285 args = shlex.split(self.definition)
285 args = shlex.split(self.definition)
286 self.cmdname = cmd = args.pop(0)
286 self.cmdname = cmd = args.pop(0)
287 args = map(util.expandpath, args)
287 args = map(util.expandpath, args)
288
288
289 for invalidarg in ("--cwd", "-R", "--repository", "--repo"):
289 for invalidarg in ("--cwd", "-R", "--repository", "--repo"):
290 if _earlygetopt([invalidarg], args):
290 if _earlygetopt([invalidarg], args):
291 def fn(ui, *args):
291 def fn(ui, *args):
292 ui.warn(_("error in definition for alias '%s': %s may only "
292 ui.warn(_("error in definition for alias '%s': %s may only "
293 "be given on the command line\n")
293 "be given on the command line\n")
294 % (self.name, invalidarg))
294 % (self.name, invalidarg))
295 return 1
295 return 1
296
296
297 self.fn = fn
297 self.fn = fn
298 self.badalias = True
298 self.badalias = True
299 return
299 return
300
300
301 try:
301 try:
302 tableentry = cmdutil.findcmd(cmd, cmdtable, False)[1]
302 tableentry = cmdutil.findcmd(cmd, cmdtable, False)[1]
303 if len(tableentry) > 2:
303 if len(tableentry) > 2:
304 self.fn, self.opts, self.help = tableentry
304 self.fn, self.opts, self.help = tableentry
305 else:
305 else:
306 self.fn, self.opts = tableentry
306 self.fn, self.opts = tableentry
307
307
308 self.args = aliasargs(self.fn, args)
308 self.args = aliasargs(self.fn, args)
309 if cmd not in commands.norepo.split(' '):
309 if cmd not in commands.norepo.split(' '):
310 self.norepo = False
310 self.norepo = False
311 if self.help.startswith("hg " + cmd):
311 if self.help.startswith("hg " + cmd):
312 # drop prefix in old-style help lines so hg shows the alias
312 # drop prefix in old-style help lines so hg shows the alias
313 self.help = self.help[4 + len(cmd):]
313 self.help = self.help[4 + len(cmd):]
314 self.__doc__ = self.fn.__doc__
314 self.__doc__ = self.fn.__doc__
315
315
316 except error.UnknownCommand:
316 except error.UnknownCommand:
317 def fn(ui, *args):
317 def fn(ui, *args):
318 ui.warn(_("alias '%s' resolves to unknown command '%s'\n") \
318 ui.warn(_("alias '%s' resolves to unknown command '%s'\n") \
319 % (self.name, cmd))
319 % (self.name, cmd))
320 try:
320 try:
321 # check if the command is in a disabled extension
321 # check if the command is in a disabled extension
322 commands.help_(ui, cmd, unknowncmd=True)
322 commands.help_(ui, cmd, unknowncmd=True)
323 except error.UnknownCommand:
323 except error.UnknownCommand:
324 pass
324 pass
325 return 1
325 return 1
326 self.fn = fn
326 self.fn = fn
327 self.badalias = True
327 self.badalias = True
328 except error.AmbiguousCommand:
328 except error.AmbiguousCommand:
329 def fn(ui, *args):
329 def fn(ui, *args):
330 ui.warn(_("alias '%s' resolves to ambiguous command '%s'\n") \
330 ui.warn(_("alias '%s' resolves to ambiguous command '%s'\n") \
331 % (self.name, cmd))
331 % (self.name, cmd))
332 return 1
332 return 1
333 self.fn = fn
333 self.fn = fn
334 self.badalias = True
334 self.badalias = True
335
335
336 def __call__(self, ui, *args, **opts):
336 def __call__(self, ui, *args, **opts):
337 if self.shadows:
337 if self.shadows:
338 ui.debug("alias '%s' shadows command '%s'\n" %
338 ui.debug("alias '%s' shadows command '%s'\n" %
339 (self.name, self.cmdname))
339 (self.name, self.cmdname))
340
340
341 if hasattr(self, 'shell'):
341 if hasattr(self, 'shell'):
342 return self.fn(ui, *args, **opts)
342 return self.fn(ui, *args, **opts)
343 else:
343 else:
344 try:
344 try:
345 util.checksignature(self.fn)(ui, *args, **opts)
345 util.checksignature(self.fn)(ui, *args, **opts)
346 except error.SignatureError:
346 except error.SignatureError:
347 args = ' '.join([self.cmdname] + self.args)
347 args = ' '.join([self.cmdname] + self.args)
348 ui.debug("alias '%s' expands to '%s'\n" % (self.name, args))
348 ui.debug("alias '%s' expands to '%s'\n" % (self.name, args))
349 raise
349 raise
350
350
351 def addaliases(ui, cmdtable):
351 def addaliases(ui, cmdtable):
352 # aliases are processed after extensions have been loaded, so they
352 # aliases are processed after extensions have been loaded, so they
353 # may use extension commands. Aliases can also use other alias definitions,
353 # may use extension commands. Aliases can also use other alias definitions,
354 # but only if they have been defined prior to the current definition.
354 # but only if they have been defined prior to the current definition.
355 for alias, definition in ui.configitems('alias'):
355 for alias, definition in ui.configitems('alias'):
356 aliasdef = cmdalias(alias, definition, cmdtable)
356 aliasdef = cmdalias(alias, definition, cmdtable)
357 cmdtable[aliasdef.cmd] = (aliasdef, aliasdef.opts, aliasdef.help)
357 cmdtable[aliasdef.cmd] = (aliasdef, aliasdef.opts, aliasdef.help)
358 if aliasdef.norepo:
358 if aliasdef.norepo:
359 commands.norepo += ' %s' % alias
359 commands.norepo += ' %s' % alias
360
360
361 def _parse(ui, args):
361 def _parse(ui, args):
362 options = {}
362 options = {}
363 cmdoptions = {}
363 cmdoptions = {}
364
364
365 try:
365 try:
366 args = fancyopts.fancyopts(args, commands.globalopts, options)
366 args = fancyopts.fancyopts(args, commands.globalopts, options)
367 except fancyopts.getopt.GetoptError, inst:
367 except fancyopts.getopt.GetoptError, inst:
368 raise error.CommandError(None, inst)
368 raise error.CommandError(None, inst)
369
369
370 if args:
370 if args:
371 cmd, args = args[0], args[1:]
371 cmd, args = args[0], args[1:]
372 aliases, entry = cmdutil.findcmd(cmd, commands.table,
372 aliases, entry = cmdutil.findcmd(cmd, commands.table,
373 ui.config("ui", "strict"))
373 ui.config("ui", "strict"))
374 cmd = aliases[0]
374 cmd = aliases[0]
375 args = aliasargs(entry[0], args)
375 args = aliasargs(entry[0], args)
376 defaults = ui.config("defaults", cmd)
376 defaults = ui.config("defaults", cmd)
377 if defaults:
377 if defaults:
378 args = map(util.expandpath, shlex.split(defaults)) + args
378 args = map(util.expandpath, shlex.split(defaults)) + args
379 c = list(entry[1])
379 c = list(entry[1])
380 else:
380 else:
381 cmd = None
381 cmd = None
382 c = []
382 c = []
383
383
384 # combine global options into local
384 # combine global options into local
385 for o in commands.globalopts:
385 for o in commands.globalopts:
386 c.append((o[0], o[1], options[o[1]], o[3]))
386 c.append((o[0], o[1], options[o[1]], o[3]))
387
387
388 try:
388 try:
389 args = fancyopts.fancyopts(args, c, cmdoptions, True)
389 args = fancyopts.fancyopts(args, c, cmdoptions, True)
390 except fancyopts.getopt.GetoptError, inst:
390 except fancyopts.getopt.GetoptError, inst:
391 raise error.CommandError(cmd, inst)
391 raise error.CommandError(cmd, inst)
392
392
393 # separate global options back out
393 # separate global options back out
394 for o in commands.globalopts:
394 for o in commands.globalopts:
395 n = o[1]
395 n = o[1]
396 options[n] = cmdoptions[n]
396 options[n] = cmdoptions[n]
397 del cmdoptions[n]
397 del cmdoptions[n]
398
398
399 return (cmd, cmd and entry[0] or None, args, options, cmdoptions)
399 return (cmd, cmd and entry[0] or None, args, options, cmdoptions)
400
400
401 def _parseconfig(ui, config):
401 def _parseconfig(ui, config):
402 """parse the --config options from the command line"""
402 """parse the --config options from the command line"""
403 configs = []
403 configs = []
404
404
405 for cfg in config:
405 for cfg in config:
406 try:
406 try:
407 name, value = cfg.split('=', 1)
407 name, value = cfg.split('=', 1)
408 section, name = name.split('.', 1)
408 section, name = name.split('.', 1)
409 if not section or not name:
409 if not section or not name:
410 raise IndexError
410 raise IndexError
411 ui.setconfig(section, name, value)
411 ui.setconfig(section, name, value)
412 configs.append((section, name, value))
412 configs.append((section, name, value))
413 except (IndexError, ValueError):
413 except (IndexError, ValueError):
414 raise util.Abort(_('malformed --config option: %r '
414 raise util.Abort(_('malformed --config option: %r '
415 '(use --config section.name=value)') % cfg)
415 '(use --config section.name=value)') % cfg)
416
416
417 return configs
417 return configs
418
418
419 def _earlygetopt(aliases, args):
419 def _earlygetopt(aliases, args):
420 """Return list of values for an option (or aliases).
420 """Return list of values for an option (or aliases).
421
421
422 The values are listed in the order they appear in args.
422 The values are listed in the order they appear in args.
423 The options and values are removed from args.
423 The options and values are removed from args.
424 """
424 """
425 try:
425 try:
426 argcount = args.index("--")
426 argcount = args.index("--")
427 except ValueError:
427 except ValueError:
428 argcount = len(args)
428 argcount = len(args)
429 shortopts = [opt for opt in aliases if len(opt) == 2]
429 shortopts = [opt for opt in aliases if len(opt) == 2]
430 values = []
430 values = []
431 pos = 0
431 pos = 0
432 while pos < argcount:
432 while pos < argcount:
433 if args[pos] in aliases:
433 if args[pos] in aliases:
434 if pos + 1 >= argcount:
434 if pos + 1 >= argcount:
435 # ignore and let getopt report an error if there is no value
435 # ignore and let getopt report an error if there is no value
436 break
436 break
437 del args[pos]
437 del args[pos]
438 values.append(args.pop(pos))
438 values.append(args.pop(pos))
439 argcount -= 2
439 argcount -= 2
440 elif args[pos][:2] in shortopts:
440 elif args[pos][:2] in shortopts:
441 # short option can have no following space, e.g. hg log -Rfoo
441 # short option can have no following space, e.g. hg log -Rfoo
442 values.append(args.pop(pos)[2:])
442 values.append(args.pop(pos)[2:])
443 argcount -= 1
443 argcount -= 1
444 else:
444 else:
445 pos += 1
445 pos += 1
446 return values
446 return values
447
447
448 def runcommand(lui, repo, cmd, fullargs, ui, options, d, cmdpats, cmdoptions):
448 def runcommand(lui, repo, cmd, fullargs, ui, options, d, cmdpats, cmdoptions):
449 # run pre-hook, and abort if it fails
449 # run pre-hook, and abort if it fails
450 ret = hook.hook(lui, repo, "pre-%s" % cmd, False, args=" ".join(fullargs),
450 ret = hook.hook(lui, repo, "pre-%s" % cmd, False, args=" ".join(fullargs),
451 pats=cmdpats, opts=cmdoptions)
451 pats=cmdpats, opts=cmdoptions)
452 if ret:
452 if ret:
453 return ret
453 return ret
454 ret = _runcommand(ui, options, cmd, d)
454 ret = _runcommand(ui, options, cmd, d)
455 # run post-hook, passing command result
455 # run post-hook, passing command result
456 hook.hook(lui, repo, "post-%s" % cmd, False, args=" ".join(fullargs),
456 hook.hook(lui, repo, "post-%s" % cmd, False, args=" ".join(fullargs),
457 result=ret, pats=cmdpats, opts=cmdoptions)
457 result=ret, pats=cmdpats, opts=cmdoptions)
458 return ret
458 return ret
459
459
460 def _getlocal(ui, rpath):
460 def _getlocal(ui, rpath):
461 """Return (path, local ui object) for the given target path.
461 """Return (path, local ui object) for the given target path.
462
462
463 Takes paths in [cwd]/.hg/hgrc into account."
463 Takes paths in [cwd]/.hg/hgrc into account."
464 """
464 """
465 try:
465 try:
466 wd = os.getcwd()
466 wd = os.getcwd()
467 except OSError, e:
467 except OSError, e:
468 raise util.Abort(_("error getting current working directory: %s") %
468 raise util.Abort(_("error getting current working directory: %s") %
469 e.strerror)
469 e.strerror)
470 path = cmdutil.findrepo(wd) or ""
470 path = cmdutil.findrepo(wd) or ""
471 if not path:
471 if not path:
472 lui = ui
472 lui = ui
473 else:
473 else:
474 lui = ui.copy()
474 lui = ui.copy()
475 lui.readconfig(os.path.join(path, ".hg", "hgrc"), path)
475 lui.readconfig(os.path.join(path, ".hg", "hgrc"), path)
476
476
477 if rpath:
477 if rpath:
478 path = lui.expandpath(rpath[-1])
478 path = lui.expandpath(rpath[-1])
479 lui = ui.copy()
479 lui = ui.copy()
480 lui.readconfig(os.path.join(path, ".hg", "hgrc"), path)
480 lui.readconfig(os.path.join(path, ".hg", "hgrc"), path)
481
481
482 return path, lui
482 return path, lui
483
483
484 def _checkshellalias(ui, args):
484 def _checkshellalias(ui, args):
485 cwd = os.getcwd()
485 cwd = os.getcwd()
486 norepo = commands.norepo
486 norepo = commands.norepo
487 options = {}
487 options = {}
488
488
489 try:
489 try:
490 args = fancyopts.fancyopts(args, commands.globalopts, options)
490 args = fancyopts.fancyopts(args, commands.globalopts, options)
491 except fancyopts.getopt.GetoptError:
491 except fancyopts.getopt.GetoptError:
492 return
492 return
493
493
494 if not args:
494 if not args:
495 return
495 return
496
496
497 _parseconfig(ui, options['config'])
497 _parseconfig(ui, options['config'])
498 if options['cwd']:
498 if options['cwd']:
499 os.chdir(options['cwd'])
499 os.chdir(options['cwd'])
500
500
501 path, lui = _getlocal(ui, [options['repository']])
501 path, lui = _getlocal(ui, [options['repository']])
502
502
503 cmdtable = commands.table.copy()
503 cmdtable = commands.table.copy()
504 addaliases(lui, cmdtable)
504 addaliases(lui, cmdtable)
505
505
506 cmd = args[0]
506 cmd = args[0]
507 try:
507 try:
508 aliases, entry = cmdutil.findcmd(cmd, cmdtable, lui.config("ui", "strict"))
508 aliases, entry = cmdutil.findcmd(cmd, cmdtable, lui.config("ui", "strict"))
509 except (error.AmbiguousCommand, error.UnknownCommand):
509 except (error.AmbiguousCommand, error.UnknownCommand):
510 commands.norepo = norepo
510 commands.norepo = norepo
511 os.chdir(cwd)
511 os.chdir(cwd)
512 return
512 return
513
513
514 cmd = aliases[0]
514 cmd = aliases[0]
515 fn = entry[0]
515 fn = entry[0]
516
516
517 if cmd and hasattr(fn, 'shell'):
517 if cmd and hasattr(fn, 'shell'):
518 d = lambda: fn(ui, *args[1:])
518 d = lambda: fn(ui, *args[1:])
519 return lambda: runcommand(lui, None, cmd, args[:1], ui, options, d, [], {})
519 return lambda: runcommand(lui, None, cmd, args[:1], ui, options, d, [], {})
520
520
521 commands.norepo = norepo
521 commands.norepo = norepo
522 os.chdir(cwd)
522 os.chdir(cwd)
523
523
524 _loaded = set()
524 _loaded = set()
525 def _dispatch(req):
525 def _dispatch(req):
526 args = req.args
526 args = req.args
527 ui = req.ui
527 ui = req.ui
528
528
529 shellaliasfn = _checkshellalias(ui, args)
529 shellaliasfn = _checkshellalias(ui, args)
530 if shellaliasfn:
530 if shellaliasfn:
531 return shellaliasfn()
531 return shellaliasfn()
532
532
533 # read --config before doing anything else
533 # read --config before doing anything else
534 # (e.g. to change trust settings for reading .hg/hgrc)
534 # (e.g. to change trust settings for reading .hg/hgrc)
535 cfgs = _parseconfig(ui, _earlygetopt(['--config'], args))
535 cfgs = _parseconfig(ui, _earlygetopt(['--config'], args))
536
536
537 # check for cwd
537 # check for cwd
538 cwd = _earlygetopt(['--cwd'], args)
538 cwd = _earlygetopt(['--cwd'], args)
539 if cwd:
539 if cwd:
540 os.chdir(cwd[-1])
540 os.chdir(cwd[-1])
541
541
542 rpath = _earlygetopt(["-R", "--repository", "--repo"], args)
542 rpath = _earlygetopt(["-R", "--repository", "--repo"], args)
543 path, lui = _getlocal(ui, rpath)
543 path, lui = _getlocal(ui, rpath)
544
544
545 # Configure extensions in phases: uisetup, extsetup, cmdtable, and
545 # Configure extensions in phases: uisetup, extsetup, cmdtable, and
546 # reposetup. Programs like TortoiseHg will call _dispatch several
546 # reposetup. Programs like TortoiseHg will call _dispatch several
547 # times so we keep track of configured extensions in _loaded.
547 # times so we keep track of configured extensions in _loaded.
548 extensions.loadall(lui)
548 extensions.loadall(lui)
549 exts = [ext for ext in extensions.extensions() if ext[0] not in _loaded]
549 exts = [ext for ext in extensions.extensions() if ext[0] not in _loaded]
550 # Propagate any changes to lui.__class__ by extensions
550 # Propagate any changes to lui.__class__ by extensions
551 ui.__class__ = lui.__class__
551 ui.__class__ = lui.__class__
552
552
553 # (uisetup and extsetup are handled in extensions.loadall)
553 # (uisetup and extsetup are handled in extensions.loadall)
554
554
555 for name, module in exts:
555 for name, module in exts:
556 cmdtable = getattr(module, 'cmdtable', {})
556 cmdtable = getattr(module, 'cmdtable', {})
557 overrides = [cmd for cmd in cmdtable if cmd in commands.table]
557 overrides = [cmd for cmd in cmdtable if cmd in commands.table]
558 if overrides:
558 if overrides:
559 ui.warn(_("extension '%s' overrides commands: %s\n")
559 ui.warn(_("extension '%s' overrides commands: %s\n")
560 % (name, " ".join(overrides)))
560 % (name, " ".join(overrides)))
561 commands.table.update(cmdtable)
561 commands.table.update(cmdtable)
562 _loaded.add(name)
562 _loaded.add(name)
563
563
564 # (reposetup is handled in hg.repository)
564 # (reposetup is handled in hg.repository)
565
565
566 addaliases(lui, commands.table)
566 addaliases(lui, commands.table)
567
567
568 # check for fallback encoding
568 # check for fallback encoding
569 fallback = lui.config('ui', 'fallbackencoding')
569 fallback = lui.config('ui', 'fallbackencoding')
570 if fallback:
570 if fallback:
571 encoding.fallbackencoding = fallback
571 encoding.fallbackencoding = fallback
572
572
573 fullargs = args
573 fullargs = args
574 cmd, func, args, options, cmdoptions = _parse(lui, args)
574 cmd, func, args, options, cmdoptions = _parse(lui, args)
575
575
576 if options["config"]:
576 if options["config"]:
577 raise util.Abort(_("option --config may not be abbreviated!"))
577 raise util.Abort(_("option --config may not be abbreviated!"))
578 if options["cwd"]:
578 if options["cwd"]:
579 raise util.Abort(_("option --cwd may not be abbreviated!"))
579 raise util.Abort(_("option --cwd may not be abbreviated!"))
580 if options["repository"]:
580 if options["repository"]:
581 raise util.Abort(_(
581 raise util.Abort(_(
582 "Option -R has to be separated from other options (e.g. not -qR) "
582 "Option -R has to be separated from other options (e.g. not -qR) "
583 "and --repository may only be abbreviated as --repo!"))
583 "and --repository may only be abbreviated as --repo!"))
584
584
585 if options["encoding"]:
585 if options["encoding"]:
586 encoding.encoding = options["encoding"]
586 encoding.encoding = options["encoding"]
587 if options["encodingmode"]:
587 if options["encodingmode"]:
588 encoding.encodingmode = options["encodingmode"]
588 encoding.encodingmode = options["encodingmode"]
589 if options["time"]:
589 if options["time"]:
590 def get_times():
590 def get_times():
591 t = os.times()
591 t = os.times()
592 if t[4] == 0.0: # Windows leaves this as zero, so use time.clock()
592 if t[4] == 0.0: # Windows leaves this as zero, so use time.clock()
593 t = (t[0], t[1], t[2], t[3], time.clock())
593 t = (t[0], t[1], t[2], t[3], time.clock())
594 return t
594 return t
595 s = get_times()
595 s = get_times()
596 def print_time():
596 def print_time():
597 t = get_times()
597 t = get_times()
598 ui.warn(_("Time: real %.3f secs (user %.3f+%.3f sys %.3f+%.3f)\n") %
598 ui.warn(_("Time: real %.3f secs (user %.3f+%.3f sys %.3f+%.3f)\n") %
599 (t[4]-s[4], t[0]-s[0], t[2]-s[2], t[1]-s[1], t[3]-s[3]))
599 (t[4]-s[4], t[0]-s[0], t[2]-s[2], t[1]-s[1], t[3]-s[3]))
600 atexit.register(print_time)
600 atexit.register(print_time)
601
601
602 uis = set([ui, lui])
602 uis = set([ui, lui])
603
603
604 if req.repo:
604 if req.repo:
605 uis.add(req.repo.ui)
605 uis.add(req.repo.ui)
606
606
607 # copy configs that were passed on the cmdline (--config) to the repo ui
607 # copy configs that were passed on the cmdline (--config) to the repo ui
608 for cfg in cfgs:
608 for cfg in cfgs:
609 req.repo.ui.setconfig(*cfg)
609 req.repo.ui.setconfig(*cfg)
610
610
611 for opt in ('verbose', 'debug', 'quiet', 'traceback'):
611 for opt in ('verbose', 'debug', 'quiet', 'traceback'):
612 val = bool(options[opt])
612 val = bool(options[opt])
613 if val:
613 if val:
614 for ui_ in uis:
614 for ui_ in uis:
615 ui_.setconfig('ui', opt, str(val))
615 ui_.setconfig('ui', opt, str(val))
616
616
617 if options['noninteractive']:
617 if options['noninteractive']:
618 for ui_ in uis:
618 for ui_ in uis:
619 ui_.setconfig('ui', 'interactive', 'off')
619 ui_.setconfig('ui', 'interactive', 'off')
620
620
621 if cmdoptions.get('insecure', False):
621 if cmdoptions.get('insecure', False):
622 for ui_ in uis:
622 for ui_ in uis:
623 ui_.setconfig('web', 'cacerts', '')
623 ui_.setconfig('web', 'cacerts', '')
624
624
625 if options['help']:
625 if options['help']:
626 return commands.help_(ui, cmd, options['version'])
626 return commands.help_(ui, cmd, options['version'])
627 elif options['version']:
627 elif options['version']:
628 return commands.version_(ui)
628 return commands.version_(ui)
629 elif not cmd:
629 elif not cmd:
630 return commands.help_(ui, 'shortlist')
630 return commands.help_(ui, 'shortlist')
631
631
632 repo = None
632 repo = None
633 cmdpats = args[:]
633 cmdpats = args[:]
634 if cmd not in commands.norepo.split():
634 if cmd not in commands.norepo.split():
635 # use the repo from the request only if we don't have -R
635 # use the repo from the request only if we don't have -R
636 if not rpath and not cwd:
636 if not rpath and not cwd:
637 repo = req.repo
637 repo = req.repo
638
638
639 if repo:
639 if repo:
640 # set the descriptors of the repo ui to those of ui
640 # set the descriptors of the repo ui to those of ui
641 repo.ui.fin = ui.fin
641 repo.ui.fin = ui.fin
642 repo.ui.fout = ui.fout
642 repo.ui.fout = ui.fout
643 repo.ui.ferr = ui.ferr
643 repo.ui.ferr = ui.ferr
644 else:
644 else:
645 try:
645 try:
646 repo = hg.repository(ui, path=path)
646 repo = hg.repository(ui, path=path)
647 if not repo.local():
647 if not repo.local():
648 raise util.Abort(_("repository '%s' is not local") % path)
648 raise util.Abort(_("repository '%s' is not local") % path)
649 repo.ui.setconfig("bundle", "mainreporoot", repo.root)
649 repo.ui.setconfig("bundle", "mainreporoot", repo.root)
650 except error.RequirementError:
650 except error.RequirementError:
651 raise
651 raise
652 except error.RepoError:
652 except error.RepoError:
653 if cmd not in commands.optionalrepo.split():
653 if cmd not in commands.optionalrepo.split():
654 if args and not path: # try to infer -R from command args
654 if args and not path: # try to infer -R from command args
655 repos = map(cmdutil.findrepo, args)
655 repos = map(cmdutil.findrepo, args)
656 guess = repos[0]
656 guess = repos[0]
657 if guess and repos.count(guess) == len(repos):
657 if guess and repos.count(guess) == len(repos):
658 req.args = ['--repository', guess] + fullargs
658 req.args = ['--repository', guess] + fullargs
659 return _dispatch(req)
659 return _dispatch(req)
660 if not path:
660 if not path:
661 raise error.RepoError(_("no repository found in %r"
661 raise error.RepoError(_("no repository found in '%s'"
662 " (.hg not found)") % os.getcwd())
662 " (.hg not found)") % os.getcwd())
663 raise
663 raise
664 if repo:
664 if repo:
665 ui = repo.ui
665 ui = repo.ui
666 args.insert(0, repo)
666 args.insert(0, repo)
667 elif rpath:
667 elif rpath:
668 ui.warn(_("warning: --repository ignored\n"))
668 ui.warn(_("warning: --repository ignored\n"))
669
669
670 msg = ' '.join(' ' in a and repr(a) or a for a in fullargs)
670 msg = ' '.join(' ' in a and repr(a) or a for a in fullargs)
671 ui.log("command", msg + "\n")
671 ui.log("command", msg + "\n")
672 d = lambda: util.checksignature(func)(ui, *args, **cmdoptions)
672 d = lambda: util.checksignature(func)(ui, *args, **cmdoptions)
673 try:
673 try:
674 return runcommand(lui, repo, cmd, fullargs, ui, options, d,
674 return runcommand(lui, repo, cmd, fullargs, ui, options, d,
675 cmdpats, cmdoptions)
675 cmdpats, cmdoptions)
676 finally:
676 finally:
677 if repo and repo != req.repo:
677 if repo and repo != req.repo:
678 repo.close()
678 repo.close()
679
679
680 def _runcommand(ui, options, cmd, cmdfunc):
680 def _runcommand(ui, options, cmd, cmdfunc):
681 def checkargs():
681 def checkargs():
682 try:
682 try:
683 return cmdfunc()
683 return cmdfunc()
684 except error.SignatureError:
684 except error.SignatureError:
685 raise error.CommandError(cmd, _("invalid arguments"))
685 raise error.CommandError(cmd, _("invalid arguments"))
686
686
687 if options['profile']:
687 if options['profile']:
688 format = ui.config('profiling', 'format', default='text')
688 format = ui.config('profiling', 'format', default='text')
689
689
690 if not format in ['text', 'kcachegrind']:
690 if not format in ['text', 'kcachegrind']:
691 ui.warn(_("unrecognized profiling format '%s'"
691 ui.warn(_("unrecognized profiling format '%s'"
692 " - Ignored\n") % format)
692 " - Ignored\n") % format)
693 format = 'text'
693 format = 'text'
694
694
695 output = ui.config('profiling', 'output')
695 output = ui.config('profiling', 'output')
696
696
697 if output:
697 if output:
698 path = ui.expandpath(output)
698 path = ui.expandpath(output)
699 ostream = open(path, 'wb')
699 ostream = open(path, 'wb')
700 else:
700 else:
701 ostream = sys.stderr
701 ostream = sys.stderr
702
702
703 try:
703 try:
704 from mercurial import lsprof
704 from mercurial import lsprof
705 except ImportError:
705 except ImportError:
706 raise util.Abort(_(
706 raise util.Abort(_(
707 'lsprof not available - install from '
707 'lsprof not available - install from '
708 'http://codespeak.net/svn/user/arigo/hack/misc/lsprof/'))
708 'http://codespeak.net/svn/user/arigo/hack/misc/lsprof/'))
709 p = lsprof.Profiler()
709 p = lsprof.Profiler()
710 p.enable(subcalls=True)
710 p.enable(subcalls=True)
711 try:
711 try:
712 return checkargs()
712 return checkargs()
713 finally:
713 finally:
714 p.disable()
714 p.disable()
715
715
716 if format == 'kcachegrind':
716 if format == 'kcachegrind':
717 import lsprofcalltree
717 import lsprofcalltree
718 calltree = lsprofcalltree.KCacheGrind(p)
718 calltree = lsprofcalltree.KCacheGrind(p)
719 calltree.output(ostream)
719 calltree.output(ostream)
720 else:
720 else:
721 # format == 'text'
721 # format == 'text'
722 stats = lsprof.Stats(p.getstats())
722 stats = lsprof.Stats(p.getstats())
723 stats.sort()
723 stats.sort()
724 stats.pprint(top=10, file=ostream, climit=5)
724 stats.pprint(top=10, file=ostream, climit=5)
725
725
726 if output:
726 if output:
727 ostream.close()
727 ostream.close()
728 else:
728 else:
729 return checkargs()
729 return checkargs()
@@ -1,24 +1,24 b''
1 Translations are optional:
1 Translations are optional:
2
2
3 $ "$TESTDIR/hghave" gettext || exit 80
3 $ "$TESTDIR/hghave" gettext || exit 80
4
4
5 Test that translations are compiled and installed correctly.
5 Test that translations are compiled and installed correctly.
6
6
7 Default encoding in tests is "ascii" and the translation is encoded
7 Default encoding in tests is "ascii" and the translation is encoded
8 using the "replace" error handler:
8 using the "replace" error handler:
9
9
10 $ LANGUAGE=pt_BR hg tip
10 $ LANGUAGE=pt_BR hg tip
11 abortado: n?o foi encontrado um reposit?rio em '$TESTTMP' (.hg n?o encontrado)!
11 abortado: no repository found in '$TESTTMP' (.hg not found)!
12 [255]
12 [255]
13
13
14 Using a more accomodating encoding:
14 Using a more accomodating encoding:
15
15
16 $ HGENCODING=UTF-8 LANGUAGE=pt_BR hg tip
16 $ HGENCODING=UTF-8 LANGUAGE=pt_BR hg tip
17 abortado: n\xc3\xa3o foi encontrado um reposit\xc3\xb3rio em '$TESTTMP' (.hg n\xc3\xa3o encontrado)! (esc)
17 abortado: no repository found in '$TESTTMP' (.hg not found)!
18 [255]
18 [255]
19
19
20 Different encoding:
20 Different encoding:
21
21
22 $ HGENCODING=Latin-1 LANGUAGE=pt_BR hg tip
22 $ HGENCODING=Latin-1 LANGUAGE=pt_BR hg tip
23 abortado: n\xe3o foi encontrado um reposit\xf3rio em '$TESTTMP' (.hg n\xe3o encontrado)! (esc)
23 abortado: no repository found in '$TESTTMP' (.hg not found)!
24 [255]
24 [255]
General Comments 0
You need to be logged in to leave comments. Login now