b5ae486486efd5af9a34a529d4b95c0be7ac5dd2
[enigma2.git] / lib / python / python.cpp
1 #include <lib/python/python.h>
2 #include <lib/base/eerror.h>
3                 /* avoid warnigs :) */
4 #undef _POSIX_C_SOURCE
5 #define _POSIX_C_SOURCE 200112L
6 #include <Python.h>
7
8 extern "C" void init_enigma();
9 extern void bsodFatal();
10
11 DEFINE_REF(TestObj);
12
13 TestObj::TestObj()
14 {
15         eDebug("create %p", this);
16 }
17
18 TestObj::~TestObj()
19 {
20         eDebug("destroy %p", this);
21 }
22
23 #if 0
24 ePyObject::ePyObject(void *ptr): m_object(ptr)
25 {
26         Py_XINCREF((PyObject*)ptr);
27 }
28
29 ePyObject::ePyObject(ePyObject &p)
30 {
31         m_object = p.m_object;
32         Py_XINCREF((PyObject*)m_object);
33 }
34
35 ePyObject::ePyObject(): m_object(0)
36 {
37 }
38
39 ePyObject::~ePyObject()
40 {
41         Py_XDECREF((PyObject*)m_object);
42 }
43
44 ePyObject &ePyObject::operator=(ePyObject &p)
45 {
46         Py_XDECREF((PyObject*)m_object);
47         m_object = p.m_object;
48         Py_XINCREF((PyObject*)m_object);
49         return *this;
50 }
51
52 ePyObject &ePyObject::operator=(void *object)
53 {
54         Py_XDECREF((PyObject*)m_object);
55         m_object = object;
56         Py_XINCREF((PyObject*)m_object);
57         return *this;
58 }
59 #endif
60
61 ePython::ePython()
62 {
63 //      Py_VerboseFlag = 1;
64         
65         Py_Initialize();
66         
67         init_enigma();
68 }
69
70 ePython::~ePython()
71 {
72         Py_Finalize();
73 }
74
75 int ePython::execute(const std::string &pythonfile, const std::string &funcname)
76 {
77         PyObject *pName, *pModule, *pDict, *pFunc, *pArgs, *pValue;
78         
79         pName = PyString_FromString(pythonfile.c_str());
80         
81         pModule = PyImport_Import(pName);
82         Py_DECREF(pName);
83         
84         if (pModule != NULL)
85         {
86                 pDict = PyModule_GetDict(pModule);
87                 
88                 pFunc = PyDict_GetItemString(pDict, funcname.c_str());
89                 
90                 if (pFunc && PyCallable_Check(pFunc))
91                 {
92                         pArgs = PyTuple_New(0);
93                                 // implement arguments..
94                         pValue = PyObject_CallObject(pFunc, pArgs);
95                         Py_DECREF(pArgs);
96                         if (pValue != NULL)
97                         {
98                                 printf("Result of call: %ld\n", PyInt_AsLong(pValue));
99                                 Py_DECREF(pValue);
100                         } else
101                         {
102                                 Py_DECREF(pModule);
103                                 PyErr_Print();
104                                 return 1;
105                         }
106                 }
107         } else
108         {
109                 if (PyErr_Occurred())
110                         PyErr_Print();
111                 return 1;
112         }
113         return 0;
114 }
115
116 int ePython::call(PyObject *pFunc, PyObject *pArgs)
117 {
118         int res = -1;
119         PyObject *pValue;
120         if (pFunc && PyCallable_Check(pFunc))
121         {
122                 pValue = PyObject_CallObject(pFunc, pArgs);
123                 if (pValue != NULL)
124                 {
125                         if (PyInt_Check(pValue))
126                                 res = PyInt_AsLong(pValue);
127                         else
128                                 res = 0;
129                         Py_DECREF(pValue);
130                 } else
131                 {
132                         PyErr_Print();
133                         bsodFatal();
134                 }
135         }
136         return res;
137 }
138
139 PyObject *ePython::resolve(const std::string &pythonfile, const std::string &funcname)
140 {
141         PyObject *pName, *pModule, *pDict, *pFunc;
142
143         pName = PyString_FromString(pythonfile.c_str());
144
145         pModule = PyImport_Import(pName);
146         Py_DECREF(pName);
147         
148         if (pModule != NULL)
149         {
150                 pDict = PyModule_GetDict(pModule);
151                 pFunc = PyDict_GetItemString(pDict, funcname.c_str());
152                 Py_XINCREF(pFunc);
153                 Py_DECREF(pModule);
154                 return pFunc;
155         } else
156         {
157                 if (PyErr_Occurred())
158                         PyErr_Print();
159                 return 0;
160         }
161 }