query-cache-common 1.9.0__tar.gz → 1.9.2__tar.gz

{query_cache_common-1.9.0 → query_cache_common-1.9.2}/.gitignore

@@ -213,9 +213,11 @@ data/
 # Service account credentials
 .creds/
 
-# Load test generated files
-tooling/load_test/data/*.json
+# Load test generated files (flat JSONs + timestamped run dirs)
+tooling/load_test/data/
 tooling/load_test/proto/
+# Active k8s load-test run ID written by k8s prepare-run.
+tooling/load_test/k8s/.run-id
 
 # Turbo integration test
 tests/integration/dbt_run_cache_turbo/projects/
@@ -229,3 +231,6 @@ docs/superpowers/
 
 # Git worktrees
 .worktrees/
+
+# Google Antigravity CLI (agy)
+.antigravitycli/

{query_cache_common-1.9.0 → query_cache_common-1.9.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: query-cache-common
-Version: 1.9.0
+Version: 1.9.2
 Summary: Common code for Query Cache shared across client and server
 License: 
                                          Apache License

{query_cache_common-1.9.0 → query_cache_common-1.9.2}/src/query_cache_common/utils.py

@@ -26,23 +26,45 @@ def str_to_bool(s: str | None) -> bool:
     return s.lower() in ("true", "1", "t", "y", "yes", "on")
 
 
+def to_bool(value: t.Any) -> bool:
+    """Coerce a bool, int, or string to a boolean.
+
+    Unlike str_to_bool, this accepts bools and ints directly (strings are delegated to
+    str_to_bool) and raises ValueError for any other type rather than silently
+    returning False.
+    """
+    if isinstance(value, bool):
+        return value
+    if isinstance(value, str):
+        return str_to_bool(value)
+    if isinstance(value, int):
+        return bool(value)
+    raise ValueError(
+        f"Cannot convert value of type {type(value).__name__} to bool: {value!r}. "
+        "Expected bool, int, or string (e.g., 'true', 'false', 'yes', 'no')."
+    )
+
+
 def current_epoch_millis() -> int:
     """Return the current time in milliseconds since Unix epoch."""
     return time.time_ns() // 1_000_000
 
 
 def extract_select_from_ctas(create: CreateStatement) -> SqlglotExpression:
-    """Extract the SELECT statement from a CREATE TABLE / VIEW AS statement.
+    """Extract the query from a CREATE TABLE / VIEW AS statement.
+
+    The query body may be a single SELECT or a set operation (UNION / INTERSECT /
+    EXCEPT), optionally wrapped in parentheses.
 
     Args:
         create: The CTAS statement.
 
     Returns:
-        The extracted SELECT statement.
+        The extracted query expression.
     """
     from sqlglot import exp
 
-    if not isinstance(create.expression, (exp.Subquery, exp.Select)):
+    if not isinstance(create.expression, (exp.Subquery, exp.Query)):
         raise ValueError(
             "Invalid CREATE statement. Only CREATE ... AS SELECT statements are supported"
         )