1

正如对使用 optparse 接受参数的大多数 pythonic 方式的接受答案中所述，我有一个程序，该程序带有一个对字符串执行操作的函数。该程序使用 argparse 检查字符串是按原样提供还是在文件中提供，并根据需要处理输入以将其传递给函数。

现在我想用我的函数的更高级版本扩展程序，但仍然保留基本版本进行比较，有点像Use argparse to run 1 of 2 functions in my script。我认为我的情况不同的地方在于，无论调用什么函数，我都希望可以选择传递我现有的输入标志。

只是向解析器添加一个新参数并将我以前的代码嵌套在检查该标志的 if/else 中是行不通的：它抱怨参数数量错误。我知道子命令，但是我对 argparse 还是很陌生，而且对于我想要的东西来说，这似乎有点过分了——但也许不是。

tl; dr：我需要选择两种功能之一和两种输入类型之一；两种输入类型都适用于这两种功能。谢谢你的帮助！

编辑添加代码：

p = argparse.ArgumentParser(description="program.py")
p.add_argument("-e", dest='extended')   #The new flag causing the trouble
p.add_argument("-s", dest="string")
p.add_argument("-f", dest="infile")

args = p.parse_args()

if args.extended:
    if args.infile:
        with open(args.infile,'r') as f:
            for line in enumerate(f.readlines()):
                print 'Input: ', line[1],    
                output = funcExtended(line[1])  #new and improved function
                print 'Extended output: ', output
    elif args.string:
        output = funcExtended(args.string)
        print output
    else:  #my future default option to grab strings from a database
        print 'This will soon work: extended'
else:   #I fully realize that I shouldn't have to essentially copy and paste here
    if args.infile:
        with open(args.infile,'r') as f:
            for line in enumerate(f.readlines()):
                print 'Input: ', line[1],    
                output = funcBasic(line[1])  #old and tired function
                print 'Basic output: ', output
    elif args.string:
        output = funcBasic(args.string)
        print output
    else:   #my future default option to grab strings from a database
        print 'This will soon work: basic'

这是一个命令行实用程序。发行

$ python program.py -s 'string'

像以前一样返回格式正确的字符串。但发出

$ python program.py -s 'string' -e

返回

program.py: error: argument -e: expected one argument

唷。再次感谢任何可以提供帮助的人！

4

1 回答 1

4

如果您将extended参数更改为布尔标志

p.add_argument("-e", dest='extended', action="store_true")

它不再期待争论。然后你可以调用你的脚本

$ python program.py -e -s 'string'

最后作为奖励，这里有一些想法可以使您的代码减少冗余：

import argparse

def funcExtended(line):
   return " ".join(line)

def funcBasic(line):
    return line.upper()

p = argparse.ArgumentParser(description="program.py")
p.add_argument("-e", "--extended", dest="func", action="store_const", const=funcExtended, default=funcBasic)
p.add_argument("-s", "--string")
p.add_argument("-f", "--infile")

args = p.parse_args()

def readlines(args):
    if args.infile:
        with open(args.infile,'r') as f:
            for line in f:
                yield line.rstrip("\n")
    elif args.string:
        yield args.string
    else:  #my future default option to grab strings from a database
        print 'This will soon work: extended'

for line in readlines(args):
    print 'Input: ', line
    output = args.func(line)
    print "Output: ", output
于 2012-09-14T07:33:45.117 回答