src/python.c

   1 /* Python scripting support
   2  *
   3  * Copyright (c) 2009 Sadrul Habib Chowdhury (sadrul@users.sf.net)
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 3, or (at your option)
   8  * any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program (see the file COPYING); if not, write to the
  17  * Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA  02111-1301  USA
  19  *
  20  ****************************************************************
  21  */
  22 #include <sys/types.h>
  23
  24 #include "config.h"
  25 #include "screen.h"
  26 #include "script.h"
  27 #include <sys/stat.h>
  28 #include <unistd.h>
  29
  30 #include <stdio.h>
  31 #include <string.h>
  32 #include <stdlib.h>
  33
  34 #include "extern.h"
  35 #include "logfile.h"
  36
  37 #include <Python.h>
  38 #include <structmember.h>
  39
  40 extern struct win *windows;
  41
  42 /** Window {{{ */
  43 typedef struct
  44 {
  45   PyObject_HEAD
  46   struct win w;
  47 } PyWindow;
  48
  49 static PyMemberDef py_window_members[] =
  50 {
  51   {"title", T_STRING, offsetof(PyWindow, w) + offsetof(struct win, w_title), 0, "Title of the window"},
  52   {"number", T_INT, offsetof(PyWindow, w) + offsetof(struct win, w_number), 0, "The window number"},
  53   {NULL}
  54 };
  55
  56 static PyTypeObject PyWindowType =
  57 {
  58   PyObject_HEAD_INIT(NULL)
  59   .ob_size = 0,
  60   .tp_name = "screen.window",
  61   .tp_basicsize = sizeof(PyWindow),
  62   .tp_flags = Py_TPFLAGS_DEFAULT,
  63   .tp_doc = "Window object",
  64   .tp_methods = NULL,
  65   .tp_members = py_window_members
  66 };
  67
  68 static PyObject *
  69 PyWindow_FromWindow(struct win *w)
  70 {
  71   PyWindow *obj = PyWindowType.tp_alloc(&PyWindowType, 0);
  72   obj->w = *w;
  73   return (PyObject *)obj;
  74 }
  75
  76 /** }}} */
  77
  78 static PyObject *
  79 screen_windows(PyObject *self)
  80 {
  81   struct win *w = windows;
  82   int count = 0;
  83   for (; w; w = w->w_next)
  84     ++count;
  85   PyObject *tuple = PyTuple_New(count);
  86
  87   for (w = windows, count = 0; w; w = w->w_next, ++count)
  88     {
  89       PyTuple_SetItem(tuple, count, PyWindow_FromWindow(w));
  90     }
  91
  92   return tuple;
  93 }
  94
  95 const PyMethodDef py_methods[] = {
  96   {"windows", (PyCFunction)screen_windows, METH_NOARGS, NULL},
  97   {NULL, NULL, 0, NULL}
  98 };
  99
 100 static int
 101 SPyInit(void)
 102 {
 103   PyObject *m;
 104
 105   Py_Initialize();
 106
 107   PyType_Ready(&PyWindowType);
 108
 109   m = Py_InitModule3 ("screen", py_methods, NULL);
 110
 111   Py_INCREF(&PyWindowType);
 112   PyModule_AddObject(m, "Window", (PyObject *)&PyWindowType);
 113   return 0;
 114 }
 115
 116 static int
 117 SPyFinit(void)
 118 {
 119   /*Py_Finalize(); // Crashes -- why? */
 120   return 0;
 121 }
 122
 123 static int
 124 SPySource(const char *file, int async)
 125 {
 126   FILE *f = fopen(file, "rb");
 127   int ret = PyRun_SimpleFile(f, file);
 128   fclose(f);
 129
 130   if (ret == 0)
 131       return 1; /* Success */
 132
 133   if (PyErr_Occurred())
 134     {
 135       PyErr_Print();
 136       return 0;
 137     }
 138
 139   return 1;
 140 }
 141
 142 struct binding py_binding =
 143 {
 144   "python",
 145   0,
 146   0,
 147   SPyInit,
 148   SPyFinit,
 149   0,
 150   SPySource,
 151   0,
 152   0
 153 };
 154