moose-lib 0.6.25__py3-none-any.whl → 0.6.26__py3-none-any.whl

moose_lib/dmv2/ingest_pipeline.py

@@ -89,7 +89,7 @@ class IngestPipeline(TypedMooseResource, Generic[T]):
             raise ValueError("Stream was not configured for this pipeline")
         return self.stream
 
-    def get_dead_letter_queue(self) -> Stream[T]:
+    def get_dead_letter_queue(self) -> DeadLetterQueue[T]:
         """Retrieves the pipeline's dead letter queue.
 
         Raises:

moose_lib/dmv2/materialized_view.py

@@ -8,6 +8,7 @@ from typing import Any, Optional, Union, Generic, TypeVar
 from pydantic import BaseModel, ConfigDict, model_validator
 
 from moose_lib import ClickHouseEngines
+from ..utilities.sql import quote_identifier
 from .types import BaseTypedResource, T
 from .olap_table import OlapTable, OlapConfig
 from .sql_resource import SqlResource
@@ -94,10 +95,10 @@ class MaterializedView(SqlResource, BaseTypedResource, Generic[T]):
             raise ValueError("Target table name cannot be the same as the materialized view name")
 
         setup = [
-            f"CREATE MATERIALIZED VIEW IF NOT EXISTS {options.materialized_view_name} TO {target_table.name} AS {options.select_statement}",
-            f"INSERT INTO {target_table.name} {options.select_statement}"
+            f"CREATE MATERIALIZED VIEW IF NOT EXISTS {quote_identifier(options.materialized_view_name)} TO {quote_identifier(target_table.name)} AS {options.select_statement}",
+            f"INSERT INTO {quote_identifier(target_table.name)} {options.select_statement}"
         ]
-        teardown = [f"DROP VIEW IF EXISTS {options.materialized_view_name}"]
+        teardown = [f"DROP VIEW IF EXISTS {quote_identifier(options.materialized_view_name)}"]
 
         super().__init__(
             options.materialized_view_name,

moose_lib/dmv2/olap_table.py

@@ -13,6 +13,7 @@ from pydantic import BaseModel
 from typing import List, Optional, Any, Literal, Union, Tuple, TypeVar, Generic, Iterator
 from moose_lib import ClickHouseEngines
 from ..config.runtime import RuntimeClickHouseConfig
+from ..utilities.sql import quote_identifier
 from .types import TypedMooseResource, T
 from ._registry import _tables
 from ..data_models import Column, is_array_nested_type, is_nested_type, _to_columns
@@ -462,7 +463,7 @@ class OlapTable(TypedMooseResource, Generic[T]):
                 settings["input_format_allow_errors_ratio"] = options.allow_errors_ratio
 
         if is_stream:
-            return table_name, data, settings
+            return quote_identifier(table_name), data, settings
 
         if not isinstance(validated_data, list):
             validated_data = [validated_data]
@@ -475,9 +476,9 @@ class OlapTable(TypedMooseResource, Generic[T]):
             preprocessed_record = self._map_to_clickhouse_record(record_dict)
             dict_data.append(preprocessed_record)
         if not dict_data:
-            return table_name, b"", settings
+            return quote_identifier(table_name), b"", settings
         json_lines = self._to_json_each_row(dict_data)
-        return table_name, json_lines, settings
+        return quote_identifier(table_name), json_lines, settings
 
     def _create_success_result(
         self,
@@ -536,7 +537,7 @@ class OlapTable(TypedMooseResource, Generic[T]):
     ) -> InsertResult[T]:
         successful: List[T] = []
         failed: List[FailedRecord[T]] = []
-        table_name = self._generate_table_name()
+        table_name = quote_identifier(self._generate_table_name())
         records_dict = []
         for record in records:
             if hasattr(record, 'model_dump'):
@@ -685,7 +686,8 @@ class OlapTable(TypedMooseResource, Generic[T]):
 
                 if len(batch) >= 1000:  # Batch size
                     json_lines = self._to_json_each_row(batch)
-                    sql = f"INSERT INTO {table_name} FORMAT JSONEachRow"
+                    quoted = quote_identifier(table_name)
+                    sql = f"INSERT INTO {quoted} FORMAT JSONEachRow"
                     # Add wait_end_of_query to batch settings using helper function
                     batch_settings = self._with_wait_end_settings(settings)
                     client.command(sql, data=json_lines, settings=batch_settings)
@@ -694,7 +696,8 @@ class OlapTable(TypedMooseResource, Generic[T]):
 
             if batch:  # Insert any remaining records
                 json_lines = self._to_json_each_row(batch)
-                sql = f"INSERT INTO {table_name} FORMAT JSONEachRow"
+                quoted = quote_identifier(table_name)
+                sql = f"INSERT INTO {quoted} FORMAT JSONEachRow"
                 # Add wait_end_of_query to final batch settings using helper function
                 final_settings = self._with_wait_end_settings(settings)
                 client.command(sql, data=json_lines, settings=final_settings)

{moose_lib-0.6.25.dist-info → moose_lib-0.6.26.dist-info}/METADATA

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

{moose_lib-0.6.25.dist-info → moose_lib-0.6.26.dist-info}/RECORD

@@ -15,10 +15,10 @@ moose_lib/dmv2/__init__.py,sha256=3DVAtNMZUoP94CMJBFhuXfYEQXDbQUNKSgg9XnKqae0,27
 moose_lib/dmv2/_registry.py,sha256=RweMQWHSvhxgSceZ5mfd0NGnJ-49VZORbaFmw-O0bFY,611
 moose_lib/dmv2/consumption.py,sha256=KJV9MKgrnVLIRhlFVZMDhTsXvOaZoMiHk4hnRePdPWA,8745
 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/ingest_pipeline.py,sha256=2tDM4gCjQMdDRDjvymI5Z1DlkXULx9nY4nG9M3YXpjk,7033
 moose_lib/dmv2/life_cycle.py,sha256=wl0k6yzwU1MJ_fO_UkN29buoY5G6ChYZvfwigP9fVfM,1254
-moose_lib/dmv2/materialized_view.py,sha256=xlYNXiOHhjGq5pE8z0dMrVs-Db01DnOlY14CmLwMDGE,4851
-moose_lib/dmv2/olap_table.py,sha256=4DfF_KMCspCvQLf7DwmkmfyFswCv8J-3NhDcnNTWvKg,31304
+moose_lib/dmv2/materialized_view.py,sha256=sNwYwNlNNoaYdmlNHBNuJXp7uc4CThxtYkYPNUaeQpw,4968
+moose_lib/dmv2/olap_table.py,sha256=WT2ZQ-G4xwLOfpKSG1Duk-DHlFFTtQQzGiMG8hc-uWg,31525
 moose_lib/dmv2/registry.py,sha256=janjYszIXDnMfuwkNhoiBmMLar3umQ3Npw9Ms-u6p-8,2373
 moose_lib/dmv2/sql_resource.py,sha256=kUZoGqxhZMHMthtBZGYJBxTFjXkspXiWLXhJRYXgGUM,1864
 moose_lib/dmv2/stream.py,sha256=jiUWBsjFalLLP63mikOxyHRdieiDAlzf9lXfLye-Wjc,10761
@@ -31,7 +31,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.6.25.dist-info/METADATA,sha256=Hq_MwwVkWIuvtATrnipVeGmricUmokjIhJa5Oh1NShY,730
-moose_lib-0.6.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-moose_lib-0.6.25.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
-moose_lib-0.6.25.dist-info/RECORD,,
+moose_lib-0.6.26.dist-info/METADATA,sha256=ckDoVeS60CWtZ2jx-PRM0lWTPlmwq89UIJltD87cmxc,730
+moose_lib-0.6.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+moose_lib-0.6.26.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
+moose_lib-0.6.26.dist-info/RECORD,,