Расширение Python C: Извлечь параметр из движка

Question

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

#include <iostream> 
#include <string> 
#include <Python.h> 
int RunTests() 
{ 
    Py_Initialize(); 

    PyRun_SimpleString("a=5"); 
    PyRun_SimpleString("b='Hello'"); 
    PyRun_SimpleString("c=1+2j"); 
    PyRun_SimpleString("d=[1,3,5,7,9]"); 

    //question here 

    Py_Finalize(); 
    return 0; 
} 

Мой вопрос: Как я могу извлечь параметры a,b,c,d в PyObject ы?

Answer 1

PyRun_SimpleString() выполняет код в контексте модуля __main__. Вы можете получить ссылку на этот модуль с помощью PyImport_AddModule(), получить словарь глобалам из этого модуля и подстановки переменных:

PyObject *main = PyImport_AddModule("__main__"); 
PyObject *globals = PyModule_GetDict(main); 
PyObject *a = PyDict_GetItemString(globals, "a"); 

Вместо того, чтобы использовать этот подход, вы могли бы быть лучше создать новый globals словарь и используя PyRun_String() для выполнения код в контексте этого globals Dict:

PyObject *globals = PyDict_New(); 
PyObject *a = PyRun_String("5", Py_single_input, globals, globals); 

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