[fix] Python 3 module init, teardown methods

osandvik · Feb 12, 2016 · bc8ad67 · bc8ad67
1 parent a6c22f0
commit bc8ad67
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/postal/pyexpand.c b/postal/pyexpand.c
@@ -157,7 +157,7 @@ static PyObject *py_expand(PyObject *self, PyObject *args, PyObject *keywords) {
 
                 language = NULL;
 
-                #if IS_PY3K
+                #ifdef IS_PY3K
 
                 if (PyBytes_Check(item)) {
                     language = PyBytes_AsString(item);
@@ -258,7 +258,7 @@ static int expand_traverse(PyObject *m, visitproc visit, void *arg) {
 
 static int expand_clear(PyObject *m) {
     Py_CLEAR(GETSTATE(m)->error);
-    libpostal_language_classifier_teardown();
+    libpostal_teardown_language_classifier();
     libpostal_teardown();
     return 0;
 }
@@ -278,14 +278,15 @@ static struct PyModuleDef module_def = {
 #define INITERROR return NULL
 
 PyObject *
-PyInit_expand(void) {
+PyInit__expand(void) {
 
 #else
 
 #define INITERROR return
 
 void cleanup_libpostal(void) {
     libpostal_teardown();
+    libpostal_teardown_language_classifier();
 }
 
 void
@@ -334,7 +335,7 @@ init_expand(void) {
     Py_AtExit(&cleanup_libpostal);
 #endif
 
-#if IS_PY3K
+#ifdef IS_PY3K
     return module;
 #endif
 }

diff --git a/postal/pyparser.c b/postal/pyparser.c
@@ -251,7 +251,7 @@ static struct PyModuleDef module_def = {
 #define INITERROR return NULL
 
 PyObject *
-PyInit_parser(void) {
+PyInit__parser(void) {
 #else
 
 #define INITERROR return