C++嵌入Python中PyRun_File与PyImport_Import兼容问题求助

当你通过PyImport_Import导入my_file模块后，Python会将该模块实例缓存到sys.modules全局字典中。后续调用PyRun_File运行同一个文件时，即便你手动设置了globals["__name__"] = "__main__"，Python会检测到该模块已存在于缓存中，直接复用已加载的模块对象，此时模块的__name__属性仍然是my_file，而非你设置的__main__，导致if __name__ == "__main__"分支不会执行，这就是输出不符合预期的核心原因。

要解决这个问题，需要在调用run_file_main之前，先将已缓存的模块从sys.modules中移除，让Python重新加载文件并正确识别为__main__脚本。以下是具体实现步骤：

1. 实现模块卸载函数 #

添加C++函数用于从sys.modules中移除指定模块，清理缓存：

void unload_module(const std::string& module_name) {
    // 获取sys.modules字典
    PyObject* sys_modules = PyImport_GetModuleDict();
    if (!sys_modules) return;

    // 创建模块名字符串对象
    PyObject* module_name_obj = PyUnicode_FromString(module_name.c_str());
    if (!module_name_obj) {
        PyErr_Clear();
        return;
    }

    // 移除并释放模块引用
    PyObject* removed_module = PyDict_GetItem(sys_modules, module_name_obj);
    if (removed_module) {
        PyDict_DelItem(sys_modules, module_name_obj);
        Py_DECREF(removed_module);
    }

    Py_DECREF(module_name_obj);
}

2. 调整调用逻辑 #

在调用run_file_main前先卸载已导入的模块，确保文件能以__main__身份重新加载：

call_module_func("my_file", "main");
unload_module("my_file"); // 清理模块缓存
run_file_main("my_file.py");

3. 修正run_file_main函数细节 #

原代码存在命名空间污染、文件打开模式不当等问题，修正后版本：

void run_file_main(std::string file_name)
{
    // 创建独立的全局/局部字典，避免污染其他模块
    PyObject* globals = PyDict_New();
    PyObject* locals = PyDict_New();
    if (!globals || !locals) {
        Py_XDECREF(globals);
        Py_XDECREF(locals);
        return;
    }

    // 设置__name__为__main__
    PyObject* main_name = PyUnicode_FromString("__main__");
    PyDict_SetItemString(globals, "__name__", main_name);
    Py_DECREF(main_name);

    // 以文本模式打开文件（适配Python 3）
    FILE* fp = fopen(file_name.c_str(), "r");
    if (!fp) {
        Py_XDECREF(globals);
        Py_XDECREF(locals);
        return;
    }

    // 运行文件并处理错误
    PyObject* result = PyRun_File(fp, file_name.c_str(), Py_file_input, globals, locals);
    if (!result) {
        PyErr_Print();
    }

    // 清理资源
    Py_XDECREF(result);
    Py_XDECREF(globals);
    Py_XDECREF(locals);
    fclose(fp);
}

4. 补全call_module_func缺失代码 #

原代码缺少函数对象获取逻辑，补全后版本：

void call_module_func(std::string module_name, std::string func_name)
{
    PyObject* pName = PyUnicode_DecodeFSDefault(module_name.c_str());
    if (!pName) {
        PyErr_Print();
        return;
    }

    PyObject* pModule = PyImport_Import(pName);
    Py_DECREF(pName);
    if (!pModule) {
        PyErr_Print();
        return;
    }

    PyObject* pFunc = PyObject_GetAttrString(pModule, func_name.c_str());
    Py_DECREF(pModule);
    if (!pFunc || !PyCallable_Check(pFunc)) {
        PyErr_Print();
        Py_XDECREF(pFunc);
        return;
    }

    PyObject* pArgs = PyTuple_New(0);
    PyObject* result = PyObject_CallObject(pFunc, pArgs);
    Py_DECREF(pArgs);
    Py_DECREF(pFunc);

    if (!result) {
        PyErr_Print();
    } else {
        Py_DECREF(result);
    }
}

修改后无论调用顺序如何，输出都会符合预期：

• 先调用call_module_func再调用run_file_main，输出为：

  Hello world
  you are main
  Hello world
  

• 先调用run_file_main再调用call_module_func，输出与原正确顺序一致。

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