lazy loading of modules and exposed utility functions

Author: jlaehne

lumispy/__init__.py

@@ -17,18 +17,13 @@
 # along with LumiSpy. If not, see <https://www.gnu.org/licenses/#GPL>.
 
 
-from importlib.metadata import version
+import importlib
 from pathlib import Path
+from typing import Any
 
-from lumispy.utils.axes import nm2eV, eV2nm, nm2invcm, invcm2nm, join_spectra
-from lumispy.utils.io import to_array, savetxt
 
-from lumispy import signals, components, utils
-
-from lumispy.utils import crop_edges
-
-
-__version__ = version("lumispy")
+# rresolve version
+__version__ = importlib.metadata.version("lumispy")
 
 # For development version, `setuptools_scm` will be used at build time
 # to get the dev version, in case of missing vcs information (git archive,
@@ -61,5 +56,44 @@
 ]
 
 
+# Map exported names that will be resolved lazily
+_lazy_modules = {
+    "signals": "lumispy.signals",
+    "components": "lumispy.components",
+    "utils": "lumispy.utils",
+}
+
+# Map top-level utility names to their submodule and attribute name
+_lazy_attributes = {
+    # name: (module, attribute)
+    "nm2eV": ("lumispy.utils.axes", "nm2eV"),
+    "eV2nm": ("lumispy.utils.axes", "eV2nm"),
+    "nm2invcm": ("lumispy.utils.axes", "nm2invcm"),
+    "invcm2nm": ("lumispy.utils.axes", "invcm2nm"),
+    "join_spectra": ("lumispy.utils.axes", "join_spectra"),
+    "to_array": ("lumispy.utils.io", "to_array"),
+    "savetxt": ("lumispy.utils.io", "savetxt"),
+}
+
+
+def __getattr__(name: str) -> Any:
+    """Lazy-load subpackages and selected attributes on demand."""
+    # Lazy subpackages
+    if name in _lazy_modules:
+        mod = importlib.import_module(_lazy_modules[name])
+        globals()[name] = mod
+        return mod
+
+    # Lazy attributes (forward from submodules)
+    if name in _lazy_attributes:
+        mod_name, attr = _lazy_attributes[name]
+        mod = importlib.import_module(mod_name)
+        val = getattr(mod, attr)
+        globals()[name] = val
+        return val
+
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
 def __dir__():
-    return sorted(__all__)
+    return sorted(list(__all__) + list(_lazy_attributes.keys()))