mirror of https://github.com/python/cpython.git
113 lines
4.3 KiB
Python
113 lines
4.3 KiB
Python
# module getopt -- Standard command line processing.
|
|
|
|
# Function getopt.getopt() has a different interface but provides the
|
|
# same functionality as the Unix getopt() function.
|
|
|
|
# It has two arguments: the first should be argv[1:] (it doesn't want
|
|
# the script name), the second the string of option letters as passed
|
|
# to Unix getopt() (i.e., a string of allowable option letters, with
|
|
# options requiring an argument followed by a colon).
|
|
|
|
# It raises the exception getopt.error with a string argument if it
|
|
# detects an error.
|
|
|
|
# It returns two items:
|
|
# (1) a list of pairs (option, option_argument) giving the options in
|
|
# the order in which they were specified. (I'd use a dictionary
|
|
# but applications may depend on option order or multiple
|
|
# occurrences.) Boolean options have '' as option_argument.
|
|
# (2) the list of remaining arguments (may be empty).
|
|
|
|
# Added by Lars Wirzenius (liw@iki.fi): A third argument is optional.
|
|
# If present, getopt.getopt works similar to the GNU getopt_long
|
|
# function (but optional arguments are not supported). The third
|
|
# argument should be a list of strings that name the long options. If
|
|
# the name ends '=', the argument requires an argument.
|
|
|
|
# (While making this addition, I rewrote the whole thing.)
|
|
|
|
import string
|
|
|
|
error = 'getopt error'
|
|
|
|
def getopt(args, shortopts, longopts = []):
|
|
list = []
|
|
longopts = longopts[:]
|
|
longopts.sort()
|
|
while args and args[0][:1] == '-' and args[0] != '-':
|
|
if args[0] == '-' or args[0] == '--':
|
|
args = args[1:]
|
|
break
|
|
if args[0][:2] == '--':
|
|
list, args = do_longs(list, args[0][2:],
|
|
longopts, args[1:])
|
|
else:
|
|
list, args = do_shorts(list, args[0][1:],
|
|
shortopts, args[1:])
|
|
|
|
return list, args
|
|
|
|
def do_longs(list, opt, longopts, args):
|
|
try:
|
|
i = string.index(opt, '=')
|
|
opt, optarg = opt[:i], opt[i+1:]
|
|
except ValueError:
|
|
optarg = ''
|
|
|
|
has_arg, opt = long_has_args(opt, longopts)
|
|
if has_arg:
|
|
if not optarg:
|
|
if not args:
|
|
raise error, 'option --' + opt + \
|
|
' requires argument'
|
|
optarg, args = args[0], args[1:]
|
|
else:
|
|
if optarg:
|
|
raise error, 'argument --' + opt + \
|
|
' must not have an argument'
|
|
list.append('--' + opt, optarg)
|
|
return list, args
|
|
|
|
# Return:
|
|
# has_arg?
|
|
# full option name
|
|
def long_has_args(opt, longopts):
|
|
optlen = len(opt)
|
|
for i in range(len(longopts)):
|
|
x, y = longopts[i][:optlen], longopts[i][optlen:]
|
|
if opt != x:
|
|
continue
|
|
if y != '' and y != '=' and i+1 < len(longopts):
|
|
if opt == longopts[i+1][:optlen]:
|
|
raise error, 'option --' + opt + \
|
|
' not a unique prefix'
|
|
if longopts[i][-1:] == '=':
|
|
return 1, longopts[i][:-1]
|
|
return 0, longopts[i]
|
|
raise error, 'option --' + opt + ' not recognized'
|
|
|
|
def do_shorts(list, optstring, shortopts, args):
|
|
while optstring != '':
|
|
opt, optstring = optstring[0], optstring[1:]
|
|
if short_has_arg(opt, shortopts):
|
|
if optstring == '':
|
|
if not args:
|
|
raise error, 'option -' + opt + \
|
|
' requires argument'
|
|
optstring, args = args[0], args[1:]
|
|
optarg, optstring = optstring, ''
|
|
else:
|
|
optarg = ''
|
|
list.append('-' + opt, optarg)
|
|
return list, args
|
|
|
|
def short_has_arg(opt, shortopts):
|
|
for i in range(len(shortopts)):
|
|
if opt == shortopts[i] != ':':
|
|
return shortopts[i+1:i+2] == ':'
|
|
raise error, 'option -' + opt + ' not recognized'
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
print getopt(sys.argv[1:], "a:b")
|