10
10
11
11
logger = logging .getLogger (__name__ )
12
12
13
- logging .basicConfig (
14
- level = logging .DEBUG ,
15
- format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s' ,
16
- handlers = [
17
- logging .StreamHandler (),
18
- ]
19
- )
20
-
21
- logger = logging .getLogger (__name__ )
22
13
23
14
class RedshiftDataSource (DataSource ):
24
15
TYPE = "redshift"
@@ -64,10 +55,8 @@ def __get_cluster_credentials(self, aws_credentials: AwsCredentials):
64
55
resolved_aws_credentials = aws_credentials .resolve_role (
65
56
role_session_name = "soda_redshift_get_cluster_credentials"
66
57
)
67
- logger .debug (f"Resolved AWS Credentials: { resolved_aws_credentials } " )
68
58
69
59
region_name = resolved_aws_credentials .region_name or "us-west-2"
70
- logger .debug (f"Region Name: { region_name } " )
71
60
72
61
client = boto3 .client (
73
62
"redshift" ,
@@ -76,28 +65,13 @@ def __get_cluster_credentials(self, aws_credentials: AwsCredentials):
76
65
aws_secret_access_key = resolved_aws_credentials .secret_access_key ,
77
66
aws_session_token = resolved_aws_credentials .session_token ,
78
67
)
79
- logger .debug ("Boto3 Redshift client initialized." )
80
68
81
69
cluster_name = "bi-edw-db"
82
70
username = self .dbuser if self .dbuser else self .username
83
71
db_name = self .dbname if self .dbname else self .database
84
-
85
- logger .debug (f"Using cluster name: { cluster_name } " )
86
- logger .debug (f"Using DbUser: { username } " )
87
- logger .debug (f"Using DbName: { db_name } " )
88
-
89
- try :
90
- cluster_creds = client .get_cluster_credentials (
91
- DbUser = username ,
92
- DbName = db_name ,
93
- ClusterIdentifier = cluster_name ,
94
- AutoCreate = False ,
95
- DurationSeconds = 3600
96
- )
97
- logger .debug (f"Cluster Credentials Retrieved: { cluster_creds } " )
98
- except botocore .exceptions .ClientError as error :
99
- logger .error (f"Error retrieving cluster credentials: { error } " )
100
- raise
72
+ cluster_creds = client .get_cluster_credentials (
73
+ DbUser = username , DbName = db_name , ClusterIdentifier = cluster_name , AutoCreate = False , DurationSeconds = 3600
74
+ )
101
75
102
76
return cluster_creds ["DbUser" ], cluster_creds ["DbPassword" ]
103
77
0 commit comments