如何借助argparse通过标志变量在Python中按需调用执行求解器脚本

完全可以用argparse实现需求，它自带的子解析器（subparsers）特性天生适配「按选择执行对应求解器+每个求解器支持独立专属参数」的场景，比零散写if判断标志的方式规范很多，扩展性也更强。

1. 调整求解器脚本（可选但推荐） #

原有求解器脚本的逻辑可以完全保留，只需要给对外暴露的执行函数加上参数入口，方便后续接收专属配置即可，示例如下：

# script1.py
import numpy
        
def test1(param1=None):
    print(f'This is script One, 接收到的专属参数param1值为: {param1}')


if __name__ == '__main__':
    test1()

# script2.py
import numpy
        
def test2(param2=0):
    print(f'This is script two, 接收到的专属参数param2值为: {param2}')


if __name__ == '__main__':
    test2()

2. 重写main.py实现参数解析和逻辑分派 #

核心逻辑分为三块：

• 定义主解析器，存放所有求解器通用的全局参数（比如调试开关、公共配置路径等）
• 为每个求解器注册独立的子解析器，分别配置各自的专属运行参数
• 解析命令行输入后，根据选中的求解器自动分派执行，传入对应参数
  完整代码如下：

# main.py
import argparse
# 不推荐用from xxx import *，避免多脚本间重名冲突，显式导入需要的函数即可
from script1 import test1
from script2 import test2
    
if __name__=='__main__':
    # 初始化主解析器
    main_parser = argparse.ArgumentParser(description='流水线求解器调度入口')
    # 示例：添加全局通用参数
    main_parser.add_argument('--debug', action='store_true', help='开启调试日志')

    # 初始化子解析器容器，dest字段存储用户选中的求解器标识，required=True要求必须指定一个求解器
    subparsers = main_parser.add_subparsers(dest='selected_solver', required=True, help='支持的求解器列表')

    # 注册script1对应的子命令，子命令名设为s1
    s1_parser = subparsers.add_parser('s1', help='运行script1求解器')
    # 配置s1的专属参数
    s1_parser.add_argument('--p1', type=str, default='default_val', help='script1专属参数p1')

    # 注册script2对应的子命令，子命令名设为s2
    s2_parser = subparsers.add_parser('s2', help='运行script2求解器')
    # 配置s2的专属参数
    s2_parser.add_argument('--p2', type=int, default=0, help='script2专属参数p2')

    # 解析命令行参数
    args = main_parser.parse_args()

    # 按选中的求解器分派执行
    if args.selected_solver == 's1':
        if args.debug:
            print('[调试模式] 即将执行script1')
        test1(param1=args.p1)
    elif args.selected_solver == 's2':
        if args.debug:
            print('[调试模式] 即将执行script2')
        test2(param2=args.p2)

• 查看帮助文档，会自动列出所有全局参数、可用求解器、各求解器专属参数说明：

  python main.py -h
  

• 仅执行script1，传入自定义专属参数：

  python main.py s1 --p1 "自定义测试值"
  

• 仅执行script2，开启全局调试模式、传入专属参数：

  python main.py --debug s2 --p2 999
  

• 不要使用from xxx import *的导入方式，后续求解器数量增多、单文件代码量变大后，很容易出现函数、变量重名冲突，显式导入需要的对象即可。
• 后续新增求解器时，只需要三步即可完成扩展：导入新求解器的执行函数、注册对应的子解析器并配置专属参数、增加对应分支的调用逻辑，维护成本很低。
• 如果偏好布尔开关标志而非子命令形式，也可以在主解析器中添加--run-s1、--run-s2这类store_true类型的参数控制执行逻辑，但这种模式下管理各求解器的专属参数会非常混乱，不适合多求解器的流水线项目使用。

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