diff --git a/python/TODO.md b/python/TODO.md index d7aff07..621e79f 100644 --- 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 index 6c794fa..aea5804 100644 --- 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 @@ -182,10 +182,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 + }