2

特定の文字列に基づいて、カスタム オブジェクト (以下の例では Bar) のインスタンスを作成する必要があります。タイプを次のコードに変更して実行しない場合:Bar

import argparse

VALID_BAR_NAMES = ['alfa', 'beta', 'gamma', 'delta']


class Bar:

    def __init__(self, name):
        if not name in VALID_BAR_NAMES:
            raise RuntimeError('Bar can not be {n}, '
                               'it must be one of {m}'.format(
                                    n=name, m=', '.join(VALID_BAR_NAMES)))
        self.name = name


if __name__ == '__main__':
    parser = argparse.ArgumentParser()

    parser.add_argument('foo', help='Specify the foo!')

    parser.add_argument('-b', '--bar', nargs='*',
                        choices=VALID_BAR_NAMES,
                        type=str,  # SELECTED TYPE
                        help=('Specify one or many valid bar(s)'))
    parsed_arguments = parser.parse_args()

に無効な引数を渡すと、次のかなり良い出力が得られhelloます-b。

usage: Example.py [-h]
                  [-b [{alfa,beta,gamma,delta} [{alfa,beta,gamma,delta} ...]]]
                  foo
Example.py: error: argument -b/--bar: invalid choice: 'hello' (choose from 'alfa', 'beta', 'gamma', 'delta')

ただし、例に変更type=strしてtype=Bar再度実行すると、次の出力が得られます。

Traceback (most recent call last):
  File "C:\PyTest\Example.py", line 25, in <module>
    parsed_arguments = parser.parse_args()
  File "C:\Python27\lib\argparse.py", line 1688, in parse_args
    args, argv = self.parse_known_args(args, namespace)
  File "C:\Python27\lib\argparse.py", line 1720, in parse_known_args
    namespace, args = self._parse_known_args(args, namespace)
  File "C:\Python27\lib\argparse.py", line 1926, in _parse_known_args
    start_index = consume_optional(start_index)
  File "C:\Python27\lib\argparse.py", line 1866, in consume_optional
    take_action(action, args, option_string)
  File "C:\Python27\lib\argparse.py", line 1778, in take_action
    argument_values = self._get_values(action, argument_strings)
  File "C:\Python27\lib\argparse.py", line 2218, in _get_values
    value = [self._get_value(action, v) for v in arg_strings]
  File "C:\Python27\lib\argparse.py", line 2233, in _get_value
    result = type_func(arg_string)
  File "C:\PyTest\Example.py", line 12, in __init__
    n=name, m=', '.join(VALID_BAR_NAMES)))
RuntimeError: Bar can not be hello, it must be one of alfa, beta, gamma, delta

これはかなり悪いようです。これは、利用可能な選択肢に対するチェックが行われる前に型変換が行われるためであることを理解しています。これを処理する最良の方法は何でしょうか?

4

3 に答える 3

2

カスタム アクションを作成する必要があります (未テスト):

class Bar: 
    ...


class BarAction(argparse.Action):
    def __call__(self,parser,namespace,values,option_string=None):
        try:  #Catch the runtime error if it occures.
           l=[Bar(v) for v in values] #Create Bars, raise RuntimeError if bad arg passed.
        except RuntimeError as E:
           #Optional:  Print some other error here.  for example: `print E; exit(1)`  
           parser.error()

        setattr(namespace,self.dest,l) #add the list to the namespace

...
parser.add_argument('-b', '--bar', nargs='*',
                    choices=VALID_BAR_NAMES,
                    action=BarAction,  # SELECTED TYPE -- The action does all the type conversion instead of the type keyword.
                    help=('Specify one or many valid bar(s)'))

...
于 2012-04-24T13:11:14.657 に答える
1

解析が完了するまで、引数を文字列として保持します。解析が完了したら、まずそれらをドメイン オブジェクトに変換します。

于 2012-04-24T12:40:23.477 に答える
0

Don't know if it helps but I looked at argparse.FileType as a reference:

class Bar:
    def __call__(self, strting):
        if not name in VALID_BAR_NAMES:
            raise argparse.ArgumentError(None,'Bar can not be {n}, '
                               'it must be one of {m}'.format(
                                    n=name, m=', '.join(VALID_BAR_NAMES)))
        self.name = string

    def __init__(self):
        pass

    def __repr__(self):
        return name

parser.add_argument('-b', '--bar', nargs='*',
                    choices=VALID_BAR_NAMES,
                    type=Bar(),  # SELECTED TYPE
                    help=('Specify one or many valid bar(s)'))
于 2014-06-16T10:38:02.913 に答える