我有一段代码来处理命令行参数.
def parse_cmd_args():
input_path = None
output_name = 'out.flv.txt'
is_detail = False
try:
opts, args = getopt.getopt(sys.argv[1:], "hi:o:d")
except getopt.GetoptError:
print 'Usage:'
print 'parse_flv -i input_path -o [output_name]'
sys.exit()
for op, value in opts:
if op == '-i':
input_path = value
elif op == '-o':
output_name = value
elif op == '-d':
is_detail = True
elif op == '-h':
print 'Usage:'
print 'parse_flv -i input_path [-o output_name]'
sys.exit()
return os.path.abspath(input_path), output_name, is_detail
如果我输入一个没有选项符号’ – ‘的命令,如下所示:
python parse_flv.py s
它会引发错误.
我的问题:
如何使用getopt模块处理没有’-i’之类的选项的参数.谢谢
解决方法:
您应该考虑使用argparse模块. getopt是有限的……
这个模块更方便(代码更少,信息量更多,帮助和错误消息更多).在你的情况下,这将简单地类似于:
parser = argparse.ArgumentParser(add_help=True)
parser.add_argument('infile', nargs=1, help='input file name')
parser.add_argument('outfile', nargs='?', help='output file name')
在此示例中,outfile是可选的,您可以指定默认输出文件名:
parser.add_argument('outfile', nargs='?', help='output file name', default='out.txt')
关于getopt和argparse here的更多细节(相互比较).
编辑:
这是getopt可以做的最好的事情(据我所知),即使用gnu_getopt切换到GNU模式:
import getopt
import sys
output_name = 'out.txt'
input_name = ''
print 'ARGV :', sys.argv[1:]
options, remainder = getopt.gnu_getopt(sys.argv[1:], 'o:', ['input-path',
'output-name=',
])
print 'OPTIONS :', options
for opt, arg in options:
if opt in ('-o', '--output-name'):
output_name = arg
else:
pass
# Get input name by yourself...
input_name = remainder[0]
print 'OUTPUTNAME :', output_name
print 'REMAINING :', remainder
print 'INPUTNAME :', input_name
呼叫:
python parse_flv.py input -o output
要么
python parse_flv.py -o output input
输出:
ARGV : ['-o', 'output', 'input']
OPTIONS : [('-o', 'output')]
OUTPUTNAME : output
REMAINING : ['input']
INPUTNAME : input
这将证实你必须自己处理剩余的清单……
但是,至少,您可以切换两个选项的顺序.
有趣的来源here.
标签:python,command-line-arguments,getopt