quillsql 2.2.5__tar.gz → 2.2.6__tar.gz

{quillsql-2.2.5 → quillsql-2.2.6}/PKG-INFO

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

{quillsql-2.2.5 → quillsql-2.2.6}/quillsql/core.py

@@ -396,13 +396,24 @@ class Quill:
                 return query
             return f"{query.rstrip(';')} limit {limit}"
 
+    def normalize_database_type(self, db_type):
+        if not db_type:
+            return None
+        lowered = db_type.lower()
+        if lowered in ("postgresql", "postgres"):
+            return "postgres"
+        return lowered
+
     def run_queries(
         self, queries, pkDatabaseType, databaseType=None, metadata=None, runQueryConfig=None
     ):
         results = {}
         if not queries:
             return {"queryResults": []}
-        if databaseType and databaseType.lower() != pkDatabaseType.lower():
+        normalized_pk = self.normalize_database_type(pkDatabaseType)
+        normalized_requested = self.normalize_database_type(databaseType)
+        should_enforce_match = normalized_requested is not None
+        if should_enforce_match and normalized_requested != normalized_pk:
             return {"dbMismatched": True, "backendDatabaseType": pkDatabaseType}
         if runQueryConfig and runQueryConfig.get("arrayToMap"):
             mapped_array = array_to_map(

{quillsql-2.2.5 → quillsql-2.2.6}/quillsql/db/bigquery.py

@@ -26,8 +26,7 @@ def format_bigquery_config(connection_string):
             "project": service_account.get("project_id"),
             "credentials": service_account,
         }
-    except (ValueError, TypeError) as e:
-        print("Invalid JSON string: ", e)
+    except (ValueError, TypeError):
     return connection_string
 
 

{quillsql-2.2.5 → quillsql-2.2.6}/quillsql/db/db_helper.py

@@ -15,47 +15,51 @@ from quillsql.db.bigquery import (
 )
 
 
+def _is_postgres(database_type):
+    return database_type and database_type.lower() in ("postgresql", "postgres")
+
+
 def get_db_credentials(database_type, connection_string):
-    if database_type.lower() == "postgresql":
+    if _is_postgres(database_type):
         return format_postgres(connection_string)
-    elif database_type.lower() == "bigquery":
+    elif database_type and database_type.lower() == "bigquery":
         return format_bigquery_config(connection_string)
     return {}
 
 
 def connect_to_db(database_type, config, using_connection_string):
-    if database_type.lower() == "postgresql":
+    if _is_postgres(database_type):
         return connect_to_postgres(config, using_connection_string)
-    elif database_type.lower() == "bigquery":
+    elif database_type and database_type.lower() == "bigquery":
         return connect_to_bigquery(config, using_connection_string)
     return None
 
 
 def run_query_by_db(database_type, query, connection):
-    if database_type.lower() == "postgresql":
+    if _is_postgres(database_type):
         return run_query_postgres(query, connection)
-    elif database_type.lower() == "bigquery":
+    elif database_type and database_type.lower() == "bigquery":
         return run_query_big_query(query, connection)
     return None
 
 
 def disconnect_from_db(database_type, connection):
-    if database_type.lower() == "postgresql":
+    if _is_postgres(database_type):
         return disconnect_from_postgres(connection)
     return None
 
 
 def get_schema_tables_by_db(database_type, connection, schema_name):
-    if database_type.lower() == "postgresql":
+    if _is_postgres(database_type):
         return get_tables_by_schema_postgres(connection, schema_name)
-    elif database_type.lower() == "bigquery":
+    elif database_type and database_type.lower() == "bigquery":
         return get_tables_by_schema_big_query(connection, schema_name)
     return None
 
 
 def get_schema_column_info_by_db(database_type, connection, schema_name, table_names):
-    if database_type.lower() == "postgresql":
+    if _is_postgres(database_type):
         return get_schema_column_info_postgres(connection, schema_name, table_names)
-    elif database_type.lower() == "bigquery":
+    elif database_type and database_type.lower() == "bigquery":
         return get_schema_column_info_big_query(connection, schema_name, table_names)
     return None

{quillsql-2.2.5 → quillsql-2.2.6}/quillsql/utils/pivot_template.py

@@ -198,7 +198,6 @@ def parse_distinct_values(query_result: Dict[str, Any], database_type: str) -> L
             distinct_values = [v.strip() for v in row["string_values"].split(",")]
     
     else:
-        print(f"Warning: Unknown database type: {database_type}")
         distinct_values = []
     
     # Filter out null/undefined/empty values

{quillsql-2.2.5 → quillsql-2.2.6}/quillsql.egg-info/PKG-INFO

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

{quillsql-2.2.5 → quillsql-2.2.6}/setup.py

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