Add _PyImport_CheckSubinterpIncompatibleExtensionAllowed().

ericsnowcurrently · ericsnowcurrently · commit ed505a67ad54 · 2023-01-11T10:35:16.000-07:00
diff --git a/Include/internal/pycore_import.h b/Include/internal/pycore_import.h
@@ -52,6 +52,9 @@ PyAPI_DATA(const struct _frozen *) _PyImport_FrozenStdlib;
 PyAPI_DATA(const struct _frozen *) _PyImport_FrozenTest;
 extern const struct _module_alias * _PyImport_FrozenAliases;
 
+PyAPI_FUNC(int) _PyImport_CheckSubinterpIncompatibleExtensionAllowed(
+    const char *name);
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/Python/import.c b/Python/import.c
@@ -2395,6 +2395,21 @@ exec_builtin_or_dynamic(PyObject *mod) {
     return PyModule_ExecDef(mod, def);
 }
 
+int
+_PyImport_CheckSubinterpIncompatibleExtensionAllowed(const char *name)
+{
+    PyInterpreterState *interp = _PyInterpreterState_Get();
+    if (_PyInterpreterState_HasFeature(
+                interp, Py_RTFLAGS_MULTI_INTERP_EXTENSIONS)) {
+        assert(!_Py_IsMainInterpreter(interp));
+        PyErr_Format(PyExc_ImportError,
+                     "module %s does not support loading in subinterpreters",
+                     name);
+        return -1;
+    }
+    return 0;
+}
+
 #ifdef HAVE_DYNAMIC_LOADING
 
 /*[clinic input]

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,9 @@ PyAPI_DATA(const struct _frozen *) _PyImport_FrozenStdlib;`
`52`	`52`	`PyAPI_DATA(const struct _frozen *) _PyImport_FrozenTest;`
`53`	`53`	`extern const struct _module_alias * _PyImport_FrozenAliases;`
`54`	`54`
	`55`	`+PyAPI_FUNC(int) _PyImport_CheckSubinterpIncompatibleExtensionAllowed(`
	`56`	`+ const char *name);`
	`57`	`+`
`55`	`58`	`#ifdef __cplusplus`
`56`	`59`	`}`
`57`	`60`	`#endif`