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

Merge branch 'add_result_config' into 'main'

Browse source 
Adding ResultConfiguration for specific output location for Athena

See merge request pmo/minerva!2
This commit is contained in:
Ari Brown 2023-08-03 18:30:19 +00:00
commit 1d509f544c
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