如何不使用subprocess模块执行Python脚本__main__块代码并传递argparse参数？

我刚好处理过类似的需求，这里分两种场景给你解决方案，优先推荐第一种更优雅的模块化方式：

场景1：可以修改目标脚本（推荐方案） #

这种方式最符合Python的模块化设计，把__main__里的逻辑封装成可调用的函数，既方便复用也便于测试。

步骤1：重构目标脚本 #

假设你的目标脚本叫target_script.py，原来的代码是这样的：

import argparse

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="你的脚本描述")
    parser.add_argument("--input", type=str, required=True, help="输入文件路径")
    parser.add_argument("--output", type=str, help="输出文件路径")
    parsed_args = parser.parse_args()
    
    # 这里是你的业务逻辑
    print(f"处理输入文件: {parsed_args.input}")
    if parsed_args.output:
        print(f"结果将保存到: {parsed_args.output}")

我们把__main__里的逻辑抽成一个main()函数，并让它接受可选的args参数：

import argparse

def main(args=None):
    # 初始化解析器
    parser = argparse.ArgumentParser(description="你的脚本描述")
    parser.add_argument("--input", type=str, required=True, help="输入文件路径")
    parser.add_argument("--output", type=str, help="输出文件路径")
    
    # 如果args为None，就读取命令行参数；否则用传入的参数列表
    parsed_args = parser.parse_args(args)
    
    # 原来的业务逻辑保持不变
    print(f"处理输入文件: {parsed_args.input}")
    if parsed_args.output:
        print(f"结果将保存到: {parsed_args.output}")

if __name__ == '__main__':
    main()  # 命令行运行时，不传参就自动读取sys.argv

步骤2：在调用脚本中导入并执行 #

新建一个caller_script.py，直接导入目标脚本的main函数，然后传入构造好的参数列表：

from target_script import main

# 构造参数列表，格式和命令行输入完全一致
args_to_pass = ["--input", "my_data.txt", "--output", "my_result.txt"]
# 如果是短参数，比如-i，就写成["-i", "my_data.txt"]

# 调用main函数，传入参数
main(args_to_pass)

这种方案的好处：

• 无全局状态依赖，不会影响其他代码
• 代码复用性高，方便单元测试（直接传测试参数即可）
• 结构清晰，符合Python的最佳实践

场景2：无法修改目标脚本（兼容方案） #

如果因为各种原因不能修改目标脚本，我们可以通过修改sys.argv来模拟命令行参数，然后让目标脚本以__main__模式执行。

假设目标脚本还是原始的未修改版本，调用脚本caller_script.py的代码如下：

import sys
from importlib.machinery import SourceFileLoader

# 保存原来的sys.argv，执行完后恢复，避免影响后续代码
original_argv = sys.argv.copy()

try:
    # 设置模拟的命令行参数：第一个元素是脚本名（随便填即可，argparse会忽略它）
    sys.argv = ["target_script.py", "--input", "my_data.txt", "--output", "my_result.txt"]
    
    # 加载并执行目标脚本，相当于在命令行运行`python target_script.py ...`
    SourceFileLoader("__main__", "target_script.py").load_module()
finally:
    # 恢复原来的sys.argv
    sys.argv = original_argv

注意事项： #

• 这种方式依赖全局的sys.argv，如果你的调用脚本还有其他依赖命令行参数的代码，一定要记得恢复原始的sys.argv
• 如果目标脚本中有其他依赖__name__的逻辑，可能会出现意外行为，所以这只是临时的兼容方案，优先推荐场景1的重构方式

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