Created
September 13, 2019 23:10
-
-
Save dannygoldstein/12902104c3fc7cc47e89111d5c9a2a93 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Use this code snippet in your app. | |
# If you need more information about configurations or implementing the sample code, visit the AWS docs: | |
# https://aws.amazon.com/developers/getting-started/python/ | |
import boto3 | |
from botocore.exceptions import ClientError, CredentialRetrievalError, MetadataRetrievalError, NoCredentialsError | |
import json | |
def get_secret(secret_name): | |
region_name = "us-east-1" | |
# Create a Secrets Manager client | |
while True: | |
try: | |
session = boto3.session.Session() | |
client = session.client( | |
service_name='secretsmanager', | |
region_name=region_name | |
) | |
except (CredentialRetrievalError, MetadataRetrievalError, NoCredentialsError): | |
print('encountered AWS ratelimit, retrying...') | |
continue | |
# In this sample we only handle the specific exceptions for the 'GetSecretValue' API. | |
# See https://docs.aws.amazon.com/secretsmanager/latest/apireference/API_GetSecretValue.html | |
# We rethrow the exception by default. | |
try: | |
get_secret_value_response = client.get_secret_value( | |
SecretId=secret_name | |
) | |
except (CredentialRetrievalError, MetadataRetrievalError, NoCredentialsError): | |
print('encountered AWS ratelimit, retrying...') | |
continue | |
except ClientError as e: | |
if e.response['Error']['Code'] == 'DecryptionFailureException': | |
# Secrets Manager can't decrypt the protected secret text using the provided KMS key. | |
# Deal with the exception here, and/or rethrow at your discretion. | |
raise e | |
elif e.response['Error']['Code'] == 'InternalServiceErrorException': | |
# An error occurred on the server side. | |
# Deal with the exception here, and/or rethrow at your discretion. | |
raise e | |
elif e.response['Error']['Code'] == 'InvalidParameterException': | |
# You provided an invalid value for a parameter. | |
# Deal with the exception here, and/or rethrow at your discretion. | |
raise e | |
elif e.response['Error']['Code'] == 'InvalidRequestException': | |
# You provided a parameter value that is not valid for the current state of the resource. | |
# Deal with the exception here, and/or rethrow at your discretion. | |
raise e | |
elif e.response['Error']['Code'] == 'ResourceNotFoundException': | |
# We can't find the resource that you asked for. | |
# Deal with the exception here, and/or rethrow at your discretion. | |
raise e | |
else: | |
# Decrypts secret using the associated KMS CMK. | |
# Depending on whether the secret is a string or binary, one of these fields will be populated. | |
secret = get_secret_value_response['SecretString'] | |
return json.loads(secret) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment