如何在不修改源码的情况下调用无def main的Python模块主逻辑？

当然有可行的办法！你不需要修改那个开源模块的代码，就能触发它的if __name__ == '__main__':分支逻辑，下面给你两种实用的实现方式，都能把调用整合到你的Python 3.5代码中：

方法1：使用importlib.util隔离加载模块（推荐） #

这个方法会在独立的命名空间中加载并执行目标模块，不会污染你自己代码的变量环境，还能轻松传递命令行参数：

import importlib.util
import sys

def run_open_source_module(module_path, args=None):
    # 创建模块的规范对象
    spec = importlib.util.spec_from_file_location("__main__", module_path)
    # 创建空的模块对象
    target_module = importlib.util.module_from_spec(spec)
    
    # 保存原有的sys.argv，避免被修改
    original_argv = sys.argv.copy()
    
    try:
        # 设置模块的__name__为'__main__'，触发主逻辑分支
        target_module.__name__ = '__main__'
        # 配置命令行参数（如果需要的话）
        sys.argv = [module_path] + (args or [])
        # 执行模块代码
        spec.loader.exec_module(target_module)
    finally:
        # 恢复原来的sys.argv，不影响后续代码
        sys.argv = original_argv

使用示例 #

假设你要调用的模块路径是./third_party_module.py，并且需要传递两个参数--input data.txt和--output result.txt：

run_open_source_module("./third_party_module.py", ["--input", "data.txt", "--output", "result.txt"])

方法2：直接读取并执行模块代码 #

这个方法更直接，但要注意可能会有变量污染的风险（因为会在当前命名空间执行代码），适合简单场景：

import sys

def run_module_as_script(module_path, args=None):
    original_argv = sys.argv.copy()
    original_name = __name__
    
    try:
        # 临时修改sys.argv模拟命令行调用
        sys.argv = [module_path] + (args or [])
        # 把当前命名空间的__name__改成'__main__'，触发目标分支
        __name__ = '__main__'
        # 读取并执行模块代码
        with open(module_path, 'r', encoding='utf-8') as f:
            exec(f.read(), globals(), locals())
    finally:
        # 恢复原有的__name__和sys.argv
        __name__ = original_name
        sys.argv = original_argv

注意事项 #

• 两种方法都会执行模块中的所有顶级代码（包括函数定义、全局变量初始化等），和直接在命令行运行该脚本的效果完全一致。
• 如果目标模块有第三方依赖，请确保你的Python环境已经安装这些依赖包。
• 如果需要获取目标模块的输出内容（比如打印的信息），可以通过重定向sys.stdout来捕获；如果需要返回值，由于原模块没有定义main函数，这种情况下可能需要结合subprocess来捕获输出（但这是备选方案，优先推荐前两种整合到代码的方式）。

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