Commits

Brian Curtin  committed 475de43

Sprinkle in some ifdefs

  • Participants
  • Parent commits 9bb21a4

Comments (0)

Files changed (1)

File src/_watcher.c

 #include "structmember.h"
 
 #include <windows.h>
-#include "stdio.h"
 
 /* ignore "conditional expression is constant" */
 #pragma warning(disable: 4100)
 
 #define MAX_BUFFER 4096
 
+#if PY_MAJOR_VERSION == 3
+#define PYTHON3
+#endif
+
 
 typedef struct {
     PyObject_HEAD
         Py_DECREF(val);
 }
 
+#ifdef PYTHON3
 static struct PyModuleDef _watchermodule = {
     PyModuleDef_HEAD_INIT,
     "_watcher",
     NULL,
     NULL
 };
+#endif
 
+#ifdef PYTHON3
 PyMODINIT_FUNC
 PyInit__watcher(void)
+#else
+PyMODINIT_FUNC
+init_watcher()
+#endif
 {
     PyObject *module, *dict;
 
     /* Initialize and acquire the GIL before we do anything else. */
     PyEval_InitThreads();
 
+#ifdef PYTHON3
     module = PyModule_Create(&_watchermodule);
+#else
+    module = Py_InitModule3("_watcher", NULL, module_doc);
+#endif
 
     if (PyType_Ready(&watcher_object_type) < 0)
         goto fail;
     setint(dict, "FILE_ACTION_RENAMED_NEW_NAME",
                  FILE_ACTION_RENAMED_NEW_NAME);
 
+#ifdef PYTHON3
     return module;
+#endif
 
 fail:
+#ifdef PYTHON3
     return NULL;
+#else
+    return;
+#endif
 }