aboutsummaryrefslogtreecommitdiffstats
path: root/plugin_python.c
blob: 7b61096b59003ff188bdf5fc8b31176ed135f16c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
/* $Id: plugin_python.c,v 1.2 2005/05/03 11:13:24 reinelt Exp $
 *
 * Python plugin
 *
 * Copyright 2005 Dan Fritz  
 * Copyright 2005 The LCD4Linux Team <lcd4linux-devel@users.sourceforge.net>
 *
 * This file is part of LCD4Linux.
 *
 * LCD4Linux is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * LCD4Linux is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 *
 * $Log: plugin_python.c,v $
 * Revision 1.2  2005/05/03 11:13:24  reinelt
 * rearranged autoconf a bit,
 * libX11 will be linked only if really needed (i.e. when the X11 driver has been selected)
 * plugin_python filled with life
 *
 * Revision 1.1  2005/05/02 10:29:20  reinelt
 * preparations for python bindings and python plugin
 *
 */

/* 
 * exported functions:
 *
 * int plugin_init_python (void)
 *  adds a python interpreter
 *
 */

#include <Python.h>

#include "config.h"

#include <stdlib.h>
#include <stdio.h>

#include "debug.h"
#include "plugin.h"
#include "hash.h"

/* 
 * Executes a python function specified by function name and module.
 *
 * This method is more or less a copy of an example found in the python 
 * documentation. Kudos goes to Guido van Rossum and Fred L. Drake.
 * 
 * Returns a char* directly from PyString_AsString() !!! DO NOT DEALLOCATE !!!
 */
static const char* 
pyt_exec_str(const char* module, const char* function, int argc, const char* argv[]) {

  PyObject *pName, *pModule, *pDict, *pFunc;
  PyObject *pArgs, *pValue;
  const char * rv = NULL;
  int i;

  pName = PyString_FromString(module);
  /* Error checking of pName left out */

  pModule = PyImport_Import(pName);
  Py_DECREF(pName);

  if (pModule != NULL) {
    pDict = PyModule_GetDict(pModule);
    /* pDict is a borrowed reference */

    pFunc = PyDict_GetItemString(pDict, function);
    /* pFun: Borrowed reference */

    if (pFunc && PyCallable_Check(pFunc)) {
      pArgs = PyTuple_New(argc);
      for (i = 0; i < argc; ++i) {
        pValue = PyString_FromString(argv[i]);
        if (!pValue) {
          Py_DECREF(pArgs);
          Py_DECREF(pModule);
          fprintf(stderr, "Cannot convert argument %s\n", argv[i]);
          return NULL;
        }
        /* pValue reference stolen here: */
        PyTuple_SetItem(pArgs, i, pValue);
      }
      pValue = PyObject_CallObject(pFunc, pArgs);
      Py_DECREF(pArgs);
      if (pValue != NULL) {
        rv = PyString_AsString(pValue);
        //printf("Result of call: %s\n", rv);
        Py_DECREF(pValue);
        return rv;
      }
      else {
        Py_DECREF(pModule);
        PyErr_Print();
        fprintf(stderr,"Call failed\n");
        return NULL;
      }
      /* pDict and pFunc are borrowed and must not be Py_DECREF-ed */
    }
    else {
      if (PyErr_Occurred())
        PyErr_Print();
      fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]);
    }
    Py_DECREF(pModule);
  }
  else {
    PyErr_Print();
    fprintf(stderr, "Failed to load \"%s\"\n", argv[1]);
    return NULL;
  }
  return NULL;
}

static int python_cleanup_responsibility = 0;
    
static void my_exec (RESULT *result, RESULT *module, RESULT *function, RESULT *arg )
{ 
  const char* args[] = {R2S(arg)};
  const char* value = pyt_exec_str(R2S(module),R2S(function),1,args);
  SetResult(&result, R_STRING, value); 
}

int plugin_init_python (void)
{
  if (!Py_IsInitialized()) {
    Py_Initialize();
    python_cleanup_responsibility = 1;
  }
  AddFunction ("python::exec", 3, my_exec);
  return 0;
}

void plugin_exit_python (void) 
{
  /* Make sure NOT to call Py_Finalize() When (and if) the entire lcd4linux process 
   * is started from inside python
   */
  if (python_cleanup_responsibility) {
    python_cleanup_responsibility = 0;
    Py_Finalize();
  }
}