cos-comparison 0.1.0__py3-none-any.whl

cos_comparison/__init__.py

@@ -0,0 +1,146 @@
+# cos_comparison package initializer
+#
+# Dynamic fallback through configured backends.  Any attribute that is not
+# already defined in this module is lazily resolved at call time by
+# searching the backends listed in config.json in order.  Once a backend
+# returns a result (or raises an unexpected error), the process stops.
+
+import importlib
+import json
+import os
+from typing import Any
+
+# ----------------------------------------------------------------------
+#  File context manager (kept from original codebase)
+# ----------------------------------------------------------------------
+class file_context:
+    __slots__ = ("file",)
+    def __init__(self, *arg, **kwarg):
+        self.file = open(*arg, **kwarg)
+    def __enter__(self):
+        self.file = self.file.__enter__()
+        return self
+    def __exit__(self, *arg):
+        self.file.__exit__(*arg)
+    def __del__(self):
+        self.close()
+    def __getattr__(self, name):
+        return getattr(self.file, name)
+
+
+# ----------------------------------------------------------------------
+#  Backend discovery from config.json
+# ----------------------------------------------------------------------
+def _load_backend_names() -> list:
+    """Return the ordered list of backend module names."""
+    conf_path = os.path.join(os.path.dirname(__file__), "config.json")
+    if not os.path.exists(conf_path):
+        return ["backend_python"]                     # sensible default
+
+    with file_context(conf_path, "r", encoding="utf-8") as f:
+        try:
+            meta, _ = json.load(f)                    # [backends, plugin_dir]
+        except Exception:
+            return ["backend_python"]
+
+    if isinstance(meta, list) and meta:
+        return meta
+    return ["backend_python"]
+
+import os
+
+def _get_backend_order():
+    try:
+        return  _load_backend_names()
+    except:
+        pass    
+    env_order = os.environ.get("COS_BACKEND", "").strip()
+    if env_order:
+        return [name.strip() for name in env_order.split(",")]
+
+    return [
+        "cos_comparison_c",
+        "cos_comparison_numpy",
+        "cos_comparison",
+    ]
+
+_BACKEND_NAMES = _get_backend_order()
+
+
+
+# ----------------------------------------------------------------------
+#  Module cache (avoid repeated imports)
+# ----------------------------------------------------------------------
+_module_cache: dict = {}          # name -> module | None
+
+def _import_backend(name: str) -> Any:
+    """Import a backend module and cache it.  Returns module or None."""
+    if name in _module_cache:
+        return _module_cache[name]
+    try:
+        mod = importlib.import_module(f".{name}", package="cos_comparison")
+        _module_cache[name] = mod
+        return mod
+    except ImportError:
+        _module_cache[name] = None
+        return None
+
+
+# ----------------------------------------------------------------------
+#  Callable object that searches backends on every invocation
+# ----------------------------------------------------------------------
+class _FallbackWrapper:
+    __slots__ = ("func_name",)
+
+    def __init__(self, func_name: str):
+        self.func_name = func_name
+
+    def __call__(self, *args, **kwargs):
+        last_exception = None
+        for bname in _BACKENDS:
+            mod = _import_backend(bname)
+            if mod is None:
+                continue                          # module not available
+            func = getattr(mod, self.func_name, None)
+            if func is None:
+                continue                          # function not provided
+
+            try:
+                return func(*args, **kwargs)
+            except NotImplementedError:
+                # Backend explicitly refuses this call -- normal fallback
+                continue
+            except Exception as exc:
+                # Unexpected error – remember it but try next backend
+                last_exception = exc
+                continue
+
+        if last_exception is not None:
+            raise last_exception
+        raise RuntimeError(
+            f"No backend among {_BACKENDS} provides a working "
+            f"implementation for '{self.func_name}'."
+        )
+
+
+# ----------------------------------------------------------------------
+#  Intercept missing attributes (Python 3.7+)
+# ----------------------------------------------------------------------
+def __getattr__(name: str):
+    """Called only when `name` is not found in the module globals.
+
+    Returns a callable wrapper that searches backends on each call.
+    Once created, the wrapper is cached in the module's dictionary so
+    that subsequent lookups are fast.
+    """
+    # No exclusion of private names – internal functions may be needed.
+    wrapper = _FallbackWrapper(name)
+    globals()[name] = wrapper
+    return wrapper
+
+
+# ----------------------------------------------------------------------
+#  Provide a useful __dir__ for introspection
+# ----------------------------------------------------------------------
+def __dir__() -> list:
+    return sorted(set(globals().keys()))