handle empty responses

IanRFerguson · Jun 7, 2024 · c5bae0f · c5bae0f
1 parent 3835de2
commit c5bae0f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 100 deletions.
diff --git a/klondike/snowflake/snowflake.py b/klondike/snowflake/snowflake.py
@@ -123,7 +123,15 @@ def __query(self, sql: str):
                 _cursor.execute(sql)
                 _resp = _cursor.fetch_arrow_batches()
 
-        return pl.from_arrow(_resp)
+        try:
+            return pl.from_arrow(_resp)
+        except ValueError as ve:
+            # NOTE - This appears to be Polars response to empty fetch_arrow_batches()
+            # This should be interrogated more, but is functional
+            if "Must pass schema, or at least one RecordBatch" in str(ve):
+                logger.debug("No results obtained via query")
+                return pl.DataFrame()
+            raise
 
     def read_dataframe(self, sql: str) -> pl.DataFrame:
         """

diff --git a/tests/test_bigquery_connector.py b/tests/test_bigquery_connector.py