允许 Argparse 参数的特定值[重复]

Question

是否可以要求 argparse 参数是几个预设值之一？

我目前的方法是手动检查参数，如果它不是允许的值之一，请调用 print_help() 并退出。

这是当前的实现：

...
parser.add_argument('--val',
                    help='Special testing value')

args = parser.parse_args(sys.argv[1:])
if args.val not in ['a', 'b', 'c']:
    parser.print_help()
    sys.exit(1)

并不是说这个特别难，而是看起来很乱。

Answer 1

可以使用choices 参数将 argparse 参数限制为特定值：

...
parser.add_argument('--val',
                    choices=['a', 'b', 'c'],
                    help='Special testing value')

args = parser.parse_args(sys.argv[1:])

有关详细信息，请参阅docs。