解析Python中的空选项

时间:2009-08-04 18:32:47

标签: python optparse optionparser

我有一个应用程序,允许您将事件数据发送到自定义脚本。您只需布置命令行参数并分配哪些事件数据与哪个参数相关。问题是这里没有真正的灵活性。您将映射的每个选项都将被使用,但并非每个选项都必须包含数据。因此,当应用程序构建要发送到脚本的字符串时,某些参数为空,并且python的OptionParser错误输出“error: - someomegument option需要参数”

由于存在超过200个数据点,因此我不能编写单独的脚本来处理可能的参数的每个组合(它需要2 ^ 200个脚本)。有没有办法在python的optionparser中处理空参数?

6 个答案:

答案 0 :(得分:8)

抱歉,我的第一个回答误解了这个问题。在定义选项时,您可以完成使命令行标志的可选参数使用回调操作类型的功能。使用以下函数作为回调(您可能希望根据您的需要定制)并为每个可以选择接收参数的标志配置它:

import optparse

def optional_arg(arg_default):
    def func(option,opt_str,value,parser):
        if parser.rargs and not parser.rargs[0].startswith('-'):
            val=parser.rargs[0]
            parser.rargs.pop(0)
        else:
            val=arg_default
        setattr(parser.values,option.dest,val)
    return func

def main(args):
    parser=optparse.OptionParser()
    parser.add_option('--foo',action='callback',callback=optional_arg('empty'),dest='foo')
    parser.add_option('--file',action='store_true',default=False)
    return parser.parse_args(args)

if __name__=='__main__':
    import sys
    print main(sys.argv)



从命令行运行,你会看到:

# python parser.py
(<Values at 0x8e42d8: {'foo': None, 'file': False}>, [])

# python parser.py --foo
(<Values at 0x8e42d8: {'foo': 'empty', 'file': False}>, [])

# python parser.py --foo bar
(<Values at 0x8e42d8: {'foo': 'bar', 'file': False}>, [])

答案 1 :(得分:1)

我认为optparse无法做到这一点。 argparse是一个不同的(非标准)模块,handle situations like this选项具有可选值。

使用optparse时,您必须指定包含该值的选项,或者同时省略它们。

答案 2 :(得分:0)

是的,添加选项时会有一个参数:

from optparse import OptionParser
parser = OptionParser()
parser.add_option("--SomeData",action="store", dest="TheData", default='')

默认参数指定您希望选项指定的值,但可以选择使用参数。

答案 3 :(得分:0)

Optparse已经允许您将空字符串作为选项参数传递。因此,如果可能,将空字符串视为“无值”。对于长期选项,可以执行以下任何工作:

my_script --opt=   --anotheroption
my_script --opt='' --anotheroption
my_script --opt="" --anotheroption
my_script --opt '' --anotheroption
my_script --opt "" --anotheroption

对于短格式选项,您可以使用以下任一项:

my_script -o '' --anotheroption
my_script -o "" --anotheroption

警告:这已经在Linux下进行了测试,并且在其他Unix类系统下应该可以正常工作; Windows以不同方式处理命令行引用,可能不接受上面列出的所有变体。

答案 4 :(得分:0)

Mark Roddy的解决方案可行,但它需要在运行时修改解析器对象的属性,并且不支持除 - 或 - 之外的其他选项格式。 稍微不那么复杂的解决方案是在运行optparse之前修改sys.argv数组,并在不需要参数的开关之后插入空字符串(“”)。 这种方法的唯一约束是你的选项默认为可插入到sys.argv中的可预测值(我在下面的例子中选择了None,但它确实没关系)。

以下代码创建一个示例解析器和一组选项,从解析器中提取一个允许的开关数组(使用一点实例变量magic),然后遍历sys.argv,每次找到一个 允许切换,它检查它是否在没有任何参数的情况下给出。如果切换后没有参数,则会在命令中插入空字符串 线。更改sys.argv后,将调用解析器,您可以检查其值为“”的选项,并相应地执行操作。

#Instantiate the parser, and add some options; set the options' default values to None, or something predictable that
#can be checked later.
PARSER_DEFAULTVAL = None
parser = OptionParser(usage="%prog -[MODE] INPUT [options]")
#This method doesn't work if interspersed switches and arguments are allowed.
parser.allow_interspersed_args = False
parser.add_option("-d", "--delete", action="store", type="string", dest="to_delete", default=PARSER_DEFAULTVAL)
parser.add_option("-a", "--add", action="store", type="string", dest="to_add", default=PARSER_DEFAULTVAL)

#Build a list of allowed switches, in this case ['-d', '--delete', '-a', '--add'] so that you can check if something
#found on sys.argv is indeed a valid switch. This is trivial to make by hand in a short example, but if a program has
#a lot of options, or if you want an idiot-proof way of getting all added options without modifying a list yourself,
#this way is durable. If you are using OptionGroups, simply run the loop below with each group's option_list field.
allowed_switches = []
for opt in parser.option_list:
    #Add the short (-a) and long (--add) form of each switch to the list.
    allowed_switches.extend(opt._short_opts + opt._long_opts)

#Insert empty-string values into sys.argv whenever a switch without arguments is found.
for a in range(len(sys.argv)):
    arg = sys.argv[a]
    #Check if the sys.argv value is a switch
    if arg in allowed_switches:
        #Check if it doesn't have an accompanying argument (i.e. if it is followed by another switch, or if it is last
        #on the command line)
        if a == len(sys.argv) - 1 or argv[a + 1] in allowed_switches:
            sys.argv.insert(a + 1, "")

options, args = parser.parse_args()

#If the option is present (i.e. wasn't set to the default value)
if not (options.to_delete == PARSER_DEFAULTVAL):
    if options.droptables_ids_csv == "":
        #The switch was not used with any arguments.
        ...
    else:
        #The switch had arguments.
        ...

答案 5 :(得分:0)

检查cp命令是否理解例如--backup=simple --backup simple,我回答了这样的问题:

import sys
from optparse import OptionParser

def add_optval_option(pog, *args, **kwargs):
    if 'empty' in kwargs:
        empty_val = kwargs.pop('empty')
        for i in range(1, len(sys.argv)):
            a = sys.argv[i]
            if a in args:
                sys.argv.insert(i+1, empty_val)
                break
    pog.add_option(*args, **kwargs)

def main(args):
    parser = OptionParser()
    add_optval_option(parser,
                      '--foo', '-f',
                      default='MISSING',
                      empty='EMPTY',
                      help='"EMPTY" if given without a value. Note: '
                      '--foo=VALUE will work; --foo VALUE will *not*!')
    o, a = parser.parse_args(args)
    print 'Options:'
    print '  --foo/-f:', o.foo
    if a[1:]:
        print 'Positional arguments:'
        for arg in a[1:]:
            print ' ', arg
    else:
        print 'No positional arguments'

if __name__=='__main__':
    import sys
    main(sys.argv)

自我广告:这是我opo包的thebops模块的一部分......; - )