From fec94523ab5f83f6a8089f9951a9a8ee77a0df5c Mon Sep 17 00:00:00 2001 From: laughingman7743 Date: Tue, 10 Jan 2017 19:49:24 +0900 Subject: [PATCH] Fix credentials provider configuration --- pyathenajdbc/connection.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pyathenajdbc/connection.py b/pyathenajdbc/connection.py index 43bd432..002add6 100644 --- a/pyathenajdbc/connection.py +++ b/pyathenajdbc/connection.py @@ -89,14 +89,14 @@ def _build_driver_args(self, **kwargs): props = jpype.java.util.Properties() if self.credential_file: props.setProperty('aws_credentials_provider_class', + 'com.amazonaws.athena.jdbc.shaded.' + 'com.amazonaws.auth.PropertiesFileCredentialsProvider') props.setProperty('aws_credentials_provider_arguments', self.credential_file) - if self.token: + elif self.token: props.setProperty('aws_credentials_provider_class', - 'com.amazonaws.athena.jdbc.CustomSessionCredentialsProvider') - props.setProperty('aws_credentials_provider_arguments', - ','.join([self.access_key, self.secret_key, self.token])) + 'com.amazonaws.athena.jdbc.shaded.' + + 'com.amazonaws.auth.InstanceProfileCredentialsProvider') else: props.setProperty('user', self.access_key) props.setProperty('password', self.secret_key)