如何使用argparse将命令行指定参数文件传入其他Python文件

不需要改动参数文件结构，也不需要重构solver中的业务代码，用Python标准库的动态导入能力即可实现命令行指定参数文件的需求，所有适配逻辑都封装在入口文件main.py中，使用者不需要修改in_para.py和solver.py的核心源码。
核心实现逻辑：

• 给命令行解析新增-I/--input参数，接收用户传入的配置文件路径
• 用importlib模块动态加载任意路径下的py格式配置文件，不需要提前硬编码import语句
• 将配置文件中定义的所有公开参数注入到solver模块的全局命名空间，solver原有代码不需要调整参数读取逻辑即可直接获取nx、c等参数值

最终代码 #

1. in_para.py 完全不需要修改 #

#in_para.py
nx = 31
c = 1

2. solver.py 仅调整打印语句匹配预期输出，核心计算逻辑不动 #

#solver.py
import numpy
import os   

def simple(output):
    ny = nx + 5
    c_2 = c + 2
    with open(os.path.join(output, 'Output.txt'), 'w') as f:
         print(f"this is a simple test {ny}", file=f)
         print(f"this is a simple test2 {c_2}", file=f)
        
if __name__=='__main__':
    pass

3. main.py 新增动态导入、参数注入逻辑 #

#main.py 
import os
import argparse
import importlib.util
import sys

def load_py_module(module_name, file_path):
    spec = importlib.util.spec_from_file_location(module_name, file_path)
    module = importlib.util.module_from_spec(spec)
    sys.modules[module_name] = module
    spec.loader.exec_module(module)
    return module

def main():
    parser = argparse.ArgumentParser() 
    parser.add_argument('-I','--input',type=str, required=True, help='指定参数配置文件路径（不需要.py后缀）')
    parser.add_argument('-C','--Chk',type=str, help='选择要执行的求解组件')
    parser.add_argument('-O','--output',type=str, default="./Output", help='指定输出目录')

    args = vars(parser.parse_args())
 
    output = args['output']
    if not os.path.exists(output):
        os.makedirs(output)

    # 动态加载用户指定的参数文件
    para_module = load_py_module("runtime_para", f"{args['input']}.py")

    # 导入solver模块并注入参数
    import solver
    for attr_name, attr_val in vars(para_module).items():
        if not attr_name.startswith('_'):
            setattr(solver, attr_name, attr_val)

    if args['Chk'] == 'compo1':
        solver.simple(output)
    
if __name__=='__main__':
    main()

运行方式 #

直接执行目标命令即可：

python3 main.py -I in_para -C compo1 -O Output_dir

执行完成后打开Output_dir/Output.txt就能看到预期输出：

this is a simple test 36
this is a simple test2 3

该方案支持任意名称的py格式参数文件，适配参数多、业务逻辑复杂的场景，不需要对原有求解代码做侵入式修改。

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