new swig build system. much better. yay.
[dana/openbox.git] / src / python.cc
1 // -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 2; -*-
2
3 #include "python.hh"
4 #include "openbox.hh"
5 #include "actions.hh"
6 #include "python.hh"
7 #include "bindings.hh"
8 #include "otk/display.hh"
9 #include "otk/util.hh"
10
11 namespace ob {
12
13 static PyObject *obdict = NULL;
14
15 void python_init(char *argv0)
16 {
17   // start the python engine
18   Py_SetProgramName(argv0);
19   Py_Initialize();
20   // prepend the openbox directories for python scripts to the sys path
21   PyRun_SimpleString("import sys");
22   PyRun_SimpleString("sys.path.insert(0, '" SCRIPTDIR "')");
23   PyRun_SimpleString(const_cast<char*>(("sys.path.insert(0, '" +
24                                         otk::expandTilde("~/.openbox/python") +
25                                         "')").c_str()));
26   //PyRun_SimpleString("import ob; import otk; import config;");
27   PyRun_SimpleString("import config;");
28   // set up convenience global variables
29   //PyRun_SimpleString("ob.openbox = ob.Openbox_instance()");
30   //PyRun_SimpleString("otk.display = otk.Display_instance()");
31
32   // set up access to the python global variables
33   PyObject *obmodule = PyImport_AddModule("config");
34   obdict = PyModule_GetDict(obmodule);
35 }
36
37 void python_destroy()
38 {
39   Py_Finalize();
40 }
41
42 bool python_exec(const std::string &path)
43 {
44   FILE *rcpyfd = fopen(path.c_str(), "r");
45   if (!rcpyfd) {
46     printf("Failed to load python file %s\n", path.c_str());
47     return false;
48   }
49   PyRun_SimpleFile(rcpyfd, const_cast<char*>(path.c_str()));
50   fclose(rcpyfd);
51   return true;
52 }
53
54 bool python_get_long(const char *name, long *value)
55 {
56   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
57   if (!(val && PyInt_Check(val))) return false;
58   
59   *value = PyInt_AsLong(val);
60   return true;
61 }
62
63 bool python_get_string(const char *name, otk::ustring *value)
64 {
65   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
66   if (!(val && PyString_Check(val))) return false;
67   
68   *value = PyString_AsString(val);
69   return true;
70 }
71
72 bool python_get_stringlist(const char *name, std::vector<otk::ustring> *value)
73 {
74   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
75   if (!(val && PyList_Check(val))) return false;
76
77   for (int i = 0, end = PyList_Size(val); i < end; ++i) {
78     PyObject *str = PyList_GetItem(val, i);
79     if (PyString_Check(str))
80       value->push_back(PyString_AsString(str));
81   }
82   return true;
83 }
84
85 }