diff --git a/aws/logs_monitoring/lambda_function.py b/aws/logs_monitoring/lambda_function.py index 454f5c7fb..90aad3b24 100644 --- a/aws/logs_monitoring/lambda_function.py +++ b/aws/logs_monitoring/lambda_function.py @@ -112,7 +112,12 @@ def compileRegex(rule, pattern): # DD_API_KEY: Datadog API Key DD_API_KEY = "" -if "DD_KMS_API_KEY" in os.environ: +if "DD_API_KEY_SECRET_ARN" in os.environ: + SECRET_ARN = os.environ["DD_API_KEY_SECRET_ARN"] + DD_API_KEY = boto3.client("secretsmanager").get_secret_value( + SecretId=SECRET_ARN + )["SecretString"] +elif "DD_KMS_API_KEY" in os.environ: ENCRYPTED = os.environ["DD_KMS_API_KEY"] DD_API_KEY = boto3.client("kms").decrypt( CiphertextBlob=base64.b64decode(ENCRYPTED)