Python: Fix imports of datetime.h and call PyDateTime_IMPORT during module init.

diff --git a/src/python/createrepo_cmodule.c b/src/python/createrepo_cmodule.c
index 9489b11..a392784 100644 (file)
--- a/src/python/createrepo_cmodule.c
+++ b/src/python/createrepo_cmodule.c
@@ -18,6 +18,7 @@
  */
 
 #include <Python.h>
+#include <datetime.h> // from python
 
 #include "src/createrepo_c.h"
 
@@ -192,6 +193,10 @@ init_createrepo_c(void)
     Py_AtExit(cr_xml_dump_cleanup);
     Py_AtExit(cr_package_parser_cleanup);
 
+    /* Python macro to use datetime objects */
+
+    PyDateTime_IMPORT;
+
     /* Module constants */
 
     /* Version */

diff --git a/src/python/updaterecord-py.c b/src/python/updaterecord-py.c
index 2f344e0..8ed4129 100644 (file)
--- a/src/python/updaterecord-py.c
+++ b/src/python/updaterecord-py.c
@@ -18,7 +18,7 @@
  */
 
 #include <Python.h>
-#include <python2.7/datetime.h>
+#include <datetime.h> // from python
 #include <assert.h>
 #include <stddef.h>
 #include <time.h>
@@ -79,9 +79,6 @@ check_UpdateRecordStatus(const _UpdateRecordObject *self)
 static PyObject *
 updaterecord_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
 {
-    /* Python macro to use datetime objects */
-    PyDateTime_IMPORT;
-
     CR_UNUSED(args);
     CR_UNUSED(kwds);
     _UpdateRecordObject *self = (_UpdateRecordObject *)type->tp_alloc(type, 0);