Add lazy import filter

Author: DinoV

Include/import.h

@@ -88,6 +88,16 @@ PyAPI_FUNC(int) PyImport_AppendInittab(
     PyObject* (*initfunc)(void)
     );
 
+typedef enum  {
+    PyLazyImportsMode_Default,
+    PyLazyImportsMode_ForcedOff,
+    PyLazyImportsMode_ForcedOn,
+} PyImport_LazyImportsMode;
+
+PyAPI_FUNC(int) PyImport_SetLazyImports(PyImport_LazyImportsMode mode, PyObject *filter);
+
+PyAPI_FUNC(PyImport_LazyImportsMode) PyImport_LazyImportsEnabled(void);
+
 #ifndef Py_LIMITED_API
 #  define Py_CPYTHON_IMPORT_H
 #  include "cpython/import.h"

Include/internal/pycore_ceval.h

@@ -298,7 +298,7 @@ PyAPI_FUNC(void) _PyEval_FormatExcUnbound(PyThreadState *tstate, PyCodeObject *c
 PyAPI_FUNC(void) _PyEval_FormatKwargsError(PyThreadState *tstate, PyObject *func, PyObject *kwargs);
 PyAPI_FUNC(PyObject *) _PyEval_ImportFrom(PyThreadState *, PyObject *, PyObject *);
 PyAPI_FUNC(PyObject *) _PyEval_LazyImportName(PyThreadState *tstate, PyObject *builtins, PyObject *globals, 
-            PyObject *locals, PyObject *name, PyObject *fromlist, PyObject *level);
+            PyObject *locals, PyObject *name, PyObject *fromlist, PyObject *level, int lazy);
 PyAPI_FUNC(PyObject *) _PyEval_LazyImportFrom(PyThreadState *tstate, PyObject *v, PyObject *name);
 PyAPI_FUNC(PyObject *) _PyEval_ImportName(PyThreadState *tstate, PyObject *builtins, PyObject *globals, PyObject *locals,
             PyObject *name, PyObject *fromlist, PyObject *level);

Include/internal/pycore_global_objects_fini_generated.h

@@ -574,6 +574,7 @@ struct _Py_global_strings {
         STRUCT_FOR_ID(last_type)
         STRUCT_FOR_ID(last_value)
         STRUCT_FOR_ID(latin1)
+        STRUCT_FOR_ID(lazy)
         STRUCT_FOR_ID(leaf_size)
         STRUCT_FOR_ID(legacy)
         STRUCT_FOR_ID(len)

Include/internal/pycore_global_strings.h

@@ -314,6 +314,8 @@ struct _import_state {
 #endif
     PyObject *import_func;
     PyObject *lazy_import_func;
+    int lazy_imports_mode;
+    PyObject *lazy_imports_filter;
     /* The global import lock. */
     _PyRecursiveMutex lock;
     /* diagnostic info in PyImport_ImportModuleLevelObject() */

Include/internal/pycore_interp_structs.h

@@ -59,7 +59,7 @@
 
 from ._bootstrap import __import__
 
-from _imp import lazy_import
+from _imp import lazy_import, set_lazy_imports
 
 
 def invalidate_caches():

Include/internal/pycore_runtime_init_generated.h

@@ -2548,6 +2548,8 @@ def tearDown(self):
             if key.startswith('test.test_import.data.lazy_imports'):
                 del sys.modules[key]
 
+        importlib.set_lazy_imports(None, None)
+
     def test_basic_unused(self):
         try:
             import test.test_import.data.lazy_imports.basic_unused
@@ -2564,6 +2566,54 @@ def test_basic_used(self):
 
         self.assertTrue("test.test_import.data.lazy_imports.basic2" in sys.modules)
 
+    def test_global_off(self):
+        try:
+            import test.test_import.data.lazy_imports.global_off
+        except ImportError as e:
+            self.fail('lazy import failed')
+
+        self.assertTrue("test.test_import.data.lazy_imports.basic2" in sys.modules)
+
+    def test_global_on(self):
+        try:
+            import test.test_import.data.lazy_imports.global_on
+        except ImportError as e:
+            self.fail('lazy import failed')
+
+        self.assertFalse("test.test_import.data.lazy_imports.basic2" in sys.modules)
+
+    def test_global_filter(self):
+        try:
+            import test.test_import.data.lazy_imports.global_filter
+        except ImportError as e:
+            self.fail('lazy import failed')
+
+        self.assertTrue("test.test_import.data.lazy_imports.basic2" in sys.modules)
+
+    def test_global_filter_true(self):
+        try:
+            import test.test_import.data.lazy_imports.global_filter_true
+        except ImportError as e:
+            self.fail('lazy import failed')
+
+        self.assertFalse("test.test_import.data.lazy_imports.basic2" in sys.modules)
+
+    def test_global_filter_from(self):
+        try:
+            import test.test_import.data.lazy_imports.global_filter
+        except ImportError as e:
+            self.fail('lazy import failed')
+
+        self.assertTrue("test.test_import.data.lazy_imports.basic2" in sys.modules)
+
+    def test_global_filter_from_true(self):
+        try:
+            import test.test_import.data.lazy_imports.global_filter_true
+        except ImportError as e:
+            self.fail('lazy import failed')
+
+        self.assertFalse("test.test_import.data.lazy_imports.basic2" in sys.modules)
+
 
 class TestSinglePhaseSnapshot(ModuleSnapshot):
     """A representation of a single-phase init module for testing.

Include/internal/pycore_unicodeobject_generated.h

@@ -0,0 +1,10 @@
+import importlib
+
+def filter(module_name, imported_name, from_list):
+    assert module_name == __name__
+    assert imported_name == "test.test_import.data.lazy_imports.basic2"
+    return False
+
+importlib.set_lazy_imports(None, filter)
+
+lazy import test.test_import.data.lazy_imports.basic2 as basic2