quillsql 2.2.3__tar.gz → 2.2.4__tar.gz

{quillsql-2.2.3 → quillsql-2.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quillsql
-Version: 2.2.3
+Version: 2.2.4
 Summary: Quill SDK for Python.
 Home-page: https://github.com/quill-sql/quill-python
 Author: Quill

{quillsql-2.2.3 → quillsql-2.2.4}/quillsql/core.py

@@ -129,13 +129,14 @@ class Quill:
                     return {
                         "status": "error",
                         "error": pivot_template_response.get("error"),
-                        "data": pivot_template_response.get("metadata") or {},
+                        "data": (pivot_template_response.get("metadata") or {}),
                     }
 
-                template = pivot_template_response.get("metadata", {}).get("template")
-                config = pivot_template_response.get("metadata", {}).get("config")
-                distinct_values_query = pivot_template_response.get("metadata", {}).get("distinctValuesQuery")
-                row_count_query = pivot_template_response.get("metadata", {}).get("rowCountQuery")
+                pivot_metadata = pivot_template_response.get("metadata") or {}
+                template = pivot_metadata.get("template")
+                config = pivot_metadata.get("config") or {}
+                distinct_values_query = pivot_metadata.get("distinctValuesQuery")
+                row_count_query = pivot_metadata.get("rowCountQuery")
 
                 # Step 2: Run the distinct values query to get unique values
                 distinct_values = []
@@ -180,10 +181,10 @@ class Quill:
                     self.target_connection.database_type,
                     metadata.get("databaseType"),
                     metadata,
-                    pivot_template_response.get("metadata", {}).get("runQueryConfig")
+                    pivot_metadata.get("runQueryConfig"),
                 )
 
-                responseMetadata = pivot_template_response.get("metadata") or {}
+                responseMetadata = pivot_metadata or {}
                 # Set rows and fields from first query result (the pivot query)
                 if final_results.get("queryResults") and len(final_results["queryResults"]) >= 1:
                     query_results = final_results["queryResults"][0]
@@ -228,7 +229,7 @@ class Quill:
                     return {
                         'status': 'error',
                         'error': response.get('error'),
-                        'data': response.get('metadata') or {},
+                        'data': (response.get('metadata') or {}),
                     }
                 
                 flag_query_results = self.run_queries(
@@ -237,7 +238,8 @@ class Quill:
                 )
                 
                 tenant_flags = []
-                query_order = response.get('metadata', {}).get('queryOrder') or []
+                response_metadata = response.get('metadata') or {}
+                query_order = response_metadata.get('queryOrder') or []
                 query_results = (flag_query_results or {}).get('queryResults') or []
                 for tenant_field, query_result in zip(query_order, query_results):
                     rows = []

{quillsql-2.2.3 → quillsql-2.2.4}/quillsql.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quillsql
-Version: 2.2.3
+Version: 2.2.4
 Summary: Quill SDK for Python.
 Home-page: https://github.com/quill-sql/quill-python
 Author: Quill

{quillsql-2.2.3 → quillsql-2.2.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="quillsql",
-    version="2.2.3",
+    version="2.2.4",
     packages=find_packages(),
     install_requires=[
         "psycopg2-binary",

{quillsql-2.2.3 → quillsql-2.2.4}/tests/test_core.py

@@ -51,3 +51,48 @@ def test_query_handles_none_tenant_flag_responses(mock_cached_connection):
     assert result["status"] == "success"
     assert result["data"] == {}
 
+
+@patch("quillsql.core.CachedConnection")
+def test_query_handles_tenant_flag_metadata_none(mock_cached_connection):
+    mock_connection = MagicMock()
+    mock_connection.database_type = "PostgreSQL"
+    mock_cached_connection.return_value = mock_connection
+
+    quill = Quill(
+        private_key="test-key",
+        database_type="PostgreSQL",
+        database_config={},
+    )
+
+    quill.run_queries = MagicMock(
+        side_effect=[
+            {"queryResults": [{"rows": [{"quill_flag": "alpha"}]}]},
+            {"queryResults": []},
+        ]
+    )
+
+    def post_quill_side_effect(path, payload):
+        if path == "tenant-mapped-flags":
+            return {
+                "queries": [
+                    'SELECT * FROM (SELECT "id" AS "customer_id", "name" AS "quill_label", "id" AS "quill_flag" FROM "public"."customers") AS "subq_owner_query" WHERE "customer_id" IN (2)'
+                ],
+                "metadata": None,
+            }
+        return {"queries": [], "metadata": {}}
+
+    quill.post_quill = MagicMock(side_effect=post_quill_side_effect)
+
+    result = quill.query(
+        tenants=[{"tenantField": "customer_id", "tenantIds": [2]}],
+        metadata={
+            "task": "report",
+            "clientId": "client-123",
+            "reportId": "report-456",
+            "databaseType": "PostgreSQL",
+        },
+    )
+
+    assert result["status"] == "success"
+    assert result["data"] == {}
+