12

これは、引数を複数回指定できるようにすることに関する質問で取り上げられているトピックに少し関連しています。

次のようにオプションを複数回指定できるようにしたいと思います。

 tool --foo 1 --foo 2 --foo 3

そしてまたこのように:

 tool a b c

また、両方を同時にサポートしたいと思います。

 tool a b c --foo 1 --foo2 --foo 3

これは次の場合に正常に機能します。

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('foo', nargs='*', action='append')
parser.add_argument('--foo', nargs='*', dest='foo', action='append')

結果リストは簡単にフラット化できます。

args = parser.parse_args('a b c --foo 1 --foo 2 --foo 3'.split())
args.foo = [el for elements in args.foo for el in elements]

収量:

>>> args
Namespace(foo=['a', 'b', 'c', '1', '2', '3'])

ユーザーが1つの引数を指定するとすぐにデフォルトが使用されないように、デフォルト値を追加するにはどうすればよいですか?

default=[['spam']]呼び出しの1つだけに追加する場合add_argument()、デフォルトは常に結果の一部です。ユーザーが自分で引数を指定するとすぐに、argparseを取得してそれを単独で削除することはできません。

私はargparseすでにそれ自体を提供しているもので解決策があることを望んでいます。

4

1 に答える 1

7

self.defaultこれは、他の回答(カスタムアクションの属性に依存)のもう少しクリーンなバリエーションだと思います。

import argparse
import sys

class Extender(argparse.Action):
    def __call__(self,parser,namespace,values,option_strings=None):
        #Need None here incase `argparse.SUPPRESS` was supplied for `dest`
        dest = getattr(namespace,self.dest,None) 
        #print dest,self.default,values,option_strings
        if(not hasattr(dest,'extend') or dest == self.default):
            dest = []
            setattr(namespace,self.dest,dest)
            #if default isn't set to None, this method might be called
            # with the default as `values` for other arguements which
            # share this destination.
            parser.set_defaults(**{self.dest:None}) 

        try:
            dest.extend(values)
        except ValueError:
            dest.append(values)

        #another option:
        #if not isinstance(values,basestring):
        #    dest.extend(values)
        #else:
        #    dest.append(values) #It's a string.  Oops.

def new_parser(args):
    parser = argparse.ArgumentParser()
    parser.add_argument('foo', nargs='*',action=Extender)
    parser.add_argument('--foo', nargs='*', dest='foo', action=Extender)
    parser.set_defaults(foo = [['spam']])
    return parser.parse_args(args.split())

tests = {'a b c --foo 1 --foo 2 --foo 3':['a','b','c','1','2','3'],
         '':[['spam']],
         'a b c --foo 1 2 3':['a','b','c','1','2','3'],
         '--foo 1':['1'],
         'a':['a']}

for s,r in tests.items():
    print ( "parsing: {0}".format(s) )
    args = new_parser(s)
    if(args.foo != r):
        print ("ERROR")
        print (args.foo)
        print (r)
        sys.exit(1)
    print ( args )
    print ('*'*80)

また、属性parser.set_defaults(...)のデフォルトを設定するために使用したことにも注意してください。foo

于 2012-09-17T14:17:45.587 に答える