C++嵌入Python解释器时如何重写print()函数使其输出到自定义控制台

优先推荐重定向Python标准输出的方案，兼容性远高于直接重写print()函数，可覆盖包括numpy打印、第三方库输出在内的所有标准输出场景，不需要单独适配任意参数类型。

方案1：重定向Python标准输出（推荐） #

实现逻辑 #

Python内置的print()函数默认输出到sys.stdout对象，你只需要替换这个对象为自定义实现，所有print()输出的内容会自动传入自定义对象的write()方法，你可以在该方法中将内容传递给C++端的自定义控制台。

具体实现步骤 #

• 第一步：C++端初始化Python解释器后，通过Python C API运行如下代码，替换标准输出对象

import sys
class CustomStdout:
    def write(self, text):
        # text为print输出的完整内容，包含自动追加的换行符
        # 此处调用C++导出的控制台输出接口即可
        pass
    def flush(self):
        # 可选实现，处理缓冲区刷新逻辑
        pass
# 同时重定向标准输出和错误输出，覆盖报错信息打印
sys.stdout = CustomStdout()
sys.stderr = CustomStdout()

• 第二步：在C++端注册可被Python调用的输出函数，示例如下：

// C++端自定义控制台输出的实现
void your_custom_console_print(const char* content) {
    // 此处填写你自己的控制台输出逻辑
}

// 封装为Python可调用的函数
static PyObject* cpp_output_proxy(PyObject* self, PyObject* args) {
    const char* text;
    if (!PyArg_ParseTuple(args, "s", &text)) {
        return NULL;
    }
    your_custom_console_print(text);
    Py_RETURN_NONE;
}

// 注册函数到Python内置模块
static PyMethodDef EmbModuleMethods[] = {
    {"cpp_output", cpp_output_proxy, METH_VARARGS, "Output to custom C++ console"},
    {NULL, NULL, 0, NULL}
};

// 初始化Python解释器前调用该函数注册模块
PyImport_AppendInittab("emb_module", []() {
    static struct PyModuleDef module_def = {
        PyModuleDef_HEAD_INIT,
        "emb_module",
        NULL,
        -1,
        EmbModuleMethods
    };
    return PyModule_Create(&module_def);
});

• 第三步：在之前定义的CustomStdout.write()方法中调用C++导出的函数即可：

def write(self, text):
    import emb_module
    emb_module.cpp_output(text)

方案2：直接重写内置print()函数 #

如果你需要完全定制print()的行为（比如参数过滤、特殊格式处理等），可以直接替换Python内置的print函数，Python会自动完成任意参数的类型转换，不需要你手动适配多类型参数。

实现示例 #

import builtins
# 保存原生print函数，兼容指定file参数的场景
original_print = builtins.print

def custom_print(*args, **kwargs):
    # 读取print原生参数
    sep = kwargs.get('sep', ' ')
    end = kwargs.get('end', '\n')
    file = kwargs.get('file', None)
    flush = kwargs.get('flush', False)
    
    if file is None:
        # 未指定输出文件时走自定义逻辑
        output_str = sep.join(str(arg) for arg in args) + end
        # 调用C++输出接口，调用逻辑和方案1一致
        import emb_module
        emb_module.cpp_output(output_str)
        if flush:
            # 自定义刷新逻辑
            pass
    else:
        # 指定输出文件时走原生逻辑，避免破坏代码兼容性
        original_print(*args, **kwargs)

# 替换内置print函数
builtins.print = custom_print

注意事项 #

• 多线程场景下需要给自定义输出逻辑加锁，避免输出内容乱序
• 常规Python安装版本的所有内置库、numpy都兼容上述两种方案，不需要额外适配
• 优先选择方案1，可覆盖所有直接写入标准输出的场景，不需要担心遗漏第三方库的打印内容

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