Clean up technique for building simpleext with different name

adang1345 · Jul 31, 2023 · 7a57e24 · 7a57e24
1 parent 4d3f523
commit 7a57e24
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 58 deletions.
diff --git a/tests/simpleext/build/setup.py b/tests/simpleext/build/setup.py
@@ -26,9 +26,10 @@
     python_requires='==3.10.*',
     zip_safe=False,
     ext_modules=[Extension(
-        f'simpleext{n}', [f'simpleext{n}.c'],
+        f'simpleext{n}', [f'simpleext.c'],
         include_dirs=['simpledll'],
         libraries=['simpledll'],
-        library_dirs=library_dirs
+        library_dirs=library_dirs,
+        define_macros=[('SIMPLEEXT_INIT', f'PyInit_simpleext{n}'), ('SIMPLEEXT_MODNAME', f'"simpleext{n}"')],
     )]
 )
diff --git a/tests/simpleext/build/simpleext.c b/tests/simpleext/build/simpleext.c
@@ -2,6 +2,13 @@
 #include <Python.h>
 #include "simpledll.h"
 
+#ifndef SIMPLEEXT_INIT
+#define SIMPLEEXT_INIT PyInit_simpleext
+#endif
+#ifndef SIMPLEEXT_MODNAME
+#define SIMPLEEXT_MODNAME "simpleext"
+#endif
+
 static PyObject *simpleext_helloworld(PyObject *self, PyObject *args)
 {
    helloworld();
@@ -15,13 +22,13 @@ static PyMethodDef SimpleExtMethods[] = {
 
 static struct PyModuleDef simpleextmodule = {
     PyModuleDef_HEAD_INIT,
-    "simpleext",
+    SIMPLEEXT_MODNAME,
     NULL,
     -1,
     SimpleExtMethods
 };
 
-PyMODINIT_FUNC PyInit_simpleext(void)
+PyMODINIT_FUNC SIMPLEEXT_INIT(void)
 {
     return PyModule_Create(&simpleextmodule);
 }
diff --git a/tests/simpleext/build/simpleext0.c b/tests/simpleext/build/simpleext0.c
diff --git a/tests/simpleext/build/simpleext1.c b/tests/simpleext/build/simpleext1.c