annotate onsub.py @ 16:44feac64ecfe

refactoring: disentangled breadth-first and depth-first search; made functions nested
author jakob krainz <jakob@hawo-net.de>
date Tue, 06 Mar 2012 19:59:36 +0100
parents 34aa014b5989
children 5ea3f7533ec5
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6
11fd0da50daa Copyright and license header
Martin Geisler <mg@aragost.com>
parents: 4
diff changeset
1 # onsub.py - execute commands recursively on subrepositories
11fd0da50daa Copyright and license header
Martin Geisler <mg@aragost.com>
parents: 4
diff changeset
2 #
11fd0da50daa Copyright and license header
Martin Geisler <mg@aragost.com>
parents: 4
diff changeset
3 # Copyright 2010, 2011 aragost Trifork
11fd0da50daa Copyright and license header
Martin Geisler <mg@aragost.com>
parents: 4
diff changeset
4 #
11fd0da50daa Copyright and license header
Martin Geisler <mg@aragost.com>
parents: 4
diff changeset
5 # This software may be used and distributed according to the terms of
11fd0da50daa Copyright and license header
Martin Geisler <mg@aragost.com>
parents: 4
diff changeset
6 # the GNU General Public License version 2 or any later version.
0
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
7
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
8 import os
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
9 from mercurial.i18n import _
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
10 from mercurial import extensions, subrepo, util
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
11
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
12 """execute a command in each subrepository"""
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
13
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
14 def onsub(ui, repo, *args, **opts):
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
15 """execute a command in each subrepository
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
16
7
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
17 Executes CMD with the current working directory set to the root of
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
18 each subrepository. By default, execution stops if CMD returns a
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
19 non-zero exit code. Use --ignore-errors to override this.
0
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
20
14
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
21 If a POST-CMD is specified, this will be executed after all
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
22 subrepositories below the current subrepository has been visited.
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
23 This corresponds to a post-order traversal of the tree.
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
24
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
25 It is an error to specify a POST-CMD together with the
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
26 --breadth-first flag.
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
27
7
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
28 Use --verbose/-v to print the command being run and the subrepo
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
29 name for each run of CMD in a subrepo. Alternately, use
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
30 --print0/-0 to print just the subrepo name followed by a NUL
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
31 character instead of a newline. This can be useful in combination
a2ad7553ba79 Improve help text (in particular, clarify meaning of --print0).
Greg Ward <greg@gerg.ca>
parents: 6
diff changeset
32 with :hg:`status --print0`.
0
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
33
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
34 The command has access to the following environment variables:
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
35
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
36 ``HG_REPO``:
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
37 Absolute path to the top-level repository in which the onsub
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
38 command was executed.
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
39
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
40 ``HG_SUBPATH``:
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
41 Relative path to the current subrepository from the top-level
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
42 repository.
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
43
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
44 ``HG_SUBURL``:
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
45 URL for the current subrepository as specified in the
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
46 containing repository's ``.hgsub`` file.
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
47
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
48 ``HG_SUBSTATE``:
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
49 State of the current subrepository as specified in the
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
50 containing repository's ``.hgsubstate`` file.
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
51 """
16
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
52
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
53 # function level "constants" - these won't be modified by the nested functions
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
54 print0 = opts.get('print0')
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
55 if opts.get('ignore_errors'):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
56 onerr = None
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
57 else:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
58 onerr = util.Abort
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
59 maxdepth = opts.get('max_depth')
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
60 precmd = None
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
61 postcmd = None
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
62 includeroot = opts.get('root_repo')
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
63
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
64 def execCmd(sub, cmd):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
65 """if sub == None, cmd is executed inside repo; else, inside sub.
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
66 If cmd == None, do nothing. If cmd == '', do only the print0 (if needed).
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
67 Else, do either print0 or the debugging message, then execute the command.
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
68 """
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
69 if sub == None:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
70 envargdict = dict(HG_SUBPATH='.',
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
71 HG_SUBURL='.',
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
72 HG_SUBSTATE=repo['.'].hex(),
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
73 HG_REPO=repo.root)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
74 relpath = '.'
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
75 cmdwd = repo.root
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
76 else:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
77 # subrepo.relpath was renamed to subrepo.subrelpath in
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
78 # 18b5b6392fcf.
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
79 if hasattr(subrepo, 'relpath'):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
80 relpath = subrepo.relpath(sub)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
81 else:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
82 relpath = subrepo.subrelpath(sub)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
83 envargdict = dict(HG_SUBPATH=relpath,
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
84 HG_SUBURL=sub._path,
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
85 HG_SUBSTATE=sub._state[1],
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
86 HG_REPO=repo.root)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
87 cmdwd = os.path.join(repo.root, relpath)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
88 if cmd != None:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
89 if print0:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
90 ui.write(relpath, "\0")
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
91 if cmd != '':
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
92 if not print0: ui.note(_("executing '%s' in %s\n") % (cmd, relpath))
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
93 util.system(cmd, environ=envargdict, cwd=cmdwd, onerr=onerr,
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
94 errprefix=_('terminated onsub in %s') % relpath)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
95
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
96 def bfs():
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
97 """execute precmd in repo.root and in each subrepository, breadth-first"""
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
98 if includeroot:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
99 execCmd(None, precmd)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
100 ctx = repo['.']
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
101 work = [(1, ctx.sub(subpath)) for subpath in sorted(ctx.substate)]
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
102 while work:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
103 (depth, sub) = work.pop(0)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
104 if depth > maxdepth >= 0:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
105 continue
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
106 execCmd(sub, precmd)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
107 if isinstance(sub, subrepo.hgsubrepo):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
108 rev = sub._state[1]
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
109 ctx = sub._repo[rev]
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
110 w = [(depth + 1, ctx.sub(subpath))
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
111 for subpath in sorted(ctx.substate)]
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
112 work.extend(w)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
113
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
114 def dfs():
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
115 """execute pre-/postcmd in repo.root and in each subrepository, depth-first"""
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
116
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
117 def dfs_rek(depth, sub):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
118 if depth > maxdepth >= 0:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
119 return
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
120 execCmd(sub, precmd)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
121 if isinstance(sub, subrepo.hgsubrepo):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
122 rev = sub._state[1]
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
123 ctx = sub._repo[rev]
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
124 for subpath in sorted(ctx.substate):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
125 dfs_rek(depth+1, ctx.sub(subpath))
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
126 execCmd(sub, postcmd)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
127
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
128 ctx = repo['.']
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
129 work = [ctx.sub(subpath) for subpath in sorted(ctx.substate)]
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
130 if includeroot:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
131 execCmd(None, precmd)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
132 for sub in work:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
133 dfs_rek(1, sub)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
134 if includeroot:
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
135 execCmd(None, postcmd)
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
136
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
137 ### start of main function part ###
12
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
138 if len(args) == 2:
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
139 precmd = args[0]
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
140 postcmd = args[1]
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
141 if opts.get('breadth_first') or opts.get('post_order'):
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
142 raise util.Abort(_("onsub: '-b' and '-p' imply the use of only one command"))
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
143 elif len(args) == 1:
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
144 if opts.get('post_order'):
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
145 precmd = None
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
146 postcmd = args[0]
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
147 else:
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
148 precmd = args[0]
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
149 postcmd = None
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
150 elif len(args) == 0:
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
151 # cmd == '' means only do print0
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
152 if opts.get('post_order'):
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
153 precmd = None
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
154 postcmd = ''
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
155 else:
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
156 precmd = ''
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
157 postcmd = None
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
158 else:
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
159 raise util.Abort(_("onsub: at most 2 command arguments required"))
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
160 if opts.get('post_order') and opts.get('breadth_first'):
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
161 raise util.Abort(_("onsub: '-b' and '-p' are mutually exclusive"))
0
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
162
16
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
163 if opts.get('breadth_first'):
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
164 bfs()
12
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
165 else:
16
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
166 dfs()
44feac64ecfe refactoring: disentangled breadth-first and depth-first search; made functions nested
jakob krainz <jakob@hawo-net.de>
parents: 14
diff changeset
167
0
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
168 cmdtable = {
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
169 "onsub":
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
170 (onsub,
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
171 [('b', 'breadth-first', None,
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
172 _('use breadth-first traversal')),
12
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
173 ('p', 'post-order', None,
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
174 _('use post-order depth-first traversal')),
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
175 ('', 'root-repo', None,
90b54c4fe4fe onsub extension: pre and post order
jakob krainz <jakob@hawo-net.de>
parents: 8
diff changeset
176 _('include root repository in traversal')),
8
ecc4fd16555d Add --max-depth option to limit depth of recursion.
Greg Ward <greg@gerg.ca>
parents: 7
diff changeset
177 ('', 'max-depth', -1,
ecc4fd16555d Add --max-depth option to limit depth of recursion.
Greg Ward <greg@gerg.ca>
parents: 7
diff changeset
178 _('limit recursion to N levels (negative for no limit)'), 'N'),
3
a2184bbf38e6 Implement --ignore-errors.
Martin Geisler <mg@lazybytes.net>
parents: 0
diff changeset
179 ('', 'ignore-errors', None,
a2184bbf38e6 Implement --ignore-errors.
Martin Geisler <mg@lazybytes.net>
parents: 0
diff changeset
180 _('continue execution despite errors')),
0
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
181 ('0', 'print0', None,
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
182 _('end subrepository names with NUL, for use with xargs'))],
14
34aa014b5989 Add help for POST-CMD
Martin Geisler <mg@aragost.com>
parents: 12
diff changeset
183 _('[-b] [-0] [--ignore-errors] CMD [POST-CMD]'))
0
e49f3bbfec4d Initial version.
Martin Geisler <mg@aragost.com>
parents:
diff changeset
184 }