moose-lib 0.6.50__tar.gz → 0.6.52__tar.gz

{moose_lib-0.6.50 → moose_lib-0.6.52}/PKG-INFO

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

{moose_lib-0.6.50 → moose_lib-0.6.52}/moose_lib/main.py

@@ -19,6 +19,8 @@ from string import Formatter
 from temporalio.client import Client as TemporalClient, TLSConfig
 from temporalio.common import RetryPolicy, WorkflowIDConflictPolicy, WorkflowIDReusePolicy
 from datetime import timedelta
+from time import perf_counter
+from humanfriendly import format_timespan
 from .config.runtime import RuntimeClickHouseConfig
 
 from moose_lib.commons import EnhancedJSONEncoder
@@ -182,6 +184,7 @@ class QueryClient:
     def execute(self, input, variables, row_type: Type[BaseModel] = None):
         params = {}
         values = {}
+        preview_params = {}
 
         for i, (_, variable_name, _, _) in enumerate(Formatter().parse(input)):
             if variable_name:
@@ -196,20 +199,63 @@ class QueryClient:
 
                 params[variable_name] = f'{{p{i}: {t}}}'
                 values[f'p{i}'] = value
+                preview_params[variable_name] = self._format_value_for_preview(value)
+
         clickhouse_query = input.format_map(params)
+        preview_query = input.format_map(preview_params)
 
         # We are not using the result of the ping
         # but this ensures that if the clickhouse cloud service is idle, we
         # wake it up, before we send the query.
         self.ch_client.ping()
 
+        print(f"[QueryClient] | Query: {' '.join(preview_query.split())}")
+        start = perf_counter()
         val = self.ch_client.query(clickhouse_query, values)
+        secs = perf_counter() - start
+        if secs < 1:
+            print(f"[QueryClient] | Query completed: {secs * 1000:.0f} ms")
+        else:
+            print(f"[QueryClient] | Query completed: {format_timespan(secs)}")
 
         if row_type is None:
             return list(val.named_results())
         else:
             return list(row_type(**row) for row in val.named_results())
 
+    def _format_value_for_preview(self, value: Any) -> str:
+        """Format a Python value as a ClickHouse SQL literal for preview logging.
+
+        This does not affect execution; it's only for human-readable query logs.
+        """
+        # NULL handling
+        if value is None:
+            return 'NULL'
+
+        # Booleans (ClickHouse accepts true/false)
+        if isinstance(value, bool):
+            return 'true' if value else 'false'
+
+        # Numbers
+        if isinstance(value, (int, float)) and not isinstance(value, bool):
+            return str(value)
+
+        # Strings
+        if isinstance(value, str):
+            # Escape backslashes and single quotes for ClickHouse single-quoted strings
+            escaped = value.replace('\\', '\\\\').replace("'", "\\'")
+            return f"'{escaped}'"
+
+        # Lists / tuples (format as [item1, item2, ...])
+        if isinstance(value, (list, tuple)):
+            formatted_items = ', '.join(self._format_value_for_preview(v) for v in value)
+            return f"[{formatted_items}]"
+
+        # Fallback: stringify and single-quote
+        fallback = str(value)
+        escaped_fallback = fallback.replace('\\', '\\\\').replace("'", "\\'")
+        return f"'{escaped_fallback}'"
+
     def close(self):
         """Close the ClickHouse client connection."""
         if self.ch_client:

{moose_lib-0.6.50 → moose_lib-0.6.52}/moose_lib.egg-info/PKG-INFO

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