fix: allow module to be imported twice in the same process

nedbat · nedbat · commit 2d39c8cb6d00 · 2025-05-31T14:43:22.000-04:00
diff --git a/coverage/ctracer/module.c b/coverage/ctracer/module.c
@@ -9,18 +9,17 @@
 
 #define MODULE_DOC PyDoc_STR("Fast coverage tracer.")
 
-static int module_loaded = 0;
+static BOOL module_inited = FALSE;
+static PyMutex modinit_mutex = {0};
 
 static int
 tracer_exec(PyObject *mod)
 {
-    // https://docs.python.org/3/howto/isolating-extensions.html#opt-out-limiting-to-one-module-object-per-process
-    if (module_loaded) {
-        PyErr_SetString(PyExc_ImportError,
-                        "cannot load module more than once per process");
-        return -1;
+    PyMutex_Lock(&modinit_mutex);
+    if (module_inited) {
+        PyMutex_Unlock(&modinit_mutex);
+        return 0;
     }
-    module_loaded = 1;
 
     if (CTracer_intern_strings() < 0) {
         return -1;
@@ -52,6 +51,9 @@ tracer_exec(PyObject *mod)
         return -1;
     }
 
+    module_inited = TRUE;
+    PyMutex_Unlock(&modinit_mutex);
+
     return 0;
 }
 
