ari/minerva
1
0
Fork 0
forked from bellwether/minerva
Code Pull requests Activity

Adding ResultConfiguration for specific output location for Athena

Browse source
This commit is contained in:
Roshan Punnoose 2023-08-03 14:28:12 -04:00
parent 72fd7fe005
commit 757bc74bfd
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
minerva/athena.py
View file

@ -44,11 +44,14 @@ class Execute:
return self.sql return self.sql
def run(self): def run(self):
config = {"OutputLocation": self.athena.output}
if self.params: if self.params:
resp = self.client.start_query_execution(QueryString=self.query(), resp = self.client.start_query_execution(QueryString=self.query(),
ResultConfiguration=config,
ExecutionParameters=self.params) ExecutionParameters=self.params)
else: else:
resp = self.client.start_query_execution(QueryString=self.query()) resp = self.client.start_query_execution(QueryString=self.query(),
ResultConfiguration=config)
self.query_id = resp['QueryExecutionId'] self.query_id = resp['QueryExecutionId']
return resp return resp