在C ++中获取Python函数参数名称

时间:2018-05-31 05:48:45

标签: python c++ embedding

我正在尝试获取从C ++调用的Python函数的参数名称。一点背景:

我有一个C ++应用程序,需要调用Python函数,按名称传递参数。到目前为止,我已经能够通过解析Python模块的.py文件来做到这一点;但我希望能够处理.pyc文件。

理想情况下,这可以通过Python C API完成(而不是尝试反编译字节码),但我看不到任何明显的方法。有很多方法可以在Python中执行此操作,例如有检查但不是用C ++。

有人知道可能的解决方案吗? TIA。

1 个答案:

答案 0 :(得分:1)

如果您想从python中执行此操作,则表示您将使用inspect模块。为什么不从C API调用inspect模块?

以下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;
}