1

我是 Python 新手，我从 getopt 学到的是，我可以在执行 python 脚本时传递命令行参数。现在我的问题是，有没有办法不设置我可以在命令行中传递的参数数量？

我的代码看起来像这样：

try:
     opts, args = getopt.getopt(sys.argv[1:],"h:1:2:3:4:",["help=", "1week=","2week=", "3week=", "4week="])
except getopt.GetoptError:
     print 'test.py -1 <week 1> -2 <week 2> -3 <week 3> -4 <week 4>'
     sys.exit(2)
for opt, arg in opts:
     if opt == '-h':
          print 'test.py -1 <week 1> -2 <week 2> -3 <week 3> -4 <week 4>'
          sys.exit()
     elif opt in ("-1", "--1week"):
          week1 = arg
     elif opt in ("-2", "--2week"):
          week2 = arg
     elif opt in ("-3", "--3week"):
          week3 = arg
     elif opt in ("-4", "--4week"):
          week4 = arg

并运行我使用的上述代码 python test.py -1 89 -2 88 -3 87 -4 86。无论如何我可以修改这个，因为我不必总是给出 4 个参数，但也有 3 个？我希望它灵活，有时我可能会给出 3 个或 4 个参数。

4

2 回答 2

2

如果您的程序接受一个或多个周数：

from argparse import ArgumentParser

parser = ArgumentParser()
parser.add_argument('weeks', nargs='+', type=int, help='weeks numbers')
args = parser.parse_args()
print(args)

例子

$ python your_script.py 1 2 3

输出

Namespace(weeks=[1, 2, 3])

'weekly_1_2_3'从上面生产args.weeks：

result = 'weekly_' + '_'.join(map(str, args.weeks))

结果'weekly_1_2_3_4'用于1 2 3 4命令行。

于 2012-10-17T18:00:42.820 回答
0

请改用argparse(python 2.7)。默认情况下，所有选项都被认为是可选的。argparse包括required“必需选项”的属性。

例如：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--firstname', dest='fn', required=True)
parser.add_argument('--greeting', dest='grt', nargs='?', choices=['hello', 'goodbye'], default='hello')
parser.add_argument('--repeat', dest='rpt', nargs='?', default=1, type=int)
args = parser.parse_args()

for n in range(args.rpt):
    print "%d: %s, %s" % (n+1, args.grt, args.fn)
于 2012-10-17T17:38:31.340 回答