lint

Author: nhulston

datadog_lambda/api.py

@@ -66,7 +66,9 @@ def get_api_key() -> str:
 
     REGION = os.environ.get("AWS_REGION", "")
     is_gov_region = REGION.startswith("us-gov-")
-    use_non_fips_endpoints = os.environ.get("DD_AWS_USE_NON_FIPS_ENDPOINTS", "").lower() == "true"
+    use_non_fips_endpoints = (
+        os.environ.get("DD_AWS_USE_NON_FIPS_ENDPOINTS", "").lower() == "true"
+    )
 
     if DD_API_KEY_SECRET_ARN:
         if is_gov_region and not use_non_fips_endpoints:
@@ -75,8 +77,7 @@ def get_api_key() -> str:
         else:
             fips_endpoint = None
         secrets_manager_client = boto3.client(
-            "secretsmanager",
-            endpoint_url=fips_endpoint
+            "secretsmanager", endpoint_url=fips_endpoint
         )
         api_key = secrets_manager_client.get_secret_value(
             SecretId=DD_API_KEY_SECRET_ARN
@@ -87,10 +88,7 @@ def get_api_key() -> str:
             fips_endpoint = f"https://ssm-fips.{REGION}.amazonaws.com"
         else:
             fips_endpoint = None
-        ssm_client = boto3.client(
-            "ssm",
-            endpoint_url=fips_endpoint
-        )
+        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"]
@@ -100,10 +98,7 @@ def get_api_key() -> str:
             fips_endpoint = f"https://kms-fips.{REGION}.amazonaws.com"
         else:
             fips_endpoint = None
-        kms_client = boto3.client(
-            "kms",
-            endpoint_url=fips_endpoint
-        )
+        kms_client = boto3.client("kms", endpoint_url=fips_endpoint)
         api_key = decrypt_kms_api_key(kms_client, DD_KMS_API_KEY)
     else:
         api_key = DD_API_KEY