从命令行选项创建数组(python :: optparse)

Question

有一个python脚本从命令行读取基准名称,如下所示:

-b benchname1

这个代码的代码是:

import optparse
import Mybench
parser = optparse.OptionParser()
# Benchmark options
parser.add_option("-b", "--benchmark", default="", help="The benchmark to be loaded.")
if options.benchmark == 'benchname1':
  process = Mybench.b1
elif options.benchmark == 'benchname2':
  process = Mybench.b2
else:
  print "no such benchmark!"

我想要做的是为这个命令行创建一个基准数组:

-b benchname1 benchname2

所以"进程"应该是一个数组:

process[0] = Mybench.b1
process[1] = Mybench.b2

对此有什么建议吗？

感谢名单

Answer 1

如果你有Python 2.7+,你可以使用argparse模块而不是optparse.

import argparse

parser = argparse.ArgumentParser(description='Process benchmarks.')
parser.add_argument("-b", "--benchmark", default=[], type=str, nargs='+',
                    help="The benchmark to be loaded.")

args = parser.parse_args()
print args.benchmark

示例运行脚本 -

$ python sample.py -h
usage: sample.py [-h] [-b BENCHMARK [BENCHMARK ...]]

Process benchmarks.

optional arguments:
  -h, --help            show this help message and exit
  -b BENCHMARK [BENCHMARK ...], --benchmark BENCHMARK [BENCHMARK ...]
                        The benchmark to be loaded.

$ python sample.py -b bench1 bench2 bench3
['bench1', 'bench2', 'bench3']