fancyopts.py
30 lines
| 849 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) | ||||
|
mpm@selenic.com
|
r959 | if not d is None and not callable(d): | ||
| 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) | ||||
| elif dt[map[opt]] is type(None): state[map[opt]] = 1 | ||||
|
mpm@selenic.com
|
r209 | |||
|
mpm@selenic.com
|
r0 | return args | ||