moose-lib 0.5.7__py3-none-any.whl → 0.5.8__py3-none-any.whl

moose_lib/dmv2/_registry.py

@@ -11,5 +11,7 @@ _tables: Dict[str, Any] = {}
 _streams: Dict[str, Any] = {}
 _ingest_apis: Dict[str, Any] = {}
 _egress_apis: Dict[str, Any] = {}
+# Alias map for O(1) fallback of sole versioned APIs: base name -> handler
+_egress_api_name_aliases: Dict[str, Any] = {}
 _sql_resources: Dict[str, Any] = {}
 _workflows: Dict[str, Any] = {} 

moose_lib/dmv2/consumption.py

@@ -12,7 +12,7 @@ from pydantic import BaseModel
 from pydantic.json_schema import JsonSchemaValue
 
 from .types import BaseTypedResource, T, U
-from ._registry import _egress_apis
+from ._registry import _egress_apis, _egress_api_name_aliases
 
 # Global base URL configuration
 _global_base_url: Optional[str] = None
@@ -116,7 +116,38 @@ class ConsumptionApi(BaseTypedResource, Generic[U]):
             
         self.query_function = query_function
         self.metadata = getattr(self.config, 'metadata', {}) or {}
-        _egress_apis[_generate_api_key(name, self.config.version)] = self
+        key = _generate_api_key(name, self.config.version)
+        _egress_apis[key] = self
+
+        # Maintain alias for base name:
+        # - If explicit unversioned registered, alias -> that
+        # - Else, if exactly one versioned exists, alias -> that
+        base = name
+        if self.config.version is None:
+            _egress_api_name_aliases[base] = self
+            return
+
+        # Versioned registration: only adjust alias if no explicit unversioned exists
+        if base in _egress_apis:
+            # Explicit unversioned present, ensure alias points to it
+            _egress_api_name_aliases[base] = _egress_apis[base]
+            return
+
+        # Determine if there is exactly one versioned API
+        prefix = f"{base}:"
+        # Early exit on 2 matches to avoid O(n) counting
+        match_count = 0
+        sole = None
+        for k in _egress_apis.keys():
+            if k.startswith(prefix):
+                match_count += 1
+                sole = _egress_apis[k]
+                if match_count > 1:
+                    break
+        if match_count == 1 and sole is not None:
+            _egress_api_name_aliases[base] = sole
+        else:
+            _egress_api_name_aliases.pop(base, None)
 
     @classmethod
     def _get_type(cls, keyword_args: dict):

moose_lib/dmv2/registry.py

@@ -11,7 +11,15 @@ from .ingest_api import IngestApi
 from .consumption import ConsumptionApi
 from .sql_resource import SqlResource
 from .workflow import Workflow
-from ._registry import _tables, _streams, _ingest_apis, _egress_apis, _sql_resources, _workflows
+from ._registry import (
+    _tables,
+    _streams,
+    _ingest_apis,
+    _egress_apis,
+    _sql_resources,
+    _workflows,
+    _egress_api_name_aliases,
+)
 
 def get_tables() -> Dict[str, OlapTable]:
     """Get all registered OLAP tables."""
@@ -42,8 +50,11 @@ def get_consumption_apis() -> Dict[str, ConsumptionApi]:
     return _egress_apis
 
 def get_consumption_api(name: str) -> Optional[ConsumptionApi]:
-    """Get a registered consumption API by name."""
-    return _egress_apis.get(name)
+    """Get a registered consumption API by name.
+
+    Supports unversioned lookup by name via alias map when only a single versioned API exists.
+    """
+    return _egress_apis.get(name) or _egress_api_name_aliases.get(name)
 
 def get_sql_resources() -> Dict[str, SqlResource]:
     """Get all registered SQL resources."""

{moose_lib-0.5.7.dist-info → moose_lib-0.5.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moose_lib
-Version: 0.5.7
+Version: 0.5.8
 Home-page: https://www.fiveonefour.com/moose
 Author: Fiveonefour Labs Inc.
 Author-email: support@fiveonefour.com

{moose_lib-0.5.7.dist-info → moose_lib-0.5.8.dist-info}/RECORD

@@ -13,14 +13,14 @@ moose_lib/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 moose_lib/config/config_file.py,sha256=NyjY6YFraBel7vBk18lLkpGaPR1viKMAEv4ZldyfLIA,2585
 moose_lib/config/runtime.py,sha256=gF1wrUCbp0MxRU92k0vyItvWX2x6P1FuBRzM61nJE_U,2776
 moose_lib/dmv2/__init__.py,sha256=N81VWGNWRZJbPjDlsCPVYUcu_VaWw5hhdNkKcbWknVk,2448
-moose_lib/dmv2/_registry.py,sha256=agdZ7xzS99Caou60Q2pEErzEwyNYHqwy6XqV79eEmwg,504
-moose_lib/dmv2/consumption.py,sha256=-MrMEOnpQZI-jivLJ8YKteJTYPjNEvO13JeXkmVgM9E,7613
+moose_lib/dmv2/_registry.py,sha256=eKeMRjO0e1_anl0G3qjGMnRI0o9OJghdQ3uVtlsggm4,625
+moose_lib/dmv2/consumption.py,sha256=V2AQWsg5jVJXqbFuB94V9YaHHUgdOHJxgTHPThykwKQ,8791
 moose_lib/dmv2/ingest_api.py,sha256=Snek9NGwaJl_BuImSWGtQq91m9D3AJ4qBoGiKZ-9yTQ,2323
 moose_lib/dmv2/ingest_pipeline.py,sha256=q7k8B50Dp_jpJA86Rr9dOGl7-EaGF0LR81TuFXlktxc,7024
 moose_lib/dmv2/life_cycle.py,sha256=wl0k6yzwU1MJ_fO_UkN29buoY5G6ChYZvfwigP9fVfM,1254
 moose_lib/dmv2/materialized_view.py,sha256=kcx-sJFTM-cH3Uc1GoldgFGodjoz0AegAQEMmohdS38,3826
 moose_lib/dmv2/olap_table.py,sha256=gAo5wE7W2P5cN3GMw7er0jXyfvXVbbo_SZPvzHoJQx0,31640
-moose_lib/dmv2/registry.py,sha256=AaGS6Xy0vKz-wHLPgRVxfKfSwW5KksMePjZ8N7-2OKU,2054
+moose_lib/dmv2/registry.py,sha256=sB6AlSg7H4gQWUN3qco2HOyN6GKQVvFu6DskedyqN78,2252
 moose_lib/dmv2/sql_resource.py,sha256=kUZoGqxhZMHMthtBZGYJBxTFjXkspXiWLXhJRYXgGUM,1864
 moose_lib/dmv2/stream.py,sha256=jiUWBsjFalLLP63mikOxyHRdieiDAlzf9lXfLye-Wjc,10761
 moose_lib/dmv2/types.py,sha256=5FsB0HLHFkYB-8cjJ0rtRUjqahVA-ToLr2JXT1lFiss,3276
@@ -32,7 +32,7 @@ tests/__init__.py,sha256=0Gh4yzPkkC3TzBGKhenpMIxJcRhyrrCfxLSfpTZnPMQ,53
 tests/conftest.py,sha256=ZVJNbnr4DwbcqkTmePW6U01zAzE6QD0kNAEZjPG1f4s,169
 tests/test_moose.py,sha256=mBsx_OYWmL8ppDzL_7Bd7xR6qf_i3-pCIO3wm2iQNaA,2136
 tests/test_redis_client.py,sha256=d9_MLYsJ4ecVil_jPB2gW3Q5aWnavxmmjZg2uYI3LVo,3256
-moose_lib-0.5.7.dist-info/METADATA,sha256=NXHtcceYfH-85R7j7uEuJ90jb6P9vKoKuZGfqgvJMAo,729
-moose_lib-0.5.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-moose_lib-0.5.7.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
-moose_lib-0.5.7.dist-info/RECORD,,
+moose_lib-0.5.8.dist-info/METADATA,sha256=ISnpoRLS2_fVNUv5VnAEomqYmBAKVXOVGmxMA9dQnok,729
+moose_lib-0.5.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+moose_lib-0.5.8.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
+moose_lib-0.5.8.dist-info/RECORD,,