query-cache-common 1.8.0__tar.gz → 1.9.1__tar.gz

{query_cache_common-1.8.0 → query_cache_common-1.9.1}/.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.8.0 → query_cache_common-1.9.1}/PKG-INFO

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

{query_cache_common-1.8.0 → query_cache_common-1.9.1}/src/query_cache_common/auth.py

@@ -75,6 +75,16 @@ class Scope:
         """Returns all organization IDs from the scope, including wildcards."""
         return list(_org_id_to_level(self._org_scopes))
 
+    @cached_property
+    def disabled_org_ids(self) -> t.List[str]:
+        """Returns organization IDs present in :app: scopes but absent from :org: scopes.
+
+        These are organizations the user is associated with but whose access has been disabled.
+        """
+        active_org_ids = set(self.org_ids)
+        app_org_ids = _org_id_to_level(self._app_scopes)
+        return [org_id for org_id in app_org_ids if org_id not in active_org_ids]
+
     def is_org_id_in_scope(self, org_id: str) -> bool:
         """Check if the given organization ID is included in the scope.
 

{query_cache_common-1.8.0 → query_cache_common-1.9.1}/src/query_cache_common/utils.py

@@ -32,17 +32,20 @@ def current_epoch_millis() -> int:
 
 
 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"
         )