Add the option to run the queries sync

.gitignore

@@ -129,4 +129,7 @@ dmypy.json
 .pyre/
 
 # visual studio code
-.vscode
+.vscode
+
+# PyCharm
+.idea

main.py

@@ -10,6 +10,7 @@
 def main():
     load_dotenv() # only on local run
     queries_list = os.environ['INPUT_QUERIES'].split(';')
+    sync = os.environ.get("INPUT_SYNC", False)
     warehouse = os.environ['INPUT_SNOWFLAKE_WAREHOUSE']
     snowflake_account = os.environ['INPUT_SNOWFLAKE_ACCOUNT']
     snowflake_username = os.environ['INPUT_SNOWFLAKE_USERNAME']
@@ -33,7 +34,10 @@ def main():
             print(f"[!] Query id - {query_result.query_id}")
             print(f"[!] Running query ### - {query}")
 
-        json_results = asyncio.run(utils.gather_all_results(query_results))
+        if not sync:
+            json_results = asyncio.run(utils.gather_all_results(query_results))
+        else:
+            json_results = utils.gather_all_results_sync(query_results)
 
     utils.set_github_action_output('queries_results', json.dumps(json_results))
 

snowflake_connector.py

@@ -64,4 +64,8 @@ async def fetch_results(self):
         while self.is_query_running():
             await asyncio.sleep(0.1)
 
-        return self._fetch_results()
+        return self._fetch_results()
+
+    def fetch_results_sync(self):
+        self.cursor.get_results_from_sfqid(self.query_id)
+        return self.cursor.fetchall()

utils.py

@@ -35,4 +35,19 @@ async def gather_all_results(query_result_list: List[QueryResult]) -> dict:
                 print(row)
                 json_total_results[done_task.get_name()].append(str(row))
 
+    return json_total_results
+
+
+def gather_all_results_sync(query_results_list: List[QueryResult]) -> dict:
+    """
+
+    Args:
+        query_results_list:
+
+    Returns:
+
+    """
+    json_total_results = {}
+    for query in query_results_list:
+        json_total_results[query.query_id] = query.fetch_results_sync()
     return json_total_results