Python встроен в CPP: как вернуть данные на CPP

Во время работы над проектом на С++ я искал стороннюю библиотеку для чего-то, что не является моим основным бизнесом. Я нашел действительно хорошую библиотеку, делая то, что нужно, но написан на Python. Я решил поэкспериментировать с вложением кода Python в С++, используя библиотеку Boost.Python.

Код С++ выглядит примерно так:

#include <string>
#include <iostream>
#include <boost/python.hpp>

using namespace boost::python;

int main(int, char **) 
{
    Py_Initialize();

    try 
    {
        object module((handle<>(borrowed(PyImport_AddModule("__main__")))));

        object name_space = module.attr("__dict__");
        object ignored = exec("from myModule import MyFunc\n"
                          "MyFunc(\"some_arg\")\n",
                          name_space);

        std::string res = extract<std::string>(name_space["result"]);
    } 
    catch (error_already_set) 
    {
        PyErr_Print();
    }

    Py_Finalize();
    return 0;
}

A (очень) упрощенная версия кода Python выглядит следующим образом:

import thirdparty

def MyFunc(some_arg):
    result = thirdparty.go()
    print result

Теперь проблема заключается в следующем: "MyFunc" отлично работает, я вижу печать "результата". То, что я не могу сделать, это прочитать "результат" из кода на С++. Команда extract никогда не находит "результат" в любом пространстве имен. Я попытался определить "результат" как глобальный, я даже попытался вернуть кортеж, но я не могу заставить его работать.

Ответ 1

Прежде всего, измените свою функцию на return значение. print ing это усложнит ситуацию, так как вы хотите вернуть значение. Предположим, что ваш MyModule.py выглядит следующим образом:

import thirdparty

def MyFunc(some_arg):
    result = thirdparty.go()
    return result

Теперь, чтобы сделать то, что вы хотите, вы должны выйти за рамки базового внедрения, о чем говорит . Вот полный код для запуска вашей функции:

#include <Python.h>

int
main(int argc, char *argv[])
{
    PyObject *pName, *pModule, *pFunc;
    PyObject *pArgs, *pArg, *pResult;
    int i;

    Py_Initialize();
    pName = PyString_FromString("MyModule.py");
    /* Error checking of pName left out as exercise */

    pModule = PyImport_Import(pName);
    Py_DECREF(pName);

    if (pModule != NULL) {
        pFunc = PyObject_GetAttrString(pModule, "MyFunc");
        /* pFunc is a new reference */

        if (pFunc) {
            pArgs = PyTuple_New(0);
            pArg = PyString_FromString("some parameter")
            /* pArg reference stolen here: */
            PyTuple_SetItem(pArgs, 0, pArg);
            pResult = PyObject_CallObject(pFunc, pArgs);
            Py_DECREF(pArgs);
            if (pResult != NULL) {
                printf("Result of call: %s\n", PyString_AsString(pResult));
                Py_DECREF(pResult);
            }
            else {
                Py_DECREF(pFunc);
                Py_DECREF(pModule);
                PyErr_Print();
                fprintf(stderr,"Call failed\n");
                return 1;
            }
        }
        else {
            if (PyErr_Occurred())
                PyErr_Print();
            fprintf(stderr, "Cannot find function");
        }
        Py_XDECREF(pFunc);
        Py_DECREF(pModule);
    }
    else {
        PyErr_Print();
        fprintf(stderr, "Failed to load module");
        return 1;
    }
    Py_Finalize();
    return 0;
}

Ответ 2

Основываясь на TΖΩΤΖΙΟΥ, ответы Джоша и Носкло, я, наконец, получил его работу с помощью boost.python:

Python:

import thirdparty

def MyFunc(some_arg):
    result = thirdparty.go()
    return result

С++:

#include <string>
#include <iostream>
#include <boost/python.hpp>

using namespace boost::python;

int main(int, char **) 
{
    Py_Initialize();

    try 
    {
        object module = import("__main__");
        object name_space = module.attr("__dict__");
        exec_file("MyModule.py", name_space, name_space);

        object MyFunc = name_space["MyFunc"];
        object result = MyFunc("some_args");

        // result is a dictionary
        std::string val = extract<std::string>(result["val"]);
    } 
    catch (error_already_set) 
    {
        PyErr_Print();
    }

    Py_Finalize();
    return 0;
}

Некоторые важные моменты:

Я изменил 'exec' на 'exec_file' из удобства, он также работает с plain 'exec'.
Основная причина, по которой это не удалось, заключается в том, что i не передал "local" name_sapce, чтобы 'exec' или 'exec_file' - теперь это исправлено путем пропуска имени name_space дважды.
Если функция python возвращается строки unicode, они не конвертируется в 'std::string', поэтому i пришлось суффиктировать все строки python с '.encode(' ASCII ',' ignore ')'.

Ответ 3

Я думаю, что вам нужен либо PyObject_CallObject(<py function>, <args>), который возвращает возвращаемое значение функции, которую вы вызываете как PyObject, или PyRun_String(<expression>, Py_eval_input, <globals>, <locals>), которая оценивает одно выражение и возвращает его результат.

Ответ 4

Вы должны иметь возможность вернуть результат из MyFunc, который затем окажется в переменной, которую вы в настоящее время вызываете "проигнорировано". Это устраняет необходимость доступа к нему любым другим способом.