From b96537de2ecc46f1f8bc05714e1667d1e6f01493 Mon Sep 17 00:00:00 2001 From: Akira Ajisaka Date: Tue, 23 Jan 2024 17:29:48 +0900 Subject: [PATCH] Revert "Fix wrong role create db#285 (#286)" This reverts commit 1416de86faea5b61ebd2b5ac16f862d8ec39f32e. --- dbt/adapters/glue/credentials.py | 2 +- dbt/adapters/glue/impl.py | 14 +++++++------- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/dbt/adapters/glue/credentials.py b/dbt/adapters/glue/credentials.py index 0ada1166..fb55e92e 100644 --- a/dbt/adapters/glue/credentials.py +++ b/dbt/adapters/glue/credentials.py @@ -29,7 +29,7 @@ class GlueCredentials(Credentials): seed_mode: Optional[str] = "overwrite" default_arguments: Optional[str] = None iceberg_glue_commit_lock_table: Optional[str] = "myGlueLockTable" - use_interactive_session_role_for_api_calls: bool = True + use_interactive_session_role_for_api_calls: bool = False lf_tags: Optional[str] = None glue_session_id: Optional[str] = None glue_session_reuse: Optional[bool] = False diff --git a/dbt/adapters/glue/impl.py b/dbt/adapters/glue/impl.py index f99aafc7..6d67bcb5 100644 --- a/dbt/adapters/glue/impl.py +++ b/dbt/adapters/glue/impl.py @@ -87,15 +87,15 @@ def get_connection(self): RoleSessionName="dbt" ) credentials = assumed_role_object['Credentials'] - glue_client = boto3.client("glue", region_name=glueSession.credentials.region, - aws_access_key_id=credentials['AccessKeyId'], - aws_secret_access_key=credentials['SecretAccessKey'], - aws_session_token=credentials['SessionToken']) - return glueSession, glue_client + session = boto3.Session( + aws_access_key_id=credentials['AccessKeyId'], + aws_secret_access_key=credentials['SecretAccessKey'], + aws_session_token=credentials['SessionToken'] + ) - glue_client = boto3.client("glue", region_name=glueSession.credentials.region) + client = boto3.client("glue", region_name=glueSession.credentials.region) - return glueSession, glue_client + return glueSession, client def list_schemas(self, database: str) -> List[str]: session, client = self.get_connection()