fancyopts.py
110 lines
| 3.1 KiB
| text/x-python
|
PythonLexer
/ mercurial / fancyopts.py
Martin Geisler
|
r8230 | # fancyopts.py - better command line parsing | ||
# | ||||
# Copyright 2005-2009 Matt Mackall <mpm@selenic.com> and others | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2, incorporated herein by reference. | ||||
mark.williamson@cl.cam.ac.uk
|
r667 | import getopt | ||
mpm@selenic.com
|
r0 | |||
Augie Fackler
|
r7772 | def gnugetopt(args, options, longoptions): | ||
"""Parse options mostly like getopt.gnu_getopt. | ||||
This is different from getopt.gnu_getopt in that an argument of - will | ||||
become an argument of - instead of vanishing completely. | ||||
""" | ||||
extraargs = [] | ||||
if '--' in args: | ||||
stopindex = args.index('--') | ||||
extraargs = args[stopindex+1:] | ||||
args = args[:stopindex] | ||||
opts, parseargs = getopt.getopt(args, options, longoptions) | ||||
args = [] | ||||
while parseargs: | ||||
arg = parseargs.pop(0) | ||||
if arg and arg[0] == '-' and len(arg) > 1: | ||||
parseargs.insert(0, arg) | ||||
topts, newparseargs = getopt.getopt(parseargs, options, longoptions) | ||||
opts = opts + topts | ||||
parseargs = newparseargs | ||||
else: | ||||
args.append(arg) | ||||
args.extend(extraargs) | ||||
return opts, args | ||||
def fancyopts(args, options, state, gnu=False): | ||||
Matt Mackall
|
r5638 | """ | ||
read args, parse options, and store options in state | ||||
each option is a tuple of: | ||||
short option or '' | ||||
long option | ||||
default value | ||||
description | ||||
option types include: | ||||
boolean or none - option sets variable in state to true | ||||
string - parameter string is stored in state | ||||
list - parameter string is added to a list | ||||
integer - parameter strings is stored as int | ||||
function - call function with parameter | ||||
mpm@selenic.com
|
r0 | |||
Matt Mackall
|
r5638 | non-option args are returned | ||
""" | ||||
namelist = [] | ||||
shortlist = '' | ||||
argmap = {} | ||||
defmap = {} | ||||
for short, name, default, comment in options: | ||||
# convert opts to getopt format | ||||
oname = name | ||||
name = name.replace('-', '_') | ||||
argmap['-' + short] = argmap['--' + oname] = name | ||||
defmap[name] = default | ||||
# copy defaults to state | ||||
if isinstance(default, list): | ||||
state[name] = default[:] | ||||
Dirkjan Ochtman
|
r8366 | elif hasattr(default, '__call__'): | ||
Matt Mackall
|
r5638 | state[name] = None | ||
Thomas Arendsen Hein
|
r5093 | else: | ||
Matt Mackall
|
r5638 | state[name] = default | ||
mpm@selenic.com
|
r0 | |||
Matt Mackall
|
r5638 | # does it take a parameter? | ||
if not (default is None or default is True or default is False): | ||||
if short: short += ':' | ||||
if oname: oname += '=' | ||||
if short: | ||||
shortlist += short | ||||
if name: | ||||
namelist.append(oname) | ||||
# parse arguments | ||||
Augie Fackler
|
r7772 | if gnu: | ||
parse = gnugetopt | ||||
else: | ||||
parse = getopt.getopt | ||||
opts, args = parse(args, shortlist, namelist) | ||||
mpm@selenic.com
|
r0 | |||
Matt Mackall
|
r5638 | # transfer result to state | ||
for opt, val in opts: | ||||
name = argmap[opt] | ||||
t = type(defmap[name]) | ||||
if t is type(fancyopts): | ||||
state[name] = defmap[name](val) | ||||
elif t is type(1): | ||||
state[name] = int(val) | ||||
elif t is type(''): | ||||
state[name] = val | ||||
elif t is type([]): | ||||
state[name].append(val) | ||||
elif t is type(None) or t is type(False): | ||||
state[name] = True | ||||
mpm@selenic.com
|
r209 | |||
Matt Mackall
|
r5638 | # return unparsed args | ||
mpm@selenic.com
|
r0 | return args | ||