Merge pull request tinyobjloader#111 from longjon/python2

syoyo · web-flow · commit e0b39341fced · 2016-12-06T14:37:05.000+09:00
Support Python 2.7
diff --git a/python/TODO.md b/python/TODO.md
@@ -1,3 +1,2 @@
 * PBR material
 * Define index_t struct
-* Python 2.7 binding
diff --git a/python/main.cpp b/python/main.cpp
@@ -1,4 +1,4 @@
-// python3 module for tinyobjloader
+// python2/3 module for tinyobjloader
 //
 // usage:
 // import tinyobjloader as tol
@@ -183,10 +183,21 @@ static PyMethodDef mMethods[] = {
 
 };
 
+#if PY_MAJOR_VERSION >= 3
+
 static struct PyModuleDef moduledef = {PyModuleDef_HEAD_INIT, "tinyobjloader",
                                        NULL, -1, mMethods};
 
 PyMODINIT_FUNC PyInit_tinyobjloader(void) {
   return PyModule_Create(&moduledef);
 }
+
+#else
+
+PyMODINIT_FUNC inittinyobjloader(void) {
+  Py_InitModule3("tinyobjloader", mMethods, NULL);
+}
+
+#endif  // PY_MAJOR_VERSION >= 3
+
 }
