Python用户输入命令参数解析：替代大量if语句的实现方案

当程序需要处理用户输入的命令及参数时，用大量if/elif判断会导致代码臃肿、维护困难。以下两种方案可以更优雅地实现需求：

方案1：拆分输入+命令处理函数映射 #

先将用户输入拆分为命令和参数列表，再用字典将命令映射到对应的处理函数，新增命令只需添加函数和字典条目即可。

示例代码 #

def handle_help(args):
    if "-extra" in args:
        print("额外帮助信息：包含所有命令的详细用法、参数说明及使用场景")
    elif "--command" in args:
        try:
            cmd_idx = args.index("--command")
            target_cmd = args[cmd_idx + 1]
            # 可扩展更多命令的说明
            if target_cmd == "exit":
                print(f"命令 {target_cmd} 的详细说明：输入exit即可终止当前程序会话，无额外参数")
            else:
                print(f"暂无命令 {target_cmd} 的详细说明")
        except IndexError:
            print("格式错误：请指定要查询的命令，示例：help --command exit")
    else:
        print("基础帮助菜单：\n- help：显示基础帮助\n- help -extra：显示额外帮助\n- help --command <命令名>：查询指定命令详情\n- exit：退出程序")

def handle_exit(args):
    print("正在退出程序...")
    exit()

# 命令与处理函数的映射字典
command_handlers = {
    "help": handle_help,
    "exit": handle_exit
}

# 主逻辑
while True:
    user_input = input("enter a cmd: ").strip()
    if not user_input:
        print("请输入有效命令")
        continue
    parts = user_input.split()
    cmd = parts[0]
    args = parts[1:]
    
    if cmd in command_handlers:
        command_handlers[cmd](args)
    else:
        print(f"未知命令：{cmd}，输入help查看可用命令")

优点 #

• 代码结构清晰，新增命令无需修改主逻辑
• 轻量无依赖，适合简单命令场景

方案2：使用argparse解析输入字符串 #

argparse是Python标准库，原本用于处理程序启动时的命令行参数，也可以用来解析用户输入的命令字符串，它会自动处理参数校验、格式检查和帮助信息生成。

示例代码 #

import argparse

# 创建主解析器
main_parser = argparse.ArgumentParser(prog="my_program", add_help=False)
subparsers = main_parser.add_subparsers(dest="command", required=True)

# 为help命令创建子解析器
help_subparser = subparsers.add_parser("help", help="显示帮助信息")
help_subparser.add_argument("-extra", action="store_true", help="显示额外帮助内容")
help_subparser.add_argument("--command", type=str, help="查询指定命令的详细说明")

# 为exit命令创建子解析器
exit_subparser = subparsers.add_parser("exit", help="退出程序")

while True:
    user_input = input("enter a cmd: ").strip()
    if not user_input:
        print("请输入有效命令")
        continue
    try:
        # 将输入字符串拆分为列表，模拟命令行参数
        parsed_args = main_parser.parse_args(user_input.split())
        
        if parsed_args.command == "help":
            if parsed_args.extra:
                print("额外帮助信息：所有命令的完整参数规则、使用示例及注意事项")
            elif parsed_args.command:
                if parsed_args.command == "exit":
                    print("exit命令：用于立即终止程序，无额外参数")
                else:
                    print(f"无法查询未知命令 {parsed_args.command} 的详情")
            else:
                print("基础帮助：输入help -extra查看更多内容，或help --command <命令>查询指定命令")
        elif parsed_args.command == "exit":
            print("程序已退出")
            break
    except SystemExit:
        # 捕获argparse的参数错误提示
        print("输入格式错误，输入help查看正确命令格式")

优点 #

• 自动处理参数校验、格式错误提示
• 适合参数规则复杂的场景，扩展性强

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