Python 参数解析

Question

我想使用 Python ArgParse 仅接受来自用户的特定输入。

所以在下面的例子中，假设我只想接受 'type1/type2/type3' 作为参数。那可能吗？

parser.add_argument('-t', '--type', type = str, help = 'type1/type2/type3')

Answer 1

使用 choices argument 将输入限制为一组有限的选择：

parser.add_argument('-t', '--type', choices=('type1', 'type2', 'type3'), 
                    help='type1/type2/type3')