Skip to content

Commit 19bee41

Browse files
authored
Handle paginated results from athena queries (#1196)
1 parent a1b4542 commit 19bee41

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cid/helpers/athena.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -348,7 +348,8 @@ def execute_query(self, sql_query, sleep_duration=1, database: str=None, catalog
348348

349349
def get_query_results(self, query_id):
350350
""" Get Query Results """
351-
return self.client.get_query_results(QueryExecutionId=query_id)
351+
paginator = self.client.get_paginator("get_query_results")
352+
return paginator.paginate(QueryExecutionId=query_id).build_full_result()
352353

353354
def parse_response_as_table(self, response, include_header=False):
354355
""" Return a query response as a table. """

0 commit comments

Comments
 (0)