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 extern "C" {
  12 // The initializer in openbox_wrap.cc
  13 extern void init_openbox(void);
  14 }
  15
  16 namespace ob {
  17
  18 static PyObject *obdict = NULL;
  19
  20 void python_init(char *argv0)
  21 {
  22   std::string path;
  23
  24   // start the python engine
  25   //Py_SetProgramName(argv0);
  26   //Py_Initialize();
  27   // initialize the C python module
  28   //init_openbox();
  29   // include the openbox directories for python scripts in the sys path
  30 //  PyRun_SimpleString("import sys");
  31 printf("SCRIPTDIR=%s\n", SCRIPTDIR);
  32 printf("1 getenv(DISPLAY)=%s\n", getenv("DISPLAY"));
  33   path = "sys.path";
  34 printf("2 getenv(DISPLAY)=%s\n", getenv("DISPLAY"));
  35   path = "sys.path.append('";
  36 printf("3 getenv(DISPLAY)=%s\n", getenv("DISPLAY"));
  37   path += SCRIPTDIR;
  38   path += "')";
  39   PyRun_SimpleString(const_cast<char*>(path.c_str()));
  40   path = "sys.path.append('";
  41   path += otk::expandTilde("~/.openbox/python");
  42   path += "')";
  43   PyRun_SimpleString(const_cast<char*>(path.c_str()));
  44   // import the otk and openbox modules into the main namespace
  45   PyRun_SimpleString("from openbox import *;");
  46   // set up convenience global variables
  47   PyRun_SimpleString("openbox = Openbox_instance()");
  48
  49   // set up access to the python global variables
  50   PyObject *obmodule = PyImport_AddModule("__main__");
  51   obdict = PyModule_GetDict(obmodule);
  52 }
  53
  54 void python_destroy()
  55 {
  56   Py_DECREF(obdict);
  57 }
  58
  59 bool python_exec(const std::string &path)
  60 {
  61   FILE *rcpyfd = fopen(path.c_str(), "r");
  62   if (!rcpyfd) {
  63     printf("failed to load python file %s\n", path.c_str());
  64     return false;
  65   }
  66   PyRun_SimpleFile(rcpyfd, const_cast<char*>(path.c_str()));
  67   fclose(rcpyfd);
  68   return true;
  69 }
  70
  71 bool python_get_long(const char *name, long *value)
  72 {
  73   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
  74   if (!(val && PyLong_Check(val))) return false;
  75
  76   *value = PyLong_AsLong(val);
  77   return true;
  78 }
  79
  80 bool python_get_string(const char *name, std::string *value)
  81 {
  82   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
  83   if (!(val && PyString_Check(val))) return false;
  84
  85   *value = PyString_AsString(val);
  86   return true;
  87 }
  88
  89 bool python_get_stringlist(const char *name, std::vector<std::string> *value)
  90 {
  91   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
  92   if (!(val && PyList_Check(val))) return false;
  93
  94   for (int i = 0, end = PyList_Size(val); i < end; ++i) {
  95     PyObject *str = PyList_GetItem(val, i);
  96     if (PyString_Check(str))
  97       value->push_back(PyString_AsString(str));
  98   }
  99   return true;
 100 }
 101
 102 // ************************************* //
 103 // Stuff for calling from Python scripts //
 104 // ************************************* //
 105
 106 PyObject *mbind(const std::string &button, ob::MouseContext context,
 107                 ob::MouseAction action, PyObject *func)
 108 {
 109   if (!PyCallable_Check(func)) {
 110     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
 111     return NULL;
 112   }
 113
 114   if (!ob::Openbox::instance->bindings()->addButton(button, context,
 115                                                     action, func)) {
 116     PyErr_SetString(PyExc_RuntimeError,"Unable to add binding.");
 117     return NULL;
 118   }
 119   Py_INCREF(Py_None); return Py_None;
 120 }
 121
 122 PyObject *ebind(ob::EventAction action, PyObject *func)
 123 {
 124   if (!PyCallable_Check(func)) {
 125     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
 126     return NULL;
 127   }
 128
 129   if (!ob::Openbox::instance->bindings()->addEvent(action, func)) {
 130     PyErr_SetString(PyExc_RuntimeError,"Unable to add binding.");
 131     return NULL;
 132   }
 133   Py_INCREF(Py_None); return Py_None;
 134 }
 135
 136 PyObject *kbind(PyObject *keylist, ob::KeyContext context, PyObject *func)
 137 {
 138   if (!PyCallable_Check(func)) {
 139     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
 140     return NULL;
 141   }
 142   if (!PyList_Check(keylist)) {
 143     PyErr_SetString(PyExc_TypeError, "Invalid keylist. Not a list.");
 144     return NULL;
 145   }
 146
 147   ob::OBBindings::StringVect vectkeylist;
 148   for (int i = 0, end = PyList_Size(keylist); i < end; ++i) {
 149     PyObject *str = PyList_GetItem(keylist, i);
 150     if (!PyString_Check(str)) {
 151       PyErr_SetString(PyExc_TypeError,
 152                      "Invalid keylist. It must contain only strings.");
 153       return NULL;
 154     }
 155     vectkeylist.push_back(PyString_AsString(str));
 156   }
 157
 158   (void)context; // XXX use this sometime!
 159   if (!ob::Openbox::instance->bindings()->addKey(vectkeylist, func)) {
 160     PyErr_SetString(PyExc_RuntimeError,"Unable to add binding.");
 161     return NULL;
 162   }
 163   Py_INCREF(Py_None); return Py_None;
 164 }
 165
 166 PyObject *kunbind(PyObject *keylist, PyObject *func)
 167 {
 168   if (!PyList_Check(keylist)) {
 169     PyErr_SetString(PyExc_TypeError, "Invalid keylist. Not a list.");
 170     return NULL;
 171   }
 172   if (!PyCallable_Check(func)) {
 173     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
 174     return NULL;
 175   }
 176
 177   ob::OBBindings::StringVect vectkeylist;
 178   for (int i = 0, end = PyList_Size(keylist); i < end; ++i) {
 179     PyObject *str = PyList_GetItem(keylist, i);
 180     if (!PyString_Check(str)) {
 181       PyErr_SetString(PyExc_TypeError,
 182                      "Invalid keylist. It must contain only strings.");
 183       return NULL;
 184     }
 185     vectkeylist.push_back(PyString_AsString(str));
 186   }
 187
 188   if (!ob::Openbox::instance->bindings()->removeKey(vectkeylist, func)) {
 189       PyErr_SetString(PyExc_RuntimeError, "Could not remove callback.");
 190       return NULL;
 191   }
 192   Py_INCREF(Py_None); return Py_None;
 193 }
 194
 195 void kunbind_all()
 196 {
 197   ob::Openbox::instance->bindings()->removeAllKeys();
 198 }
 199
 200 void set_reset_key(const std::string &key)
 201 {
 202   ob::Openbox::instance->bindings()->setResetKey(key);
 203 }
 204
 205 PyObject *send_client_msg(Window target, int type, Window about,
 206                           long data, long data1, long data2,
 207                           long data3, long data4)
 208 {
 209   if (type < 0 || type >= otk::OBProperty::NUM_ATOMS) {
 210       PyErr_SetString(PyExc_TypeError,
 211                      "Invalid atom type. Must be from otk::OBProperty::Atoms");
 212       return NULL;
 213   }
 214
 215   XEvent e;
 216   e.xclient.type = ClientMessage;
 217   e.xclient.format = 32;
 218   e.xclient.message_type =
 219     Openbox::instance->property()->atom((otk::OBProperty::Atoms)type);
 220   e.xclient.window = about;
 221   e.xclient.data.l[0] = data;
 222   e.xclient.data.l[1] = data1;
 223   e.xclient.data.l[2] = data2;
 224   e.xclient.data.l[3] = data3;
 225   e.xclient.data.l[4] = data4;
 226
 227   XSendEvent(otk::OBDisplay::display, target, false,
 228              SubstructureRedirectMask | SubstructureNotifyMask,
 229              &e);
 230   Py_INCREF(Py_None); return Py_None;
 231 }
 232
 233 }