Как получить текущее имя функции в Python C-API? - PullRequest
Новая
       7

Как получить текущее имя функции в Python C-API?

3 голосов
/ 04 июля 2011

Я реализовал несколько функций, и они отправляются из той же функции C, вызываемой интерпретатором Python: 

PyObject *
CmdDispatch(PyObject *self, PyObject *args, PyObject *kwargs)

Неожиданно self равно NULL, и мне нужно получить имя вызываемой функции. Есть ли способ получить эту информацию?

У меня есть десятки функций, которые проходят через эту процедуру. Эта команда обрабатывает все параметры в карте C ++ и передает их реализации каждой команды.

Обновление: http://docs.python.org/extending/extending.html#a-simple-example специально говорит: «Аргумент self указывает на объект модуля для функций уровня модуля; для метода он будет указывать на экземпляр объекта.», Но я получаю значение null при связывании с python 2.6.

Ответы [ 4 ]

4 голосов
/ 25 августа 2011

Я пытался решить очень похожую проблему.

Я пришел к выводу, что нет способа определить имя текущей функции или вызывающей стороны на уровне Python C API.,Причина в том, что интерпретатор Python помещает в стек вызовов только чистые функции Python (реализованные в самом Python).Функции, реализованные в C, независимо от того, зарегистрированы ли они в таблице методов модуля, не помещаются в стек Python, поэтому их невозможно найти для проверки фреймов стека.

Вот быстрый пример в Python, иллюстрирующий то, что я хотелдостижения (я предполагаю, что Хуан запрашивает похожее поведение): 

import sys
def foo():
    print('foo:', sys._getframe(0).f_code.co_name)
def bar():
    print('bar:', sys._getframe(0).f_code.co_name)
    foo()
bar()

Вот близкий эквивалент этого примера (на основе Python 3 документов ), но реализованный с использованием Python C API: 

// Based on example from Python 3.2.1 documentation, 5.4. Extending Embedded Python
// http://docs.python.org/release/3.2.1/extending/embedding.html#extending-embedded-python
//
#include <Python.h>
#include <frameobject.h>

static void foo()
{
    PyThreadState * ts = PyThreadState_Get();
    PyFrameObject* frame = ts->frame;
    while (frame != 0)
    {
        char const* filename = _PyUnicode_AsString(frame->f_code->co_filename);
        char const* name = _PyUnicode_AsString(frame->f_code->co_name);
        printf("foo: filename=%s, name=%s\n", filename, name);
        frame = frame->f_back;
    }
}

static void bar()
{
    PyRun_SimpleString(
    "import sys\n"
    "print(\"bar: filename=%s, name=%s\" % (sys._getframe(0).f_code.co_filename, sys._getframe(0).f_code.co_name))"
    );
}

static PyObject* emb_numargs(PyObject *self, PyObject *args)
{
    foo();
    bar();

    PyRun_SimpleString(
    "import sys\n"
    "print(\"emb_numargs: filename=%s, name=%s\" % (sys._getframe(0).f_code.co_filename, sys._getframe(0).f_code.co_name))"
    );

    return PyLong_FromLong(0);
}

static PyMethodDef EmbMethods[] = {
    {"numargs", emb_numargs, METH_VARARGS, "Return number 0"},
    {NULL, NULL, 0, NULL}
};

static PyModuleDef EmbModule = {
    PyModuleDef_HEAD_INIT, "emb", NULL, -1, EmbMethods,
    NULL, NULL, NULL, NULL
};

static PyObject* PyInit_emb(void)
{
    return PyModule_Create(&EmbModule);
}

int main(int argc, char* argv[])
{
    PyImport_AppendInittab("emb", &PyInit_emb);
    Py_Initialize();
    PyRun_SimpleString(
    "import emb\n"
    "print('This is Zero: ', emb.numargs())\n"
    );
    Py_Finalize();
    return 0;
}

Надеюсь, это также завершит ответ Неда.

2 голосов
/ 31 мая 2017

УВЕДОМЛЕНИЕ Проверка ошибок для API не предоставляется для наглядности;

В этом примере вставляется новая функция непосредственно в модуль python __buildin __ , позволяющий вызывать метод без схемы class.method. Просто замените mymodule на любой другой модуль по вашему желанию. 

PyObject* py_cb(PyObject *self, PyObject *args)
{
    const char *name = (const char *) PyCObject_AsVoidPtr(self);

    printf("%s\n", name);
    Py_RETURN_NONE;
}

int main(int argc, char *argv)
{
    PyObject        *mod, *modname, *dict, *fnc, *usrptr;
    const char      *mymodule = "__builtin__";
    PyMethodDef     *m;
    const char      *method = "hello_python";

    Py_Initialize();
    mod = PyImport_ImportModule(mymodule);
    modname = PyString_FromString(mymodule);
    dict = PyModule_GetDict(mod);
    m = (PyMethodDef *) calloc(1, sizeof(PyMethodDef));
    m->ml_name = strdup(method);
    m->ml_meth = py_cb;
    usrptr = PyCObject_FromVoidPtr("I'm am the hello_python!", NULL);
    fnc = PyCFunction_NewEx(m, usrptr, modname);
    PyDict_SetItemString(dict, method, fnc);
    ...

Когда скрипт Python выполняет hello_python , функция расширения py_cb покажет:

Я привет! Python!

self используется для отправки реального указателя, такого как контекст библиотеки, вместо этого const char * этого примера, теперь нужно изменить его на что-то интересное хотя.

2 голосов
/ 04 июля 2011

API Python не создан, чтобы сообщать вам, какую функцию он вызывает. Вы создали функцию, и она вызывает ее, API предполагает, что вы знаете, какую функцию написали. Вам нужно будет создать небольшую функцию-обертку для каждой из ваших функций Python. Лучший способ сделать это - зарегистрировать одну функцию C как одну функцию Python, которая принимает строку в качестве первого аргумента. Затем в своем слое Python создайте столько функций Python, сколько вам нужно, каждая из которых вызывает вашу функцию C с соответствующим строковым аргументом, определяющим, какую функцию вы действительно хотите вызвать.

Другая альтернатива - переосмыслить структуру вашего кода C и иметь столько функций, сколько вам нужно, каждая из которых вызывает ваш общий вспомогательный код для обработки параметров и т. Д.

0 голосов
/ 04 июля 2011

Я не знаю, можно ли это сделать напрямую из C-API. В худшем случае вы можете вызвать модуль traceback из C, чтобы получить имя вызывающего абонента.

...