“重载”不适用于stderr

时间:2012-05-04 01:00:18

标签: python c python-c-api python-embedding

/*---------------stdcallbk_module.h---------------------*/
#ifndef STDCALLBK_MODULE_H
#include <Python.h>
#define STDCALLBK_MODULE_H


// Type definition for the callback function.
typedef void __stdcall(CALLBK_FUNC_T)(const char* p);


#ifdef __cplusplus
extern "C" {
#endif
  void register_callback(CALLBK_FUNC_T* callback);
  void import_stdcallbk(void);
#ifdef __cplusplus
}
#endif


#endif    /* #define STDCALLBK_MODULE_H */


/*---------------stdcallbk_module.c---------------------*/
#include "stdcallbk_module.h"

static CALLBK_FUNC_T *callback_write_func = NULL;
static FILE *fp;    /*for debugging*/
static PyObject* g_stdout;

typedef struct
{
    PyObject_HEAD
    CALLBK_FUNC_T *write;
} Stdout;


static PyObject* write2(PyObject* self, PyObject* args)
{
    const char* p;

    if (!PyArg_ParseTuple(args, "s", &p))
        return NULL;

    fputs(p, fp);    fflush(fp);
    if(callback_write_func)
        callback_write_func(p);
    else
        printf("----%s----", p);

    return PyLong_FromLong(strlen(p));
}

static PyObject* flush2(PyObject* self, PyObject* args)
{
    // no-op
    return Py_BuildValue("");
}


static PyMethodDef Stdout_methods[] =
{
    {"write", write2, METH_VARARGS, "sys-stdout-write"},
    {"flush", flush2, METH_VARARGS, "sys-stdout-write"},
    {NULL, NULL, 0, NULL}
};



static PyTypeObject StdoutType =
{
    PyVarObject_HEAD_INIT(0, 0)
    "stdcallbk.StdoutType", /* tp_name */
    sizeof(Stdout), /* tp_basicsize */
    0, /* tp_itemsize */
    0, /* tp_dealloc */
    0, /* tp_print */
    0, /* tp_getattr */
    0, /* tp_setattr */
    0, /* tp_reserved */
    0, /* tp_repr */
    0, /* tp_as_number */
    0, /* tp_as_sequence */
    0, /* tp_as_mapping */
    0, /* tp_hash */
    0, /* tp_call */
    0, /* tp_str */
    0, /* tp_getattro */
    0, /* tp_setattro */
    0, /* tp_as_buffer */
    Py_TPFLAGS_DEFAULT, /* tp_flags */
    "stdcallbk.Stdout objects", /* tp_doc */
    0, /* tp_traverse */
    0, /* tp_clear */
    0, /* tp_richcompare */
    0, /* tp_weaklistoffset */
    0, /* tp_iter */
    0, /* tp_iternext */
    Stdout_methods, /* tp_methods */
    0, /* tp_members */
    0, /* tp_getset */
    0, /* tp_base */
    0, /* tp_dict */
    0, /* tp_descr_get */
    0, /* tp_descr_set */
    0, /* tp_dictoffset */
    0, /* tp_init */
    0, /* tp_alloc */
    0, /* tp_new */
};


static struct PyModuleDef stdcallbkmodule = {
    PyModuleDef_HEAD_INIT,
    "stdcallbk",   /* name of module */
    NULL, /* module documentation, may be NULL */
    -1,       /* size of per-interpreter state of the module,
                or -1 if the module keeps state in global variables. */
    NULL,
    NULL,
    NULL,
    NULL,
    NULL
};


PyMODINIT_FUNC PyInit_stdcallbk(void)
{
    PyObject* m;

    fp = fopen("debuggered.txt", "wt");
    fputs("got to here Vers 12\n", fp);  fflush(fp);

    StdoutType.tp_new = PyType_GenericNew;
    if (PyType_Ready(&StdoutType) < 0)
        return 0;

    m = PyModule_Create(&stdcallbkmodule);
    if (m)
    {
        Py_INCREF(&StdoutType);
        PyModule_AddObject(m, "Stdout", (PyObject*) &StdoutType);
        fputs("PyModule_AddObject() Called\n", fp); fflush(fp);
    }
    g_stdout = StdoutType.tp_new(&StdoutType, 0, 0);

    /*PySys_SetObject("stdout", g_stdout)*/
    fprintf(fp, "PySys_SetObject(stdout) returned %d\n", PySys_SetObject("stdout", g_stdout));
    fflush(fp);

    /*PySys_SetObject("stderr", g_stdout)*/
    fprintf(fp, "PySys_SetObject(stderr) returned %d\n", PySys_SetObject("stderr", g_stdout));
    fflush(fp);

    return m;
}




/* called by cpp exe _after_ Py_Initialize(); */
void __declspec(dllexport) import_stdcallbk(void)
{
    PyImport_ImportModule("stdcallbk");
}



/* called by cpp exe _before_ Py_Initialize(); */
void __declspec(dllexport) register_callback(CALLBK_FUNC_T* callback)
{
    PyImport_AppendInittab("stdcallbk", &PyInit_stdcallbk);
    callback_write_func = callback;
}

/*------------- Embarcadero C++ Builder exe ---------------------*/
#include "/py_module/stdcallbk_module.h"



void __stdcall callback_write_func(const char* p)
{
    ShowMessage(p);
}
//---------------------------------------------------------------------------
void __fastcall TForm1::Button1Click(TObject *Sender)
{
    PyObject *pName, *pModule;

    register_callback(callback_write_func);

    Py_Initialize();
    import_stdcallbk();
    //PyRun_SimpleString("import stdcallbk\n");

    pName = PyUnicode_FromString("hello_emb.py");
    pModule = PyImport_Import(pName);
    Py_Finalize();

}
//---------------------------------------------------------------------------



-------------- python script - hello_emb.py ----------------
#import stdcallbk

print("HELLO FRED !")


assert 1==2


------------------------------------------------------------

上面的代码是从这里借来的:How to redirect stderr in Python? Via Python C API?

从cpp exe运行时,我收到一个消息框“HELLO FRED!”接着是一个空的消息框,我假设是回车。到目前为止一切都很好,但是当异常被提出时,我什么也得不到。没有任何内容写入文本文件。

但是......如果我从python脚本取消注释“import stdcallbk”并从命令行运行它,我明白了:

D:\projects\embed_python3\Win32\Debug>hello_emb.py
----HELLO FRED !--------
--------Traceback (most recent call last):
--------  File "D:\projects\embed_python3\Win32\Debug\hello_emb.py", line 7, in <module>
--------    --------assert 1==2--------
--------AssertionError----------------
----

和此:

D:\projects\embed_python3\Win32\Debug>type debuggered.txt
got to here Vers 12
PyModule_AddObject() Called
PySys_SetObject(stdout) returned 0
PySys_SetObject(stderr) returned 0
HELLO FRED !
Traceback (most recent call last):
  File "D:\projects\embed_python3\Win32\Debug\hello_emb.py", line 7, in <module>
    assert 1==2
AssertionError

所以它可以正常工作,而不是从cpp exe运行时。

任何人都知道这里究竟发生了什么?

干杯

1 个答案:

答案 0 :(得分:2)

您的断言将导致hello_emb导入失败。导入模块时执行模块的主体。在您的情况下,这发生在PyImport_Import()调用内。它将打印“HELLO FRED!”很好但是它会到达会引发异常的断言。

PyImport_Import()将返回NULL来表明。您必须检查并采取适当的措施(这可能并不总是打印回溯,这就是为什么它留给您)。

如果要将错误打印到stderr,则必须致电PyErr_Print()

pName = PyUnicode_FromString("hello_emb.py");
pModule = PyImport_Import(pName);
if (pModule == NULL)
    PyErr_Print();

代码无关的问题:

这不是唯一缺少错误检查的地方(但只有上述错误是缺少追溯的原因)。调用fopen或几乎任何Python C / API调用都会失败,您应该处理它。

此外,PyInit_stdcallbk中存在引用泄漏。在创建stdout对象之后,它的引用计数为1.然后将其设置为sys.stdoutsys.stderr,使其成为3.之后,您应Py_DECREF将它设为2如果/从stdoutstderr删除它,它将从内存中释放。现在它永远不会被释放。