def __poll_status(self, query_execution_id):
status = self.get_query_status(query_execution_id)
if status in ['SUCCEEDED', 'FAILED']:
return status
else:
raise Exceptions.QueryExecutionTimeoutException(
"Query to athena has timed out. Try running the query in the athena or asynchronously"
)
def __poll_status(self, query_execution_id):
res = self.__athena.get_query_execution(
QueryExecutionId=query_execution_id)
status = res['QueryExecution']['Status']['State']
if status == 'SUCCEEDED':
return status
elif status == 'FAILED':
return status
else:
raise Exceptions.QueryExecutionTimeoutException(
"Query to athena has timed out. Try running in query in the athena"
)