如何在Python中用argparse实现[-A [-b value]]格式的子解析？

你的问题出在误用了subparser——subparser是用来处理子命令（比如test.py start这种不带短横线的命令）的，而-A是一个普通的命令行选项，应该直接用argparse的add_argument来定义。

正确实现代码 #

下面是符合你需求的代码，能支持test.py -A -b 123、test.py -A以及不带任何参数的调用：

import argparse

def main():
    parser = argparse.ArgumentParser(description="Example for [-A [-b value]] arguments")
    # 定义-A为开关选项，不需要参数，触发时args.A为True
    parser.add_argument("-A", action="store_true", help="Enable feature A")
    # 定义-b为可选参数，仅当-A启用时才需要关注它的值
    parser.add_argument("-b", type=int, help="Optional integer value for feature A")

    args = parser.parse_args()

    # 可以在这里添加逻辑，比如只有-A开启时才处理-b
    if args.A:
        print(f"Feature A is enabled. b value: {args.b if args.b is not None else 'not provided'}")
    else:
        print("Feature A is disabled.")

if __name__ == "__main__":
    main()

额外限制：强制-b必须和-A一起使用 #

如果你的需求是不能单独使用-b（即只有指定了-A时才能用-b），可以在解析后添加校验逻辑：

args = parser.parse_args()
# 检查如果-b被指定但-A没开启，就抛出错误
if args.b is not None and not args.A:
    parser.error("-b can only be used when -A is enabled")

测试用例 #

• 执行python test.py -A -b 123：会输出Feature A is enabled. b value: 123
• 执行python test.py -A：会输出Feature A is enabled. b value: not provided
• 执行python test.py：会输出Feature A is disabled.
• 如果添加了校验逻辑，执行python test.py -b 123会直接抛出错误：error: -b can only be used when -A is enabled

内容的提问来源于stack exchange，提问作者Panda