带有选择的 Python argparse

Question

我想使用一些选项从命令行运行我的 python 程序。例如。假设我的程序有 4 种模式，我希望它通过给它以下参数来使用模式 2:

$ python main.py --mode 2

(对于模式 1、3、4 也类似)。我如何使用 argparse 实现这一目标？

Answer 1

import argparse


parser = argparse.ArgumentParser(description='PROJECT_NAME')
parser.add_argument(
    '--mode', '-m',
    help='Set mode',
    default=1,
    type=int,
    choices=[1,2,3,4],
)
args = parser.parse_args()

print(args.mode)

有关可用选项的完整列表，请访问文档:

https://docs.python.org/3/library/argparse.html

更新:

添加了 MaLiN2223 评论中的建议