Add compatiblity mode

Author: DinoV

Include/internal/pycore_global_objects_fini_generated.h

@@ -156,6 +156,7 @@ struct _Py_global_strings {
         STRUCT_FOR_ID(__itruediv__)
         STRUCT_FOR_ID(__ixor__)
         STRUCT_FOR_ID(__lazy_import__)
+        STRUCT_FOR_ID(__lazy_modules__)
         STRUCT_FOR_ID(__le__)
         STRUCT_FOR_ID(__len__)
         STRUCT_FOR_ID(__length_hint__)
@@ -574,7 +575,6 @@ 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

@@ -2614,6 +2614,23 @@ def test_global_filter_from_true(self):
 
         self.assertFalse("test.test_import.data.lazy_imports.basic2" in sys.modules)
 
+    def test_compatibility_mode(self):
+        try:
+            import test.test_import.data.lazy_imports.basic_compatibility_mode
+        except ImportError as e:
+            self.fail('lazy import failed')
+
+        self.assertFalse("test.test_import.data.lazy_imports.basic2" in sys.modules)
+
+    def test_compatibility_mode_relative(self):
+        try:
+            import test.test_import.data.lazy_imports.basic_compatibility_mode_relative
+        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_runtime_init_generated.h

@@ -0,0 +1,2 @@
+__lazy_modules__ = ['test.test_import.data.lazy_imports.basic2']
+import test.test_import.data.lazy_imports.basic2

Include/internal/pycore_unicodeobject_generated.h

@@ -0,0 +1,2 @@
+__lazy_modules__ = ['test.test_import.data.lazy_imports.basic2']
+lazy from .basic2 import f

Lib/test/test_import/__init__.py

@@ -3862,6 +3862,7 @@ PyImport_ImportModuleLevelObject(PyObject *name, PyObject *globals,
     PyObject *final_mod = NULL;
     PyObject *mod = NULL;
     PyObject *package = NULL;
+    PyObject *lazy_modules = NULL;
     PyInterpreterState *interp = tstate->interp;
     int has_from;
 
@@ -3870,6 +3871,11 @@ PyImport_ImportModuleLevelObject(PyObject *name, PyObject *globals,
         goto error;
     }
 
+    if (globals != NULL &&
+        PyMapping_GetOptionalItem(globals, &_Py_ID(__lazy_modules__), &lazy_modules) < 0) {
+        goto error;
+    }
+
     /* The below code is importlib.__import__() & _gcd_import(), ported to C
        for added performance. */
 
@@ -3888,6 +3894,24 @@ PyImport_ImportModuleLevelObject(PyObject *name, PyObject *globals,
         goto error;
     }
 
+    if (lazy_modules != NULL) {
+        // Check and see if the module is opting in w/o syntax for backwards compatibility
+        // with older Python versions.
+        int contains = PySequence_Contains(lazy_modules, name);
+        if (contains < 0) {
+            goto error;
+        } else if (contains == 1) {
+            _PyInterpreterFrame *frame = _PyEval_GetFrame();
+            if (frame == NULL) {
+                PyErr_SetString(PyExc_RuntimeError, "no current frame");
+                goto error;
+            }
+            final_mod = _PyImport_LazyImportModuleLevelObject(tstate, name, frame->f_builtins, globals,
+                                                              locals, fromlist, level);
+            goto error;
+        }
+    }
+
     mod = import_get_module(tstate, abs_name);
     if (mod == NULL && _PyErr_Occurred(tstate)) {
         goto error;
@@ -3980,6 +4004,7 @@ PyImport_ImportModuleLevelObject(PyObject *name, PyObject *globals,
     }
 
   error:
+    Py_XDECREF(lazy_modules);
     Py_XDECREF(abs_name);
     Py_XDECREF(mod);
     Py_XDECREF(package);