X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fcpython.h;h=4b8aa72143167864ff575147c5ef19949997ab50;hb=204cf4f8f68cadeac474a75584f27b7339aca4c4;hp=661bf6a20c0e8fce3e8c47a9ee50a2b216cbe82e;hpb=0bd574a28c06090f0c4b056f8a78e455455a6330;p=collectd.git

diff --git a/src/cpython.h b/src/cpython.h
index 661bf6a2..4b8aa721 100644
--- a/src/cpython.h
+++ b/src/cpython.h
@@ -24,6 +24,10 @@
  *   Sven Trenkel <collectd at semidefinite.de>  
  **/
 
+/* Some python versions don't include this by default. */
+
+#include <longintrepr.h>
+
 /* These two macros are basicly Py_BEGIN_ALLOW_THREADS and Py_BEGIN_ALLOW_THREADS
  * from the other direction. If a Python thread calls a C function
  * Py_BEGIN_ALLOW_THREADS is used to allow other python threads to run because
@@ -74,6 +78,91 @@
 # define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
 #endif
 
+/* This macro is a shortcut for calls like
+ * x = PyObject_Repr(x);
+ * This can't be done like this example because this would leak
+ * a reference the the original x and crash in case of x == NULL.
+ * This calling syntax is less than elegant but it works, saves
+ * a lot of lines and avoids potential refcount errors. */
+
+#define CPY_SUBSTITUTE(func, a, ...) do {\
+	if ((a) != NULL) {\
+		PyObject *__tmp = (a);\
+		(a) = func(__VA_ARGS__);\
+		Py_DECREF(__tmp);\
+	}\
+} while(0)
+
+/* Python3 compatibility layer. To keep the actual code as clean as possible
+ * do a lot of defines here. */
+
+#if PY_MAJOR_VERSION >= 3
+#define IS_PY3K
+#endif
+
+#ifdef IS_PY3K
+
+#define PyInt_FromLong PyLong_FromLong
+#define CPY_INIT_TYPE         PyVarObject_HEAD_INIT(NULL, 0)
+#define IS_BYTES_OR_UNICODE(o) (PyUnicode_Check(o) || PyBytes_Check(o))
+#define CPY_STRCAT_AND_DEL(a, b) do {\
+	CPY_STRCAT((a), (b));\
+	Py_XDECREF((b));\
+} while (0)
+static inline void CPY_STRCAT(PyObject **a, PyObject *b) {
+	PyObject *ret;
+	
+	if (!a || !*a)
+		return;
+	
+	ret = PyUnicode_Concat(*a, b);
+	Py_DECREF(*a);
+	*a = ret;
+}
+
+#else
+
+#define CPY_INIT_TYPE         PyObject_HEAD_INIT(NULL) 0,
+#define IS_BYTES_OR_UNICODE(o) (PyUnicode_Check(o) || PyString_Check(o))
+#define CPY_STRCAT_AND_DEL PyString_ConcatAndDel
+#define CPY_STRCAT PyString_Concat
+
+#endif
+
+static inline const char *cpy_unicode_or_bytes_to_string(PyObject **o) {
+	if (PyUnicode_Check(*o)) {
+		PyObject *tmp;
+		tmp = PyUnicode_AsEncodedString(*o, NULL, NULL); /* New reference. */
+		if (tmp == NULL)
+			return NULL;
+		Py_DECREF(*o);
+		*o = tmp;
+	}
+#ifdef IS_PY3K
+	return PyBytes_AsString(*o);
+#else
+	return PyString_AsString(*o);
+#endif
+}
+
+static inline PyObject *cpy_string_to_unicode_or_bytes(const char *buf) {
+#ifdef IS_PY3K
+/* Python3 preferrs unicode */
+	PyObject *ret;
+	ret = PyUnicode_Decode(buf, strlen(buf), NULL, NULL);
+	if (ret != NULL)
+		return ret;
+	PyErr_Clear();
+	return PyBytes_FromString(buf);
+#else
+	return PyString_FromString(buf);
+#endif	
+}
+
+void cpy_log_exception(const char *context);
+
+/* Python object declarations. */
+
 typedef struct {
 	PyObject_HEAD        /* No semicolon! */
 	PyObject *parent;    /* Config */
@@ -81,7 +170,6 @@ typedef struct {
 	PyObject *values;    /* Sequence */
 	PyObject *children;  /* Sequence */
 } Config;
-
 PyTypeObject ConfigType;
 
 typedef struct {
@@ -93,21 +181,29 @@ typedef struct {
 	char type[DATA_MAX_NAME_LEN];
 	char type_instance[DATA_MAX_NAME_LEN];
 } PluginData;
-
 PyTypeObject PluginDataType;
+#define PluginData_New() PyObject_CallFunctionObjArgs((PyObject *) &PluginDataType, (void *) 0)
 
 typedef struct {
 	PluginData data;
 	PyObject *values;    /* Sequence */
-	int interval;
+	PyObject *meta;      /* dict */
+	double interval;
 } Values;
-
 PyTypeObject ValuesType;
+#define Values_New() PyObject_CallFunctionObjArgs((PyObject *) &ValuesType, (void *) 0)
 
 typedef struct {
 	PluginData data;
 	int severity;
 	char message[NOTIF_MAX_MSG_LEN];
 } Notification;
-
 PyTypeObject NotificationType;
+#define Notification_New() PyObject_CallFunctionObjArgs((PyObject *) &NotificationType, (void *) 0)
+
+typedef PyLongObject Signed;
+PyTypeObject SignedType;
+
+typedef PyLongObject Unsigned;
+PyTypeObject UnsignedType;
+