Added PYTHON3 define

This commit is contained in:
James Booth
2016-07-14 00:00:46 +01:00
parent 10814b044d
commit ace2715cf7
4 changed files with 168 additions and 3 deletions

View File

@@ -32,6 +32,8 @@
*
*/
#include "config.h"
#include <Python.h>
#include <frameobject.h>
@@ -128,7 +130,11 @@ python_api_register_command(PyObject *self, PyObject *args)
Py_ssize_t i = 0;
for (i = 0; i < len; i++) {
PyObject *item = PyList_GetItem(synopsis, i);
#ifdef PYTHON3
char *c_item = PyBytes_AS_STRING(PyUnicode_AsUTF8String(item));
#else
char *c_item = PyString_AsString(item);
#endif
c_synopsis[i] = c_item;
}
c_synopsis[len] = NULL;
@@ -143,10 +149,18 @@ python_api_register_command(PyObject *self, PyObject *args)
return Py_BuildValue("");
}
PyObject *arg = PyList_GetItem(item, 0);
#ifdef PYTHON3
char *c_arg = PyBytes_AS_STRING(PyUnicode_AsUTF8String(arg));
#else
char *c_arg = PyString_AsString(arg);
#endif
PyObject *desc = PyList_GetItem(item, 1);
char *c_desc = PyString_AsString(desc);
#ifdef PYTHON3
char *c_desc = PyBytes_AS_STRING(PyUnicode_AsUTF8String(desc));
#else
char *c_desc = PyString_AsString(desc);
#endif
c_arguments[i][0] = c_arg;
c_arguments[i][1] = c_desc;
}
@@ -159,7 +173,11 @@ python_api_register_command(PyObject *self, PyObject *args)
i = 0;
for (i = 0; i < len; i++) {
PyObject *item = PyList_GetItem(examples, i);
#ifdef PYTHON3
char *c_item = PyBytes_AS_STRING(PyUnicode_AsUTF8String(item));
#else
char *c_item = PyString_AsString(item);
#endif
c_examples[i] = c_item;
}
c_examples[len] = NULL;
@@ -218,7 +236,11 @@ python_api_completer_add(PyObject *self, PyObject *args)
Py_ssize_t i = 0;
for (i = 0; i < len; i++) {
PyObject *item = PyList_GetItem(items, i);
#ifdef PYTHON3
char *c_item = PyBytes_AS_STRING(PyUnicode_AsUTF8String(item));
#else
char *c_item = PyString_AsString(item);
#endif
c_items[i] = c_item;
}
c_items[len] = NULL;
@@ -251,7 +273,11 @@ python_api_completer_remove(PyObject *self, PyObject *args)
Py_ssize_t i = 0;
for (i = 0; i < len; i++) {
PyObject *item = PyList_GetItem(items, i);
#ifdef PYTHON3
char *c_item = PyBytes_AS_STRING(PyUnicode_AsUTF8String(item));
#else
char *c_item = PyString_AsString(item);
#endif
c_items[i] = c_item;
}
c_items[len] = NULL;
@@ -783,7 +809,7 @@ python_window_callback(PluginWindowCallback *window_callback, char *tag, char *l
static PyMethodDef apiMethods[] = {
{ "cons_alert", python_api_cons_alert, METH_NOARGS, "Highlight the console window in the status bar." },
{ "cons_show", python_api_cons_show, METH_VARARGS, "Print a line to the console." },
{ "cons_show", (PyCFunction)python_api_cons_show, METH_VARARGS, "Print a line to the console." },
{ "cons_show_themed", python_api_cons_show_themed, METH_VARARGS, "Print a themed line to the console" },
{ "cons_bad_cmd_usage", python_api_cons_bad_cmd_usage, METH_VARARGS, "Show invalid command message in console" },
{ "register_command", python_api_register_command, METH_VARARGS, "Register a command." },
@@ -819,10 +845,25 @@ static PyMethodDef apiMethods[] = {
{ NULL, NULL, 0, NULL }
};
#ifdef PYTHON3
static struct PyModuleDef profModule =
{
PyModuleDef_HEAD_INIT,
"prof", /* name of module */
"", /* module documentation, may be NULL */
-1, /* size of per-interpreter state of the module, or -1 if the module keeps state in global variables. */
apiMethods
};
#endif
void
python_api_init(void)
{
#ifdef PYTHON3
PyModule_Create(&profModule);
#else
Py_InitModule("prof", apiMethods);
#endif
}
static char*
@@ -830,7 +871,11 @@ _python_plugin_name(void)
{
PyThreadState *ts = PyThreadState_Get();
PyFrameObject *frame = ts->frame;
#ifdef PYTHON3
char const *filename = PyBytes_AS_STRING(PyUnicode_AsUTF8String(frame->f_code->co_filename));
#else
char const* filename = PyString_AsString(frame->f_code->co_filename);
#endif
gchar **split = g_strsplit(filename, "/", 0);
char *plugin_name = strdup(split[g_strv_length(split)-1]);
g_strfreev(split);