Enable Ruff format

leandrodamascena · leandrodamascena · commit 21ee75cf495d · 2025-04-17T17:11:56.000+01:00
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -12,13 +12,13 @@ repos:
       - id: check-toml
   - repo: local
     hooks:
-      - id: black
-        name: formatting::black
-        entry: poetry run black
+      - id: ruff
+        name: formatting::ruff
+        entry: poetry run ruff format
         language: system
         types: [python]
       - id: ruff
-        name: linting-format::ruff
+        name: linting::ruff
         entry: poetry run ruff check
         language: system
         types: [python]
diff --git a/Makefile b/Makefile
@@ -26,7 +26,7 @@ check-licenses:
 	poetry run licensecheck -u poetry:dev
 
 format:
-	poetry run black aws_lambda_powertools tests examples
+	poetry run ruff format aws_lambda_powertools tests examples
 
 lint: format
 	poetry run ruff check aws_lambda_powertools tests examples
diff --git a/aws_lambda_powertools/event_handler/api_gateway.py b/aws_lambda_powertools/event_handler/api_gateway.py
@@ -1604,7 +1604,6 @@ def _validate_response_validation_error_http_code(
         response_validation_error_http_code: HTTPStatus | int | None,
         enable_validation: bool,
     ) -> HTTPStatus:
-
         if response_validation_error_http_code and not enable_validation:
             msg = "'response_validation_error_http_code' cannot be set when enable_validation is False."
             raise ValueError(msg)
@@ -1613,7 +1612,6 @@ def _validate_response_validation_error_http_code(
             not isinstance(response_validation_error_http_code, HTTPStatus)
             and response_validation_error_http_code is not None
         ):
-
             try:
                 response_validation_error_http_code = HTTPStatus(response_validation_error_http_code)
             except ValueError:
diff --git a/aws_lambda_powertools/event_handler/openapi/models.py b/aws_lambda_powertools/event_handler/openapi/models.py
@@ -36,7 +36,6 @@ class OpenAPIExtensions(BaseModel):
     @model_validator(mode="before")
     def serialize_openapi_extension_v2(self):
         if isinstance(self, dict) and self.get("openapi_extensions"):
-
             openapi_extension_value = self.get("openapi_extensions")
 
             for extension_key in openapi_extension_value:
diff --git a/aws_lambda_powertools/logging/logger.py b/aws_lambda_powertools/logging/logger.py
@@ -242,7 +242,6 @@ def __init__(
         buffer_config: LoggerBufferConfig | None = None,
         **kwargs,
     ) -> None:
-
         # Used in case of sampling
         self.initial_log_level = self._determine_log_level(level)
 
diff --git a/aws_lambda_powertools/utilities/batch/base.py b/aws_lambda_powertools/utilities/batch/base.py
@@ -296,8 +296,7 @@ def _clean(self):
 
         if self._entire_batch_failed() and self.raise_on_entire_batch_failure:
             raise BatchProcessingError(
-                msg=f"All records failed processing. {len(self.exceptions)} individual errors logged "
-                f"separately below.",
+                msg=f"All records failed processing. {len(self.exceptions)} individual errors logged separately below.",
                 child_exceptions=self.exceptions,
             )
 
diff --git a/aws_lambda_powertools/utilities/data_classes/code_pipeline_job_event.py b/aws_lambda_powertools/utilities/data_classes/code_pipeline_job_event.py
@@ -314,7 +314,6 @@ def put_artifact(self, artifact_name: str, body: Any, content_type: str) -> None
         # So we are using if/else instead.
 
         if self.data.encryption_key:
-
             encryption_key_id = self.data.encryption_key.get_id
             encryption_key_type = self.data.encryption_key.get_type
             if encryption_key_type == "KMS":
diff --git a/aws_lambda_powertools/utilities/data_classes/transfer_family_event.py b/aws_lambda_powertools/utilities/data_classes/transfer_family_event.py
@@ -39,7 +39,6 @@ def source_ip(self) -> str:
 
 
 class TransferFamilyAuthorizerResponse:
-
     def _build_authentication_response(
         self,
         role_arn: str,
@@ -51,7 +50,6 @@ def _build_authentication_response(
         user_uid: int | None = None,
         public_keys: str | None = None,
     ) -> dict[str, Any]:
-
         response: dict[str, Any] = {}
 
         if home_directory_type == "PATH":
diff --git a/aws_lambda_powertools/utilities/data_masking/provider/base.py b/aws_lambda_powertools/utilities/data_masking/provider/base.py
@@ -82,7 +82,6 @@ def erase(
         masking_rules: dict | None = None,
         **kwargs,
     ) -> Any:
-
         result: Any = DATA_MASKING_STRING
 
         if not any([dynamic_mask, custom_mask, regex_pattern, mask_format, masking_rules]):
diff --git a/aws_lambda_powertools/utilities/idempotency/serialization/dataclass.py b/aws_lambda_powertools/utilities/idempotency/serialization/dataclass.py
@@ -38,7 +38,6 @@ def from_dict(self, data: dict) -> DataClass:
 
     @classmethod
     def instantiate(cls, model_type: Any) -> BaseIdempotencySerializer:
-
         model_type = get_actual_type(model_type=model_type)
 
         if model_type is None:
diff --git a/aws_lambda_powertools/utilities/idempotency/serialization/pydantic.py b/aws_lambda_powertools/utilities/idempotency/serialization/pydantic.py
@@ -35,7 +35,6 @@ def from_dict(self, data: dict) -> BaseModel:
 
     @classmethod
     def instantiate(cls, model_type: Any) -> BaseIdempotencySerializer:
-
         model_type = get_actual_type(model_type=model_type)
 
         if model_type is None:
diff --git a/examples/batch_processing/src/getting_started_error_handling.py b/examples/batch_processing/src/getting_started_error_handling.py
@@ -12,8 +12,7 @@
 logger = Logger()
 
 
-class InvalidPayload(Exception):
-    ...
+class InvalidPayload(Exception): ...
 
 
 @tracer.capture_method
diff --git a/examples/data_masking/src/aws_encryption_provider_example.py b/examples/data_masking/src/aws_encryption_provider_example.py
@@ -16,7 +16,8 @@
     local_cache_capacity=200,
     max_cache_age_seconds=400,
     max_messages_encrypted=200,
-    max_bytes_encrypted=2000)
+    max_bytes_encrypted=2000,
+)
 
 data_masker = DataMasking(provider=encryption_provider)
 
diff --git a/examples/event_handler_bedrock_agents/cdk/bedrock_agent_stack.py b/examples/event_handler_bedrock_agents/cdk/bedrock_agent_stack.py
@@ -8,7 +8,6 @@
 
 
 class AgentsCdkStack(Stack):
-
     def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
         super().__init__(scope, construct_id, **kwargs)
 
diff --git a/examples/event_handler_graphql/src/enable_exceptions_batch_resolver.py b/examples/event_handler_graphql/src/enable_exceptions_batch_resolver.py
@@ -14,8 +14,7 @@
 }
 
 
-class PostRelatedNotFound(Exception):
-    ...
+class PostRelatedNotFound(Exception): ...
 
 
 @app.batch_resolver(type_name="Query", field_name="relatedPosts", raise_on_error=True)  # (1)!
diff --git a/examples/event_handler_rest/src/raising_http_errors.py b/examples/event_handler_rest/src/raising_http_errors.py
@@ -66,6 +66,7 @@ def service_error():
 def service_unavailable_error():
     raise ServiceUnavailableError("Service is temporarily unavailable")  # HTTP 503
 
+
 @app.get("/todos")
 @tracer.capture_method
 def get_todos():
diff --git a/examples/event_sources/src/cloudWatchDashboard.py b/examples/event_sources/src/cloudWatchDashboard.py
@@ -26,6 +26,6 @@ def lambda_handler(event: CloudWatchDashboardCustomWidgetEvent, context):
         "markdown": f"""
         Dashboard: {event.widget_context.dashboard_name}
         Time Range: {time_range.start} to {time_range.end}
-        Theme: {event.widget_context.theme or 'default'}
+        Theme: {event.widget_context.theme or "default"}
         """,
     }
diff --git a/examples/event_sources/src/s3_batch_operation.py b/examples/event_sources/src/s3_batch_operation.py
@@ -33,5 +33,4 @@ def lambda_handler(event: S3BatchOperationEvent, context: LambdaContext):
     return response.asdict()
 
 
-def do_some_work(s3_client, src_bucket: str, src_key: str):
-    ...
+def do_some_work(s3_client, src_bucket: str, src_key: str): ...
diff --git a/examples/homepage/install/arm64/cdk_arm64.py b/examples/homepage/install/arm64/cdk_arm64.py
@@ -3,7 +3,6 @@
 
 
 class SampleApp(Stack):
-
     def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
         super().__init__(scope, construct_id, **kwargs)
 
diff --git a/examples/homepage/install/sar/cdk_sar.py b/examples/homepage/install/sar/cdk_sar.py
@@ -10,7 +10,6 @@
 
 
 class SampleApp(Stack):
-
     def __init__(self, scope: Construct, id_: str) -> None:
         super().__init__(scope, id_)
 
diff --git a/examples/homepage/install/x86_64/cdk_x86.py b/examples/homepage/install/x86_64/cdk_x86.py
@@ -3,7 +3,6 @@
 
 
 class SampleApp(Stack):
-
     def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
         super().__init__(scope, construct_id, **kwargs)
 
diff --git a/examples/idempotency/src/working_with_custom_idempotency_key_prefix.py b/examples/idempotency/src/working_with_custom_idempotency_key_prefix.py
@@ -22,7 +22,7 @@ class Payment:
 class PaymentError(Exception): ...
 
 
-@idempotent(persistence_store=persistence_layer, key_prefix="my_custom_prefix") # (1)!
+@idempotent(persistence_store=persistence_layer, key_prefix="my_custom_prefix")  # (1)!
 def lambda_handler(event: dict, context: LambdaContext):
     try:
         payment: Payment = create_subscription_payment(event)
diff --git a/examples/idempotency/src/working_with_custom_idempotency_key_prefix_standalone.py b/examples/idempotency/src/working_with_custom_idempotency_key_prefix_standalone.py
@@ -29,15 +29,15 @@ class Order:
     data_keyword_argument="order",
     config=config,
     persistence_store=dynamodb,
-    key_prefix="my_custom_prefix", # (1)!
+    key_prefix="my_custom_prefix",  # (1)!
 )
-def process_order(order: Order): 
+def process_order(order: Order):
     return f"processed order {order.order_id}"
 
 
 def lambda_handler(event: dict, context: LambdaContext):
     # see Lambda timeouts section
-    config.register_lambda_context(context) 
+    config.register_lambda_context(context)
 
     order_item = OrderItem(sku="fake", description="sample")
     order = Order(item=order_item, order_id=1)
diff --git a/examples/jmespath_functions/src/powertools_json_idempotency_jmespath.py b/examples/jmespath_functions/src/powertools_json_idempotency_jmespath.py
@@ -16,8 +16,7 @@
 config = IdempotencyConfig(event_key_jmespath="powertools_json(body)")
 
 
-class PaymentError(Exception):
-    ...
+class PaymentError(Exception): ...
 
 
 @idempotent(config=config, persistence_store=persistence_layer)
diff --git a/examples/logger/src/append_keys_vs_extra.py b/examples/logger/src/append_keys_vs_extra.py
@@ -8,8 +8,7 @@
 logger = Logger(service="payment")
 
 
-class PaymentError(Exception):
-    ...
+class PaymentError(Exception): ...
 
 
 def lambda_handler(event, context):
diff --git a/examples/metrics/src/capture_cold_start_metric.py b/examples/metrics/src/capture_cold_start_metric.py
@@ -5,5 +5,4 @@
 
 
 @metrics.log_metrics(capture_cold_start_metric=True)
-def lambda_handler(event: dict, context: LambdaContext):
-    ...
+def lambda_handler(event: dict, context: LambdaContext): ...
diff --git a/examples/metrics/src/flush_metrics.py b/examples/metrics/src/flush_metrics.py
@@ -5,7 +5,7 @@
 metrics = Metrics()
 
 
-def book_flight(flight_id: str, **kwargs): 
+def book_flight(flight_id: str, **kwargs):
     # logic to book flight
     ...
     metrics.add_metric(name="SuccessfulBooking", unit=MetricUnit.Count, value=1)
diff --git a/examples/metrics/src/single_metric_with_different_timestamp.py b/examples/metrics/src/single_metric_with_different_timestamp.py
@@ -6,9 +6,7 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     for record in event:
-
         record_id: str = record.get("record_id")
         amount: int = record.get("amount")
         timestamp: int = record.get("timestamp")
diff --git a/examples/parameters/src/builtin_provider_dynamodb_custom_endpoint.py b/examples/parameters/src/builtin_provider_dynamodb_custom_endpoint.py
@@ -9,7 +9,6 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Usually an endpoint is not sensitive data, so we store it in DynamoDB Table
         endpoint_comments: Any = dynamodb_provider.get("comments_endpoint")
diff --git a/examples/parameters/src/builtin_provider_dynamodb_recursive_parameter.py b/examples/parameters/src/builtin_provider_dynamodb_recursive_parameter.py
@@ -9,15 +9,13 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Retrieve multiple parameters using HASH KEY
         all_parameters: Any = dynamodb_provider.get_multiple("config")
         endpoint_comments = "https://jsonplaceholder.typicode.com/noexists/"
         limit = 2
 
         for parameter, value in all_parameters.items():
-
             if parameter == "endpoint_comments":
                 endpoint_comments = value
 
diff --git a/examples/parameters/src/builtin_provider_dynamodb_single_parameter.py b/examples/parameters/src/builtin_provider_dynamodb_single_parameter.py
@@ -9,7 +9,6 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Usually an endpoint is not sensitive data, so we store it in DynamoDB Table
         endpoint_comments: Any = dynamodb_provider.get("comments_endpoint")
diff --git a/examples/parameters/src/builtin_provider_secret.py b/examples/parameters/src/builtin_provider_secret.py
@@ -11,7 +11,6 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Usually an endpoint is not sensitive data, so we store it in SSM Parameters
         endpoint_comments: Any = parameters.get_parameter("/lambda-powertools/endpoint_comments")
diff --git a/examples/parameters/src/builtin_provider_ssm_with_no_recursive.py b/examples/parameters/src/builtin_provider_ssm_with_no_recursive.py
@@ -8,8 +8,7 @@
 ssm_provider = parameters.SSMProvider()
 
 
-class ConfigNotFound(Exception):
-    ...
+class ConfigNotFound(Exception): ...
 
 
 def lambda_handler(event: dict, context: LambdaContext):
@@ -22,7 +21,6 @@ def lambda_handler(event: dict, context: LambdaContext):
         endpoint_comments = "https://jsonplaceholder.typicode.com/comments/"
 
         for parameter, value in all_parameters.items():
-
             # query parameter is used to query endpoint
             if "query" in parameter:
                 endpoint_comments = f"{endpoint_comments}{value}"
diff --git a/examples/parameters/src/getting_started_secret.py b/examples/parameters/src/getting_started_secret.py
@@ -7,7 +7,6 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Usually an endpoint is not sensitive data, so we store it in SSM Parameters
         endpoint_comments: Any = parameters.get_parameter("/lambda-powertools/endpoint_comments")
diff --git a/examples/parameters/src/recursive_ssm_parameter_force_fetch.py b/examples/parameters/src/recursive_ssm_parameter_force_fetch.py
@@ -13,7 +13,6 @@ def lambda_handler(event: dict, context: LambdaContext):
         endpoint_comments = "https://jsonplaceholder.typicode.com/noexists/"
 
         for parameter, value in all_parameters.items():
-
             if parameter == "endpoint_comments":
                 endpoint_comments = value
 
diff --git a/examples/parameters/src/recursive_ssm_parameter_with_cache.py b/examples/parameters/src/recursive_ssm_parameter_with_cache.py
@@ -13,7 +13,6 @@ def lambda_handler(event: dict, context: LambdaContext):
         endpoint_comments = "https://jsonplaceholder.typicode.com/noexists/"
 
         for parameter, value in all_parameters.items():
-
             if parameter == "endpoint_comments":
                 endpoint_comments = value
 
diff --git a/examples/parameters/src/secret_force_fetch.py b/examples/parameters/src/secret_force_fetch.py
@@ -7,7 +7,6 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Usually an endpoint is not sensitive data, so we store it in SSM Parameters
         endpoint_comments: Any = parameters.get_parameter("/lambda-powertools/endpoint_comments")
diff --git a/examples/parameters/src/secret_with_cache.py b/examples/parameters/src/secret_with_cache.py
@@ -7,7 +7,6 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Usually an endpoint is not sensitive data, so we store it in SSM Parameters
         endpoint_comments: Any = parameters.get_parameter("/lambda-powertools/endpoint_comments")
diff --git a/examples/parameters/src/working_with_own_provider_s3.py b/examples/parameters/src/working_with_own_provider_s3.py
@@ -12,7 +12,6 @@
 
 
 def lambda_handler(event: dict, context: LambdaContext):
-
     try:
         # Retrieve a single parameter using key
         endpoint_comments: Any = s3_provider.get("comments_endpoint")
diff --git a/examples/tracer/src/disable_capture_error.py b/examples/tracer/src/disable_capture_error.py
@@ -9,8 +9,7 @@
 ENDPOINT = os.getenv("PAYMENT_API", "")
 
 
-class PaymentError(Exception):
-    ...
+class PaymentError(Exception): ...
 
 
 @tracer.capture_method(capture_error=False)
diff --git a/examples/tracer/src/ignore_endpoints.py b/examples/tracer/src/ignore_endpoints.py
diff --git a/examples/validation/src/getting_started_validator_decorator_function.py b/examples/validation/src/getting_started_validator_decorator_function.py
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
diff --git a/tests/e2e/utils/infrastructure.py b/tests/e2e/utils/infrastructure.py
diff --git a/tests/functional/event_handler/_pydantic/test_openapi_config.py b/tests/functional/event_handler/_pydantic/test_openapi_config.py
diff --git a/tests/functional/event_handler/_pydantic/test_openapi_with_pep563.py b/tests/functional/event_handler/_pydantic/test_openapi_with_pep563.py
diff --git a/tests/functional/event_handler/required_dependencies/appsync/test_appsync_batch_resolvers.py b/tests/functional/event_handler/required_dependencies/appsync/test_appsync_batch_resolvers.py
diff --git a/tests/functional/feature_flags/_boto3/test_schema_validation.py b/tests/functional/feature_flags/_boto3/test_schema_validation.py
diff --git a/tests/functional/logger/required_dependencies/test_logger.py b/tests/functional/logger/required_dependencies/test_logger.py
diff --git a/tests/functional/logger/required_dependencies/test_powertools_logger_buffer.py b/tests/functional/logger/required_dependencies/test_powertools_logger_buffer.py
diff --git a/tests/unit/data_classes/_boto3/test_code_pipeline_job_event.py b/tests/unit/data_classes/_boto3/test_code_pipeline_job_event.py
diff --git a/tests/unit/data_classes/required_dependencies/test_dynamo_db_stream_event.py b/tests/unit/data_classes/required_dependencies/test_dynamo_db_stream_event.py
diff --git a/tests/unit/data_classes/required_dependencies/test_transfer_family_event.py b/tests/unit/data_classes/required_dependencies/test_transfer_family_event.py
diff --git a/tests/unit/event_handler/_pydantic/conftest.py b/tests/unit/event_handler/_pydantic/conftest.py
diff --git a/tests/unit/event_handler/_pydantic/test_openapi_models_pydantic_v2.py b/tests/unit/event_handler/_pydantic/test_openapi_models_pydantic_v2.py
diff --git a/tests/unit/logger/required_dependencies/test_logger_buffer_cache.py b/tests/unit/logger/required_dependencies/test_logger_buffer_cache.py
diff --git a/tests/unit/parser/_pydantic/test_apigw_websockets.py b/tests/unit/parser/_pydantic/test_apigw_websockets.py
diff --git a/tests/unit/parser/_pydantic/test_iot_registry_events.py b/tests/unit/parser/_pydantic/test_iot_registry_events.py

Original file line number	Diff line number	Diff line change
`@@ -296,8 +296,7 @@ def _clean(self):`
`296`	`296`
`297`	`297`	`if self._entire_batch_failed() and self.raise_on_entire_batch_failure:`
`298`	`298`	`raise BatchProcessingError(`
`299`		`- msg=f"All records failed processing. {len(self.exceptions)} individual errors logged "`
`300`		`- f"separately below.",`
	`299`	`+ msg=f"All records failed processing. {len(self.exceptions)} individual errors logged separately below.",`
`301`	`300`	`child_exceptions=self.exceptions,`
`302`	`301`	`)`
`303`	`302`