moose-lib 0.6.10__tar.gz → 0.6.12__tar.gz

{moose_lib-0.6.10 → moose_lib-0.6.12}/PKG-INFO

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

{moose_lib-0.6.10 → moose_lib-0.6.12}/moose_lib/data_models.py

@@ -26,6 +26,15 @@ class ClickhouseSize:
     size: int
 
 
+@dataclasses.dataclass
+class ClickhouseDefault:
+    expression: str
+
+
+def clickhouse_default(expression: str) -> ClickhouseDefault:
+    return ClickhouseDefault(expression=expression)
+
+
 def clickhouse_decimal(precision: int, scale: int) -> Type[Decimal]:
     return Annotated[Decimal, Field(max_digits=precision, decimal_places=scale)]
 
@@ -135,6 +144,7 @@ class Column(BaseModel):
     required: bool
     unique: Literal[False]
     primary_key: bool
+    default: str | None = None
     annotations: list[Tuple[str, Any]] = []
 
 
@@ -255,6 +265,12 @@ def _to_columns(model: type[BaseModel]) -> list[Column]:
 
         column_name = field_name if field_info.alias is None else field_info.alias
 
+        # Extract default expression from metadata, if provided
+        default_expr = next(
+            (md.expression for md in mds if isinstance(md, ClickhouseDefault)),
+            None,
+        )
+
         columns.append(
             Column(
                 name=column_name,
@@ -262,6 +278,7 @@ def _to_columns(model: type[BaseModel]) -> list[Column]:
                 required=not optional,
                 unique=False,
                 primary_key=primary_key,
+                default=default_expr,
                 annotations=annotations,
             )
         )

{moose_lib-0.6.10 → moose_lib-0.6.12}/moose_lib/dmv2/__init__.py

@@ -46,6 +46,9 @@ from .consumption import (
     Api,
     get_moose_base_url,
     set_moose_base_url,
+    # Backward compatibility aliases
+    ConsumptionApi,
+    EgressConfig,
 )
 
 from .sql_resource import (
@@ -85,6 +88,9 @@ from .registry import (
     get_sql_resource,
     get_workflows,
     get_workflow,
+    # Backward compatibility aliases
+    get_consumption_apis,
+    get_consumption_api,
 )
 
 __all__ = [
@@ -123,6 +129,9 @@ __all__ = [
     'Api',
     'get_moose_base_url',
     'set_moose_base_url',
+    # Backward compatibility aliases (deprecated)
+    'ConsumptionApi',
+    'EgressConfig',
 
     # SQL
     'SqlResource',
@@ -152,4 +161,7 @@ __all__ = [
     'get_sql_resource',
     'get_workflows',
     'get_workflow',
+    # Backward compatibility aliases (deprecated)
+    'get_consumption_apis',
+    'get_consumption_api',
 ]

{moose_lib-0.6.10 → moose_lib-0.6.12}/moose_lib/dmv2/consumption.py

@@ -228,3 +228,11 @@ class Api(BaseTypedResource, Generic[U]):
         # Parse JSON response and return as the expected type
         response_data = response.json()
         return self._u.model_validate(response_data)
+
+
+# Backward compatibility aliases (deprecated)
+ConsumptionApi = Api
+"""@deprecated: Use Api instead of ConsumptionApi"""
+
+EgressConfig = ApiConfig
+"""@deprecated: Use ApiConfig instead of EgressConfig"""

{moose_lib-0.6.10 → moose_lib-0.6.12}/moose_lib/dmv2/registry.py

@@ -70,4 +70,12 @@ def get_workflows() -> Dict[str, Workflow]:
 
 def get_workflow(name: str) -> Optional[Workflow]:
     """Get a registered workflow by name."""
-    return _workflows.get(name)
+    return _workflows.get(name)
+
+
+# Backward compatibility aliases (deprecated)
+get_consumption_apis = get_apis
+"""@deprecated: Use get_apis instead of get_consumption_apis"""
+
+get_consumption_api = get_api
+"""@deprecated: Use get_api instead of get_consumption_api"""

{moose_lib-0.6.10 → moose_lib-0.6.12}/moose_lib/main.py

@@ -147,6 +147,11 @@ class ApiResult:
     body: Any
 
 
+# Backward compatibility alias (deprecated)
+ConsumptionApiResult = ApiResult
+"""@deprecated: Use ApiResult instead of ConsumptionApiResult"""
+
+
 class QueryClient:
     """Client for executing queries, typically against ClickHouse.
 

{moose_lib-0.6.10 → moose_lib-0.6.12}/moose_lib.egg-info/PKG-INFO

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