lint

Author: nhulston

datadog_lambda/api.py

@@ -71,7 +71,11 @@ def get_api_key() -> str:
 
     if DD_API_KEY_SECRET_ARN:
         # Secrets manager endpoints: https://docs.aws.amazon.com/general/latest/gr/asm.html
-        fips_endpoint = f"https://secretsmanager-fips.{REGION}.amazonaws.com" if is_gov_region else None
+        fips_endpoint = (
+            f"https://secretsmanager-fips.{REGION}.amazonaws.com"
+            if is_gov_region
+            else None
+        )
         secrets_manager_client = boto3.client(
             "secretsmanager", endpoint_url=fips_endpoint
         )
@@ -80,14 +84,22 @@ def get_api_key() -> str:
         )["SecretString"]
     elif DD_API_KEY_SSM_NAME:
         # SSM endpoints: https://docs.aws.amazon.com/general/latest/gr/ssm.html
-        fips_endpoint = f"https://ssm-fips.{REGION}.amazonaws.com" if is_gov_region else None
+        fips_endpoint = (
+            f"https://ssm-fips.{REGION}.amazonaws.com"
+            if is_gov_region
+            else None
+        )
         ssm_client = boto3.client("ssm", endpoint_url=fips_endpoint)
         api_key = ssm_client.get_parameter(
             Name=DD_API_KEY_SSM_NAME, WithDecryption=True
         )["Parameter"]["Value"]
     elif DD_KMS_API_KEY:
         # KMS endpoints: https://docs.aws.amazon.com/general/latest/gr/kms.html
-        fips_endpoint = f"https://kms-fips.{REGION}.amazonaws.com" if is_gov_region else None
+        fips_endpoint = (
+            f"https://kms-fips.{REGION}.amazonaws.com"
+            if is_gov_region
+            else None
+        )
         kms_client = boto3.client("kms",endpoint_url=fips_endpoint)
         api_key = decrypt_kms_api_key(kms_client, DD_KMS_API_KEY)
     else: