ArgParse：如何将选择映射到不同的值？

Question

使用ArgParse时，我希望用户从选择中选择，但是我希望他们选择确定更复杂的值（类似于store_const的工作方式）。

例如，从吸烟者选择 ['Current'，''，''，'Never']的状态时，我希望Current to映射到“当前每天吸烟者。”

有没有一种优雅的方法来做到这一点？

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--smoker', choices=['current','former','never'])
print(vars(parser.parse_args()))

正常输出：

$ ./script.py --smoker current
{'smoker': 'current'}

需要：

$ ./script.py --smoker current
{'smoker': 'Current every day smoker.'}

我认为我可以使用lambda键入参数来执行此操作，但是argparse强制执行选择列表：

choice_dict = {'current': 'Current everyday...'}
parser.add_argument('--smoker', type=lambda x: choice_dict[x], choices=choice_dict.keys())
print(vars(parser.parse_args()))

./script.py --smoker current
usage: script.py [-h] [--smoker {current}]
script.py: error: argument --smoker: invalid choice: 'Current everyday...' (choose from 'current')

Answer 1

尽管我同意0x5452的评论，即最好将这种格式与解析器解），但您可以使用action来做您想要的事情：

import argparse

CONVERSION_TABLE = {'current': 'Currently smokes',
                    'former': 'Used to smoke',
                    'never': 'Never smoke'}

class RenameOption(argparse.Action):
     def __call__(self, parser, namespace, values, option_string=None):
         setattr(namespace, self.dest, CONVERSION_TABLE[values])

parser = argparse.ArgumentParser()
parser.add_argument('--smoker', action=RenameOption, choices=CONVERSION_TABLE)
print(vars(parser.parse_args()))

结果是：

$ python test.py --smoker current
{'smoker': 'Currently smokes'}

$ python test.py --smoker no
usage: test.py [-h] [--smoker {current,former,never}]
test.py: error: argument --smoker: invalid choice: 'no' (choose from 'current', 'former', 'never')