Rewrite _coconut_lazy_module as function creating per-module class+metaclass

Instead of a class with complex __getattribute__/__setattr__ overrides,
_coconut_lazy_module is now a function that creates a unique class per
lazy module. The class IS the lazy module proxy, and its metaclass handles
attribute access and dunder forwarding via closures over the load state.

This eliminates the need for _coconut_base_getattr/_coconut_base_setattr
hacks and __slots__ with prefixed names. Also simplifies trollius handling
by removing the _coconut_trollius_module subclass.

https://claude.ai/code/session_01KBLQLYTdfjLPaR2UJhgnkq

Author: claude

coconut/compiler/header.py

@@ -322,13 +322,13 @@ def process_header_args(which, use_hash, target, no_tco, strict, no_wrap):
             (2, 7),
             if_lt='''
 import imp
-_coconut_lazy_module._coconut_base_setattr(self, "_coconut_module", imp.load_module(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_name"), *imp.find_module(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_name"))))
+return imp.load_module(name, *imp.find_module(name))
             ''',
             if_ge='''
 import importlib
-_coconut_lazy_module._coconut_base_setattr(self, "_coconut_module", importlib.import_module(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_name")))
+return importlib.import_module(name)
             ''',
-            indent=4,
+            indent=1,
         ),
         import_OrderedDict=prepare(
             r'''
@@ -761,21 +761,9 @@ class you_need_to_install_typing_extensions{object}:
         import_asyncio=pycondition(
             (3, 4),
             if_lt='''
-class _coconut_trollius_module(_coconut_lazy_module):
-    __slots__ = ()
-    def coroutine(self, func):
-        try:
-            mod = self._coconut_load()
-        except ImportError:
-            def raise_import_error(*args, **kwargs):
-                raise _coconut_lazy_module._coconut_base_getattr(self, "_coconut_import_err")
-            return raise_import_error
-        return mod.coroutine(func)
-    def Return(self, obj):
-        return self._coconut_load().Return(obj)
-asyncio = _coconut_trollius_module("trollius")
-asyncio_Return = asyncio.Return
-            '''.format(**format_dict),
+asyncio = _coconut_lazy_module("trollius")
+asyncio_Return = _coconut_lazy_module("trollius", attr="Return")
+            ''',
             if_ge='''
 import asyncio
 asyncio_Return = StopIteration

coconut/compiler/templates/header.py_template

@@ -1,57 +1,45 @@
-class _coconut_lazy_module{object}:
-    __slots__ = ("_coconut_name", "_coconut_module", "_coconut_on_import", "_coconut_import_err", "_coconut_attr")
-    _coconut_base_getattr = object.__getattribute__
-    _coconut_base_setattr = object.__setattr__
-    def __init__(self, name, on_import=None, attr=None):
-        _coconut_lazy_module._coconut_base_setattr(self, "_coconut_name", name)
-        _coconut_lazy_module._coconut_base_setattr(self, "_coconut_module", None)
-        _coconut_lazy_module._coconut_base_setattr(self, "_coconut_on_import", on_import)
-        _coconut_lazy_module._coconut_base_setattr(self, "_coconut_import_err", None)
-        _coconut_lazy_module._coconut_base_setattr(self, "_coconut_attr", attr)
-    def _coconut_load(self):
-        if _coconut_lazy_module._coconut_base_getattr(self, "_coconut_module") is None and _coconut_lazy_module._coconut_base_getattr(self, "_coconut_import_err") is None:
-            try:
+def _coconut_import_module(name):
 {lazy_module_import}
+_coconut_lazy_dunders = ("__call__", "__iter__", "__next__", "__len__", "__contains__", "__getitem__", "__setitem__", "__delitem__", "__reversed__", "__missing__", "__repr__", "__str__", "__format__", "__bytes__", "__hash__", "__int__", "__float__", "__complex__", "__index__", "__round__", "__ceil__", "__floor__", "__trunc__", "__neg__", "__pos__", "__abs__", "__invert__", "__add__", "__radd__", "__iadd__", "__sub__", "__rsub__", "__isub__", "__mul__", "__rmul__", "__imul__", "__truediv__", "__rtruediv__", "__itruediv__", "__floordiv__", "__rfloordiv__", "__ifloordiv__", "__mod__", "__rmod__", "__imod__", "__pow__", "__rpow__", "__ipow__", "__matmul__", "__rmatmul__", "__imatmul__", "__lshift__", "__rlshift__", "__ilshift__", "__rshift__", "__rrshift__", "__irshift__", "__and__", "__rand__", "__iand__", "__or__", "__ror__", "__ior__", "__xor__", "__rxor__", "__ixor__", "__lt__", "__le__", "__gt__", "__ge__", "__eq__", "__ne__", "__enter__", "__exit__", "__aiter__", "__anext__", "__aenter__", "__aexit__", "__await__", "__divmod__", "__rdivmod__", "__nonzero__", "__length_hint__")
+def _coconut_lazy_module(name, on_import=None, attr=None):
+    _coconut_lazy_state = [None, None]
+    def _coconut_lazy_load():
+        if _coconut_lazy_state[0] is None and _coconut_lazy_state[1] is None:
+            try:
+                _coconut_lazy_state[0] = _coconut_import_module(name)
             except _coconut.ImportError as err:
-                _coconut_lazy_module._coconut_base_setattr(self, "_coconut_import_err", err)
+                _coconut_lazy_state[1] = err
             else:
-                on_import = _coconut_lazy_module._coconut_base_getattr(self, "_coconut_on_import")
                 if on_import is not None:
-                    on_import(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_module"))
-        _coconut_import_err = _coconut_lazy_module._coconut_base_getattr(self, "_coconut_import_err")
-        if _coconut_import_err is not None:
-            raise _coconut_import_err
-        _coconut_module = _coconut_lazy_module._coconut_base_getattr(self, "_coconut_module")
-        _coconut_attr = _coconut_lazy_module._coconut_base_getattr(self, "_coconut_attr")
-        if _coconut_attr is not None:
-            return _coconut.getattr(_coconut_module, _coconut_attr)
-        return _coconut_module
-    def __getattribute__(self, name):
-        if name in _coconut_lazy_module.__slots__ or name == "_coconut_load":
-            return _coconut_lazy_module._coconut_base_getattr(self, name)
-        try:
-            return _coconut_lazy_module._coconut_base_getattr(self, name)
-        except _coconut.AttributeError:
-            pass
-        return _coconut.getattr(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_load")(), name)
-    def __setattr__(self, name, value):
-        if name in _coconut_lazy_module.__slots__:
-            _coconut_lazy_module._coconut_base_setattr(self, name, value)
-        else:
-            _coconut.setattr(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_load")(), name, value)
-    def __dir__(self):
-        try:
-            return _coconut.dir(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_load")())
-        except _coconut.ImportError:
-            return _coconut_lazy_module.__slots__
-    def __bool__(self):
-        return True
-def _coconut_make_lazy_dunder(_coconut_dunder_name):
-    def _coconut_lazy_dunder(self, *args, **kwargs):
-        return _coconut.getattr(_coconut_lazy_module._coconut_base_getattr(self, "_coconut_load")(), _coconut_dunder_name)(*args, **kwargs)
-    return _coconut_lazy_dunder
-for _coconut_dunder in ("__call__", "__iter__", "__next__", "__len__", "__contains__", "__getitem__", "__setitem__", "__delitem__", "__reversed__", "__missing__", "__repr__", "__str__", "__format__", "__bytes__", "__hash__", "__int__", "__float__", "__complex__", "__index__", "__round__", "__ceil__", "__floor__", "__trunc__", "__neg__", "__pos__", "__abs__", "__invert__", "__add__", "__radd__", "__iadd__", "__sub__", "__rsub__", "__isub__", "__mul__", "__rmul__", "__imul__", "__truediv__", "__rtruediv__", "__itruediv__", "__floordiv__", "__rfloordiv__", "__ifloordiv__", "__mod__", "__rmod__", "__imod__", "__pow__", "__rpow__", "__ipow__", "__matmul__", "__rmatmul__", "__imatmul__", "__lshift__", "__rlshift__", "__ilshift__", "__rshift__", "__rrshift__", "__irshift__", "__and__", "__rand__", "__iand__", "__or__", "__ror__", "__ior__", "__xor__", "__rxor__", "__ixor__", "__lt__", "__le__", "__gt__", "__ge__", "__eq__", "__ne__", "__enter__", "__exit__", "__aiter__", "__anext__", "__aenter__", "__aexit__", "__await__", "__divmod__", "__rdivmod__", "__nonzero__", "__length_hint__"):
-    setattr(_coconut_lazy_module, _coconut_dunder, _coconut_make_lazy_dunder(_coconut_dunder))
+                    on_import(_coconut_lazy_state[0])
+        if _coconut_lazy_state[1] is not None:
+            raise _coconut_lazy_state[1]
+        if attr is not None:
+            return _coconut.getattr(_coconut_lazy_state[0], attr)
+        return _coconut_lazy_state[0]
+    class _coconut_lazy_meta(type):
+        def __getattr__(cls, name):
+            return _coconut.getattr(_coconut_lazy_load(), name)
+        def __setattr__(cls, name, value):
+            if name.startswith("__") and name.endswith("__"):
+                type.__setattr__(cls, name, value)
+            else:
+                _coconut.setattr(_coconut_lazy_load(), name, value)
+        def __dir__(cls):
+            try:
+                return _coconut.dir(_coconut_lazy_load())
+            except _coconut.ImportError:
+                return []
+        def __bool__(cls):
+            return True
+        __nonzero__ = __bool__
+    def _coconut_make_lazy_dunder(dname):
+        def _coconut_lazy_dunder(cls, *args, **kwargs):
+            return _coconut.getattr(_coconut_lazy_load(), dname)(*args, **kwargs)
+        return _coconut_lazy_dunder
+    for _coconut_dunder in _coconut_lazy_dunders:
+        setattr(_coconut_lazy_meta, _coconut_dunder, _coconut_make_lazy_dunder(_coconut_dunder))
+    return _coconut_lazy_meta("_coconut_lazy_module", (), {{}})
 @_coconut_wraps(_coconut_py_super)
 def _coconut_super(type=None, object_or_type=None):
     if type is None: