fancyopts.py
32 lines
| 968 B
| text/x-python
|
PythonLexer
/ mercurial / fancyopts.py
mark.williamson@cl.cam.ac.uk
|
r667 | import getopt | ||
mpm@selenic.com
|
r0 | |||
mpm@selenic.com
|
r596 | def fancyopts(args, options, state): | ||
twaldmann@thinkmo.de
|
r1541 | long = [] | ||
short = '' | ||||
map = {} | ||||
dt = {} | ||||
mpm@selenic.com
|
r0 | |||
for s, l, d, c in options: | ||||
Bryan O'Sullivan
|
r1056 | pl = l.replace('-', '_') | ||
map['-'+s] = map['--'+l] = pl | ||||
state[pl] = d | ||||
dt[pl] = type(d) | ||||
Alexis S. L. Carvalho
|
r3749 | if (d is not None and d is not True and d is not False and | ||
not callable(d)): | ||||
mpm@selenic.com
|
r959 | if s: s += ':' | ||
if l: l += '=' | ||||
mpm@selenic.com
|
r0 | if s: short = short + s | ||
if l: long.append(l) | ||||
mpm@selenic.com
|
r293 | opts, args = getopt.getopt(args, short, long) | ||
mpm@selenic.com
|
r0 | |||
for opt, arg in opts: | ||||
Thomas Arendsen Hein
|
r3673 | if dt[map[opt]] is type(fancyopts): state[map[opt]](state, map[opt], arg) | ||
mpm@selenic.com
|
r0 | elif dt[map[opt]] is type(1): state[map[opt]] = int(arg) | ||
elif dt[map[opt]] is type(''): state[map[opt]] = arg | ||||
elif dt[map[opt]] is type([]): state[map[opt]].append(arg) | ||||
Matt Mackall
|
r3742 | elif dt[map[opt]] is type(None): state[map[opt]] = True | ||
Alexis S. L. Carvalho
|
r3749 | elif dt[map[opt]] is type(False): state[map[opt]] = True | ||
mpm@selenic.com
|
r209 | |||
mpm@selenic.com
|
r0 | return args | ||