如何在Python中以编程方式带参数调用main函数？

算力阿峰

2026-8-18

解决方案

方法1：修改`sys.argv`模拟命令行参数（几乎无需改动main.py）

argparse默认读取sys.argv列表解析参数，你可以在util.py里临时替换该列表，再直接调用main()函数，完全模拟命令行调用上下文：

# root/folder1/folder2/util.py
import sys
from root import main

def trigger_main(param1, param2):
    # 替换sys.argv：第一个元素为程序名（argparse会忽略具体值），后续是参数
    sys.argv = [
        'main.py',
        '--param-1', param1,
        '--param-2', param2
    ]
    main.main()

这个方案不需要修改main.py的任何代码，直接复用原有参数解析逻辑。

方法2：拆分核心逻辑（少量修改main.py）

如果main.py的main()函数是先解析参数再执行业务逻辑，可将核心处理逻辑抽为独立函数，既保留命令行调用方式，又支持模块内直接调用：

修改后的`root/main.py`：

import argparse

# 抽离核心处理函数
def process_params(param1, param2):
    # 这里放原来main函数中解析参数后的业务逻辑
    print(f"处理参数：param1={param1}, param2={param2}")

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--param-1')
    parser.add_argument('--param-2')
    args = parser.parse_args()
    process_params(args.param_1, args.param_2)

if __name__ == '__main__':
    main()

在`util.py`中直接调用核心函数：

# root/folder1/folder2/util.py
from root.main import process_params

def some_task():
    param1 = "test_val1"
    param2 = "test_val2"
    process_params(param1, param2)

这个改动极小，仅解耦参数解析和业务逻辑，后续维护更灵活。