Question

我编写了一个带有一个或多个输入文件的脚本。我希望能够分别指定脚本（A和B）和每个输入文件的选项（C和D）。

用法应如下：

script.py [-A] [-B]  [-C] [-D] file1 [[-C] [-D] file2] ...

如何使用argparse完成？

谢谢！

Answer 1

如果可能，请尝试docopt。它更容易使用，并有足够的示例来开始。

Answer 2

我想暂时这样做，但从来没有想过用例。你找到了一个：谢谢！

你可以通过两个阶段的argparsing来做到这一点。在第一阶段，您只需查看-A或-B选项。

在第二阶段，您将剩余的参数拆分为片段（在本例中使用生成器函数）并在片段上调用parse_args：

import argparse

def fileargs(args):
    result = []
    for arg in args:
        result.append(arg)
        if not arg.startswith('-'):
            yield result
            result = []

parser = argparse.ArgumentParser()
parser.add_argument('-A', action = 'store_true')
parser.add_argument('-B', action = 'store_true')
args, unk = parser.parse_known_args()
print(args)

file_parser = argparse.ArgumentParser()
file_parser.add_argument('-C', action = 'store_true')
file_parser.add_argument('-D', action = 'store_true')
file_parser.add_argument('file')
for filearg in fileargs(unk):
    fargs = file_parser.parse_args(filearg)
    print(fargs)

然后test.py -A -B -C -D file1 -C file2产生

Namespace(A=True, B=True)
Namespace(C=True, D=True, file='file1')
Namespace(C=True, D=False, file='file2')

Answer 3

我的，这个答案很复杂：

import sys

#Unforunately, you can't split up positional arguments in a reasonable way if you
#don't know about all of them...  Count positional arguments (files)

def how_many_files(lst):
    return sum(1 for x in lst if not x.startswith('-'))

args = sys.argv[1:]
Nfiles = how_many_files(args)

import argparse

#Create our own NameSpace class so that we can have an easy handle on the
#attributes that the namespace actually holds.
class MyNameSpace(argparse.Namespace,dict):
    def __init__(self):
        argparse.Namespace.__init__(self)
        dict.__init__(self)

    def __setattr__(self,k,o):
        argparse.Namespace.__setattr__(self,k,o)
        self[k] = o

class MyParser(argparse.ArgumentParser):
    def __init__(self,*args,**kwargs):
        self.my_parents = kwargs.get('parents',[])
        argparse.ArgumentParser.__init__(self,*args,**kwargs)

class FooAction(argparse.Action):
    def __call__(self,parser,namespace,value,option_string=None):
        ref = namespace.pop('accumulated',{})
        try:
            del namespace.accumulated
        except AttributeError:
            pass

        #get a new namespace and populate it with the arguments we've picked up
        #along the way        
        new_namespace = self.__default_namespace(parser)
        for k,v in namespace.items():
            setattr(new_namespace,k,v)
            delattr(namespace,k)  #delete things from `namespace.__dict__`

        namespace.clear() #also remove things from the dictionary side.
        namespace.accumulated = ref
        new_namespace.file = value
        ref[value] = new_namespace

    def __default_namespace(self,parser):
        n = argparse.Namespace()
        for parent in parser.my_parents:
            parent.parse_args([],namespace=n)
        return n


parser = argparse.ArgumentParser()
parser.add_argument('-A',action='store_true')
parser.add_argument('-B',action='store_true')
parser.add_argument('-C',action='store_true')
parser.add_argument('-D',action='store_true')


parser2 = MyParser(parents=[parser],conflict_handler='resolve')
for i in range(Nfiles):
    parser2.add_argument('files%d'%i,action=FooAction,default=argparse.SUPPRESS)


n = parser2.parse_args(args,namespace = MyNameSpace())
for k,v in n.accumulated.items():
    print k,v

用以下方式调用：

~ $ python test.py -A foo bar -A -B -C qux

的产率：

qux Namespace(A=True, B=True, C=True, D=False, file='qux')
foo Namespace(A=True, B=False, C=False, D=False, file='foo')
bar Namespace(A=False, B=False, C=False, D=False, file='bar')

Answer 4

通常，问题似乎并不完全是需要的......对不起。这是我的工作结果（受unutbu的答案启发），它也允许带有参数的每个文件选项。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-A', action = 'store_true')
parser.add_argument('-B', action = 'store_true')
args, unk = parser.parse_known_args()

file_parser = argparse.ArgumentParser()
file_parser.add_argument('-C', action = 'store_true')
file_parser.add_argument('-D', action = 'store_true')
file_parser.add_argument('-V', "--variable-list")
file_parser.add_argument('file')

fargs=[]
n=len(unk)
while True:
        i=0
        for i in  range(n): # finding longest fully parsable tail
            Parsed, unkf = file_parser.parse_known_args(unk[i:n])
            if not unkf: break
        if i==n: # did not found parsable tail
            file_parser.parse_args(unk[0:n]) # cause error 
        else:
            fargs.append(Parsed)
            n=i
        if (n<=0): break
fargs.reverse()

print args
for argl in fargs:
        print argl

用以下方式调用：

myscript.py -A -B -C -D file1 -C -V a,b,c file

的产率：

Namespace(A=True, B=True)
Namespace(C=True, D=True, file='file1', variable_list=None)
Namespace(C=True, D=False, file='file2', variable_list='a,b,c')

Answer 5

action=append选项可能会有所帮助。这将允许您多次指定选项，并且具有特定选项的所有参数将存储在各自的列表中。

...这是一个例子，我们称之为sample.py：

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("-c", "--cfilein", action="append")
    parser.add_argument("-d", "--dfilein", action="append")
    args = parser.parse_args()
    print args.cfilein
    print args.dfilein

Execute: ./sample.py -c f1 -d f2 -c f3 -d f4
Output:  ['f1', 'f3']
         ['f2', 'f4']

退房：http://docs.python.org/2/library/argparse.html#action ... 希望这会有所帮助...

Python argparse：位置参数的单独选项

5 个答案: