quillsql 2.1.3__py3-none-any.whl → 2.1.5__py3-none-any.whl

quillsql/core.py

@@ -112,7 +112,7 @@ class Quill:
         if not queries:
             return {"queryResults": []}
         if databaseType and databaseType.lower() != pkDatabaseType.lower():
-            return {"dbMismatched": True, "backendDatbaseType": pkDatabaseType}
+            return {"dbMismatched": True, "backendDatabaseType": pkDatabaseType}
         if runQueryConfig and runQueryConfig.get("arrayToMap"):
             array_to_map(
                 queries, runQueryConfig.get("arrayToMap"), metadata, self.target_connection
@@ -141,10 +141,8 @@ class Quill:
                 queries = [query.strip().rstrip(";") + " limit " + runQueryConfig.get("limitBy") for query in queries]
             query_results = [self.target_connection.query(query) for query in queries]
             results["queryResults"] = query_results
-            if runQueryConfig and runQueryConfig.get("removeFields"):
-                results["queryResults"] = remove_fields(
-                    query_results, runQueryConfig.get("removeFields")
-                )
+            if runQueryConfig and runQueryConfig.get("fieldsToRemove"):
+                results["queryResults"] = [ remove_fields(query_result, runQueryConfig.get("fieldsToRemove")) for query_result in results["queryResults"]]
             if runQueryConfig and runQueryConfig.get("convertDatatypes"):
                 for query_result in results["queryResults"]:
                     query_result["fields"] = [

quillsql/utils/run_query_processes.py

@@ -1,18 +1,17 @@
 import json
 
-def remove_fields(query_result, cursor_description, fields_to_remove):
+def remove_fields(query_result, fields_to_remove):
   fields = [
-      {"name": desc[0], "dataTypeID": desc[1]}
-      for desc in cursor_description
-      if desc[0] in fields_to_remove
+      {"name": field['name'], "dataTypeID": field['dataTypeID']}
+      for field in query_result['fields']
+      if field['name'] not in fields_to_remove
   ]
-  rows = [dict(json.loads(json.dumps(row, default=str))) for row in query_result]
-
+  rows = [row for row in query_result['rows']]
   for row in rows:
     for field in fields_to_remove:
       if field in row:
           del row[field]
-  return {fields, rows}
+  return {"fields": fields, "rows": rows}
 
 def array_to_map(queries, array_to_map, metadata, target_pool):
   for i in range(len(queries)):

{quillsql-2.1.3.dist-info → quillsql-2.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quillsql
-Version: 2.1.3
+Version: 2.1.5
 Summary: Quill SDK for Python.
 Home-page: https://github.com/quill-sql/quill-python
 Author: Quill
@@ -11,7 +11,7 @@ Requires-Dist: requests
 Requires-Dist: redis
 Requires-Dist: python-dotenv
 Requires-Dist: pytest
-Requires-Dist: google-cloud
+Requires-Dist: google-cloud-bigquery
 Requires-Dist: google-auth
 
 # Quill Python SDK

{quillsql-2.1.3.dist-info → quillsql-2.1.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 quillsql/__init__.py,sha256=xxVDrfIXLIbt45yHKGYRbqmgXSPe03uNbqENMLUnUC8,70
-quillsql/core.py,sha256=_UmRX70ynssicla9WLuFjahhKdl7p_HPaAXw7vZrzDE,7544
+quillsql/core.py,sha256=YWMRZMIWgZlwccbWeIo8aK8CFxs83YW064shHMQDvtU,7557
 quillsql/error.py,sha256=XltjRgdBcq3lQN0KdUba5eDzA21yQ7KisIjvqqUSELI,161
 quillsql/assets/__init__.py,sha256=PX43SCko4w5P197YXghGI2VAq3-X1J91IO_tWY7rpiM,44
 quillsql/assets/pgtypes.py,sha256=qWAcGO6QfWN6XXtGI7cMreNSeTjw-xNgcc1aLoq9EV0,40768
@@ -9,9 +9,9 @@ quillsql/db/cached_connection.py,sha256=1y_GzW5mIAYpmGKJjxpBqmFafpmgUA0Qm9oW_ufy
 quillsql/db/db_helper.py,sha256=PFAcX2V26sIaR-hF8orGGltKXdEzR5b_enMSOHlAMUg,2054
 quillsql/db/postgres.py,sha256=iw71feFvnCDBMKpeGbhiJ_4Z4h7AlUi4Hk12Pfyzx6c,2799
 quillsql/utils/__init__.py,sha256=jJLp0OkFQdWohu8vIijfggiWf4eWmeAafVzx3YpdE4E,75
-quillsql/utils/run_query_processes.py,sha256=vzqzcYzYoIOMMLpHnvdhCuYTwpeIsSQBTOyQlbecc3g,678
+quillsql/utils/run_query_processes.py,sha256=hElCVACq6bz1_jVIjQUrZzI1CPY188I3_5sHc21jtOc,673
 quillsql/utils/schema_conversion.py,sha256=MwFk95SIZ8OX5EAv141QXLjmCef-1z8jiVKlW71-sk8,297
-quillsql-2.1.3.dist-info/METADATA,sha256=6Qv5Olg9WQI3P96qetKvNppNUY6GmHbT_O5vwWjNs-g,1800
-quillsql-2.1.3.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-quillsql-2.1.3.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
-quillsql-2.1.3.dist-info/RECORD,,
+quillsql-2.1.5.dist-info/METADATA,sha256=_OK0xXtETMyotLC-Jc2moid0HYzkpggOQWOtsyGEG68,1809
+quillsql-2.1.5.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+quillsql-2.1.5.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
+quillsql-2.1.5.dist-info/RECORD,,