chore: mid work

Author: dgandhi62

packages/@jsii/python-runtime/src/jsii/__init__.py

@@ -14,6 +14,7 @@
     kernel,
     proxy_for,
 )
+from ._utils import _LazyImport
 from . import python
 
 
@@ -49,6 +50,7 @@
 __all__ = [
     "__version__",
     "__jsii_runtime_version__",
+    "_LazyImport",
     "JSIIAssembly",
     "JSIIMeta",
     "JSIIAbstractClass",

packages/@jsii/python-runtime/src/jsii/_utils.py

@@ -1,4 +1,5 @@
 import functools
+import importlib
 
 from typing import Any, MutableMapping, Type
 
@@ -24,3 +25,28 @@ def wrapped(self):
         return stored[0]
 
     return property(wrapped)
+
+
+class _LazyImport:
+    """Defers ``importlib.import_module()`` until first attribute access.
+
+    This is used by jsii-pacmak generated code to lazily import cross-module
+    dependencies, breaking circular import chains and reducing module load time.
+
+    The imported module is cached after first successful resolution. Failed
+    imports are NOT cached, allowing retry on subsequent access.
+    """
+
+    def __init__(self, module_name: str) -> None:
+        self._module_name = module_name
+        self._module: Any = None
+
+    def __getattr__(self, name: str) -> Any:
+        if self._module is None:
+            self._module = importlib.import_module(self._module_name)
+        return getattr(self._module, name)
+
+    def __repr__(self) -> str:
+        if self._module is not None:
+            return repr(self._module)
+        return f"_LazyImport({self._module_name!r})"

packages/jsii-pacmak/lib/targets/python.ts

@@ -2164,30 +2164,13 @@ class PythonModule implements PythonType {
   }
 
   /**
-   * Emit the `_LazyImport` helper class that defers `importlib.import_module()`
-   * until first attribute access via `__getattr__`.
-   *
-   * The class is emitted once per module that has cross-module imports. It wraps
-   * a module name string and caches the imported module after first access.
-   * Failed imports are NOT cached, allowing retry on subsequent access.
+   * Emit the `_LazyImport` alias that references `jsii._LazyImport` from the
+   * runtime package. This avoids duplicating the class definition in every
+   * generated module.
    */
-  private emitLazyImportClass(code: CodeMaker) {
+  private emitLazyImportAlias(code: CodeMaker) {
     code.line();
-    code.openBlock('class _LazyImport');
-    // __init__
-    code.openBlock('def __init__(self, module_name: str) -> None');
-    code.line('self._module_name = module_name');
-    code.line('self._module: typing.Any = None');
-    code.closeBlock();
-    // __getattr__
-    code.openBlock('def __getattr__(self, name: str) -> typing.Any');
-    code.openBlock('if self._module is None');
-    code.line('import importlib');
-    code.line('self._module = importlib.import_module(self._module_name)');
-    code.closeBlock();
-    code.line('return getattr(self._module, name)');
-    code.closeBlock();
-    code.closeBlock();
+    code.line('_LazyImport = jsii._LazyImport');
   }
 
   /**
@@ -2258,8 +2241,8 @@ class PythonModule implements PythonType {
       return;
     }
 
-    // Emit _LazyImport class definition
-    this.emitLazyImportClass(code);
+    // Emit a local alias to jsii._LazyImport for use in lazy proxy assignments
+    this.emitLazyImportAlias(code);
 
     // Emit TYPE_CHECKING block with original import statements for static type checkers,
     // and lazy proxy assignments in the else branch for runtime.