lazy submodule imports

optype/__init__.py

@@ -1,6 +1,6 @@
-from importlib import metadata as _metadata
+import importlib as _importlib
+import importlib.metadata as _metadata
 
-from . import copy, dataclasses, inspect, json, pickle, string, types, typing
 from ._core._can import (
     CanAEnter,
     CanAEnterSelf,
@@ -315,6 +315,13 @@
 )
 
 
+__version__: str = _metadata.version(__package__ or __file__.split("/")[-1])
+
+
+def __c(s: str, k: str, /) -> str:
+    return "".join(chr(ord(c) ^ ord(k)) for c in s)
+
+
 __all__ = [
     "CanAEnter",
     "CanAEnterSelf",
@@ -539,8 +546,6 @@
     "HasTypeParams",
     "HasWrapped",
     "__version__",
-    "copy",
-    "dataclasses",
     "do_abs",
     "do_add",
     "do_aiter",
@@ -623,20 +628,37 @@
     "do_truediv",
     "do_trunc",
     "do_xor",
-    "inspect",
-    "json",
-    "pickle",
-    "rick",
-    "string",
-    "types",
-    "typing",
 ]
 
-__version__: str = _metadata.version(__package__ or __file__.split("/")[-1])
+
+_submodules = {
+    "copy": "copy",
+    "dataclasses": "dataclasses",
+    "dlpack": "dlpack",
+    "inspect": "inspect",
+    "json": "json",
+    "numpy": "numpy",
+    "pickle": "pickle",
+    "string": "string",
+    "types": "types",
+    "typing": "typing",
+}
+__all__ += list(_submodules)  # pyright: ignore[reportUnsupportedDunderAll]
 
 # stop digging for hidden layers and be impressed
-rick = pickle
+_submodules[__c("🦞🦅🦏🦇", "🧬")] = __c("🤢🤻🤱🤹🤾🤷", "🥒")
 
 
 def __dir__() -> list[str]:
     return __all__
+
+
+def __getattr__(name: str) -> object:
+    if submodule := _submodules.get(name):
+        return _importlib.import_module(f"{__name__}.{submodule}")
+    try:
+        return globals()[name]  # pyright: ignore[reportAny]
+    except KeyError:
+        msg = f"module '{__name__}' has no attribute '{name}'"
+        module = _importlib.import_module(__name__)
+        raise AttributeError(msg, name=name, obj=module) from None