fixing black formatter issues

Author: cyphronix

aws_sra_examples/solutions/genai/bedrock_org/lambda/rules/sra_bedrock_check_guardrail_encryption/app.py

@@ -60,11 +60,9 @@ def evaluate_compliance(rule_parameters: dict) -> tuple[str, str]:  # noqa: CFQ0
         return "COMPLIANT", "All Bedrock guardrails are encrypted with a KMS key"
 
     except ClientError as e:
-        if e.response['Error']['Code'] == 'AccessDeniedException':
+        if e.response["Error"]["Code"] == "AccessDeniedException":
             LOGGER.info("Access denied. If guardrail uses KMS encryption, ensure Lambda's IAM role has permissions to the KMS key.")
-            return "NON_COMPLIANT", (
-                "Access denied. If guardrail uses KMS encryption, ensure Lambda's IAM role has permissions to the KMS key."
-            )
+            return "NON_COMPLIANT", ("Access denied. If guardrail uses KMS encryption, ensure Lambda's IAM role has permissions to the KMS key.")
         LOGGER.error(f"Error evaluating Bedrock guardrails encryption: {str(e)}")
         return "ERROR", f"Error evaluating compliance: {str(e)}"
 

aws_sra_examples/solutions/genai/bedrock_org/lambda/rules/sra_bedrock_check_guardrails/app.py

@@ -97,7 +97,7 @@ def lambda_handler(event: dict, context: Any) -> dict:  # noqa: CCR001, C901, U1
         except bedrock.exceptions.ResourceNotFoundException:
             LOGGER.warning(f"Guardrail {guardrail_name} (ID: {guardrail_id}) not found")
         except ClientError as client_error:
-            if client_error.response['Error']['Code'] == 'AccessDeniedException':
+            if client_error.response["Error"]["Code"] == "AccessDeniedException":
                 LOGGER.info(
                     f"Access denied to guardrail {guardrail_name} (ID: {guardrail_id}). "
                     + "If guardrail uses KMS encryption, ensure Lambda's IAM role has permissions to the KMS key."

aws_sra_examples/solutions/genai/bedrock_org/lambda/rules/sra_bedrock_check_invocation_log_s3/app.py

@@ -74,7 +74,7 @@ def evaluate_compliance(rule_parameters: dict) -> tuple[str, str]:  # noqa: CFQ0
                 if not any(rule.get("Expiration") for rule in lifecycle.get("Rules", [])):
                     issues.append("retention not set")
             except botocore.exceptions.ClientError as client_error:
-                if client_error.response['Error']['Code'] == 'NoSuchLifecycleConfiguration':
+                if client_error.response["Error"]["Code"] == "NoSuchLifecycleConfiguration":
                     LOGGER.info(f"No lifecycle configuration found for S3 bucket: {bucket_name}")
                     issues.append("lifecycle not set")
 

aws_sra_examples/solutions/genai/bedrock_org/lambda/src/sra_dynamodb.py

@@ -96,7 +96,7 @@ def create_table(self, table_name: str) -> None:
                 TableName=table_name,
                 KeySchema=key_schema,
                 AttributeDefinitions=attribute_definitions,
-                BillingMode='PAY_PER_REQUEST'  # on-demand capacity mode
+                BillingMode="PAY_PER_REQUEST",  # on-demand capacity mode
             )
             self.LOGGER.info(f"{table_name} dynamodb table created successfully.")
         except Exception as e:

aws_sra_examples/solutions/genai/bedrock_org/lambda/src/sra_iam.py

@@ -105,9 +105,11 @@ def create_role(self, role_name: str, trust_policy: dict, solution_name: str) ->
         """
         self.LOGGER.info("Creating role %s.", role_name)
         try:
-            return dict(self.IAM_CLIENT.create_role(
-                RoleName=role_name, AssumeRolePolicyDocument=json.dumps(trust_policy), Tags=[{"Key": "sra-solution", "Value": solution_name}]
-            ))
+            return dict(
+                self.IAM_CLIENT.create_role(
+                    RoleName=role_name, AssumeRolePolicyDocument=json.dumps(trust_policy), Tags=[{"Key": "sra-solution", "Value": solution_name}]
+                )
+            )
         except ClientError as error:
             if error.response["Error"]["Code"] == "EntityAlreadyExists":
                 self.LOGGER.info(f"{role_name} role already exists!")
@@ -126,9 +128,11 @@ def create_policy(self, policy_name: str, policy_document: dict, solution_name:
         """
         self.LOGGER.info(f"Creating {policy_name} IAM policy")
         try:
-            return dict(self.IAM_CLIENT.create_policy(
-                PolicyName=policy_name, PolicyDocument=json.dumps(policy_document), Tags=[{"Key": "sra-solution", "Value": solution_name}]
-            ))
+            return dict(
+                self.IAM_CLIENT.create_policy(
+                    PolicyName=policy_name, PolicyDocument=json.dumps(policy_document), Tags=[{"Key": "sra-solution", "Value": solution_name}]
+                )
+            )
         except ClientError as error:
             if error.response["Error"]["Code"] == "EntityAlreadyExists":
                 self.LOGGER.info(f"{policy_name} policy already exists!")

aws_sra_examples/solutions/genai/bedrock_org/lambda/src/sra_s3.py

@@ -92,7 +92,7 @@ def create_s3_bucket(self, bucket: str) -> None:
                 ACL="private",
                 Bucket=bucket,
                 CreateBucketConfiguration={"LocationConstraint": self.REGION},  # type: ignore
-                ObjectOwnership="BucketOwnerPreferred"
+                ObjectOwnership="BucketOwnerPreferred",
             )
         else:
             create_bucket = self.S3_CLIENT.create_bucket(ACL="private", Bucket=bucket, ObjectOwnership="BucketOwnerPreferred")