amazon_secretsmanager_get_secret.py

from boto3.session import Session
import base64
from botocore.exceptions import ClientError


def get_secret(secret_name, region_name='us-west-2'):
    secret_value_response = {}
    try:
        secret_value_response = (
            Session(region_name=region_name)
            .client(service_name='secretsmanager')
            .get_secret_value(SecretId=secret_name)
        )
    except ClientError as exception:
        if exception.response['Error']['Code'] == 'DecryptionFailureException':
            raise exception
        elif exception.response['Error']['Code'] == 'InternalServiceErrorException':
            raise exception
        elif exception.response['Error']['Code'] == 'InvalidParameterException':
            raise exception
        elif exception.response['Error']['Code'] == 'InvalidRequestException':
            raise exception
        elif exception.response['Error']['Code'] == 'ResourceNotFoundException':
            raise exception
        else:
            if 'SecretString' in secret_value_response:
                secret = secret_value_response['SecretString']
            else:
                secret = base64.b64decode(secret_value_response['SecretBinary'])

            return secret