Получение имен аргументов функции Python в C ++ - PullRequest
Новая
       10

Получение имен аргументов функции Python в C ++

0 голосов
/ 31 мая 2018

Я пытаюсь получить имена аргументов функции Python, вызываемой из C ++.Немного предыстории:

У меня есть приложение на C ++, которому нужно вызывать функции Python, передавая им аргументы по имени.До сих пор я был в состоянии сделать это путем анализа файла .py модуля Python;однако я хотел бы иметь возможность обрабатывать также файлы .pyc.

В идеале это должно быть сделано с помощью Python C API (а не пытаться декомпилировать байт-код), но я не вижу никакого очевидного способаделать это.Есть способы сделать это в Python, например, с помощью inspect, но не в C ++.

Кто-нибудь знает возможное решение?ТИА.

1 Ответ

0 голосов
/ 31 мая 2018

Вы сказали, что будете использовать модуль inspect, если хотите сделать это с python.Почему бы не вызвать модуль inspect из C API?

Следующий код C печатает все аргументы функции python.Это должен быть действительный код C ++.Я не использую Python C API очень часто, поэтому, пожалуйста, скажите, можно ли улучшить некоторые вещи. 

#include <Python.h>
#include <stdio.h>

int main(int argc, char* argv[])
{
    wchar_t* const program = Py_DecodeLocale(argv[0], nullptr);
    Py_SetProgramName(program);
    Py_Initialize();

    // Define a python function f with argument names x and y.
    // >>> def f(x, y): return x*y
    PyRun_SimpleString("def f(x, y): return x*y\n");

    // Get the function as python object.
    // >>> import sys
    // >>> f_function = sys.modules["__main__"].f
    PyObject* const sys_module_name = PyUnicode_DecodeFSDefault("sys");
    PyObject* const sys_module = PyImport_Import(sys_module_name);
    PyObject* const modules_dict = PyObject_GetAttrString(sys_module, "modules");
    PyObject* const main_name = PyUnicode_DecodeFSDefault("__main__");
    PyObject* const main_module = PyDict_GetItem(modules_dict, main_name);
    PyObject* const f_function = PyObject_GetAttrString(main_module, "f");

    // Get the inspect.getargspec function.
    // >>> import inspect
    // >>> getargspec_function = inspect.getargspec
    PyObject* const inspect_module_name = PyUnicode_DecodeFSDefault("inspect");
    PyObject* const inspect_module = PyImport_Import(inspect_module_name);
    PyObject* const getargspec_function = PyObject_GetAttrString(inspect_module, "getargspec");

    // Call the inspect.getargspec function.
    // >>> argspec = getargspec_function(f_function)
    PyObject* const argspec_call_args = PyTuple_New(1);
    PyTuple_SetItem(argspec_call_args, 0, f_function);
    PyObject* const argspec = PyObject_CallObject(getargspec_function, argspec_call_args);

    // Get args from argspec.
    // >>> f_args = argspec.args
    PyObject* const f_args = PyObject_GetAttrString(argspec, "args");

    // f_args now holds a python list with all arguments of f.

    // As example usage, you can print the arguments:
    // >>> for i, a in enumerate(f_args):
    // ...     print("Repr of arg", i, "is", repr(a))
    Py_ssize_t const num_args = PyList_Size(f_args);
    for (Py_ssize_t i = 0; i < num_args; ++i)
    {
        PyObject* const arg = PyList_GetItem(f_args, i);
        PyObject* const arg_repr = PyObject_Repr(arg);
        PyObject* const arg_str = PyUnicode_AsASCIIString(arg_repr);
        char const* const arg_c_str = PyBytes_AS_STRING(arg_str);
        printf("Repr of arg %ld is %s\n", i, arg_c_str);
    }

    Py_Finalize();
    PyMem_RawFree(program);

    return 0;
}
...