|
|
@@ -0,0 +1,55 @@
|
|
|
+import boto3
|
|
|
+import base64
|
|
|
+import json
|
|
|
+from botocore.exceptions import ClientError
|
|
|
+
|
|
|
+
|
|
|
+def get_secret():
|
|
|
+
|
|
|
+ secret_name = "api_secret"
|
|
|
+ region_name = "cn-northwest-1"
|
|
|
+
|
|
|
+ session = boto3.session.Session()
|
|
|
+ client = session.client(
|
|
|
+ service_name='secretsmanager',
|
|
|
+ region_name=region_name
|
|
|
+ )
|
|
|
+
|
|
|
+ try:
|
|
|
+ get_secret_value_response = client.get_secret_value(
|
|
|
+ SecretId=secret_name
|
|
|
+ )
|
|
|
+ except ClientError as e:
|
|
|
+ if e.response['Error']['Code'] == 'DecryptionFailureException':
|
|
|
+ raise e
|
|
|
+ elif e.response['Error']['Code'] == 'InternalServiceErrorException':
|
|
|
+ raise e
|
|
|
+ elif e.response['Error']['Code'] == 'InvalidParameterException':
|
|
|
+ raise e
|
|
|
+ elif e.response['Error']['Code'] == 'InvalidRequestException':
|
|
|
+ raise e
|
|
|
+ elif e.response['Error']['Code'] == 'ResourceNotFoundException':
|
|
|
+ raise e
|
|
|
+ else:
|
|
|
+ if 'SecretString' in get_secret_value_response:
|
|
|
+ # secret = json.loads(get_secret_value_response['SecretString'])
|
|
|
+ return json.loads(get_secret_value_response['SecretString'])
|
|
|
+ else:
|
|
|
+ # decoded_binary_secret = base64.b64decode(get_secret_value_response['SecretBinary'])
|
|
|
+ return base64.b64decode(get_secret_value_response['SecretBinary'])
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+# {'username': 'admin', 'password': '12345678', 'engine': 'mysql', 'host': 'database-1.ce6iux1s3ccn.rds.cn-northwest-1.amazonaws.com.cn', 'port': 3306, 'dbInstanceIdentifier': 'database-1'}
|
|
|
+
|
|
|
+get_value = get_secret()['username']
|
|
|
+username = get_value['username']
|
|
|
+password = get_value['password']
|
|
|
+host = get_value['host']
|
|
|
+port = get_value['port']
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|