Documentation

Author: leandrodamascena

docs/utilities/data_classes.md

@@ -1083,80 +1083,42 @@ This example is based on the AWS S3 Batch Operations documentation [Example Lamb
 
 === "app.py"
 
-    ```python  hl_lines="4-8 17-19 24-26 38-40 55-57 61-63 72 75 77"
+    ```python  hl_lines="5-8 13 15 26 31 33 35"
     import boto3
+
     from botocore.exceptions import ClientError
 
     from aws_lambda_powertools.utilities.data_classes import (
         S3BatchOperationEvent,
-        S3BatchOperationResult,
         S3BatchOperationResponse,
+        event_source
     )
+    from aws_lambda_powertools.utilities.typing import LambdaContext
 
-    import logging
-
-    logger = logging.getLogger(__name__)
-    logger.setLevel("INFO")
 
-    s3 = boto3.client("s3")
+    @event_source(data_class=S3BatchOperationEvent)
+    def lambda_handler(event: S3BatchOperationEvent, context: LambdaContext):
+        response = S3BatchOperationResponse(event.invocation_schema_version, event.invocation_id, "PermanentFailure")
 
-    @event_route(data_class=S3BatchOperationEvent)
-    def lambda_handler(event: S3BatchOperationEvent, context):
-        response = S3BatchOperationResponse(event.invocation_schema_version, event.invocation_id)
         result = None
         task = event.task
+        src_key: str = task.s3_key
+        src_bucket: str = task.s3_bucket
+
+        s3 = boto3.client("s3", region_name='us-east-1')
 
         try:
-            obj_key = task.s3_key
-            obj_version_id = task.s3_version_id
-            bucket_name = task.s3_bucket
-
-            logger.info(
-                "Got task: remove delete marker %s from object %s.", obj_version_id, obj_key
-            )
-
-            try:
-                # If this call does not raise an error, the object version is not a delete
-                # marker and should not be deleted.
-                head_response = s3.head_object(
-                    Bucket=bucket_name, Key=obj_key, VersionId=obj_version_id
-                )
-                result = S3BatchOperationResult.as_permanent_failure(task,
-                    f"Object {obj_key}, ID {obj_version_id} is not a delete marker."
-                )
-
-                logger.debug(head_response)
-            except ClientError as error:
-                delete_marker = error.response["ResponseMetadata"]["HTTPHeaders"].get(
-                    "x-amz-delete-marker", "false"
-                )
-                if delete_marker == "true":
-                    logger.info(
-                        "Object %s, version %s is a delete marker.", obj_key, obj_version_id
-                    )
-                    try:
-                        s3.delete_object(
-                            Bucket=bucket_name, Key=obj_key, VersionId=obj_version_id
-                        )
-                        result = S3BatchOperationResult.as_succeeded(task,
-                            f"Successfully removed delete marker {obj_version_id} from object {obj_key}."
-                        )
-                    except ClientError as error:
-                        # Mark request timeout as a temporary failure so it will be retried.
-                        if error.response["Error"]["Code"] == "RequestTimeout":
-                            result = S3BatchOperationResult.as_temporary_failure(
-                                task, f"Attempt to remove delete marker from object {obj_key} timed out."
-                            )
-                        else:
-                            raise
-                else:
-                    raise ValueError(
-                        f"The x-amz-delete-marker header is either not present or is not 'true'."
-                    )
-        except Exception as error:
-            # Mark all other exceptions as permanent failures.
-            result = S3BatchOperationResult.as_permanent_failure(task, str(error))
-            logger.exception(error)
+            dest_bucket, dest_key = do_some_work(s3, src_bucket, src_key)
+            result = task.build_task_batch_response("Succeeded", f"s3://{dest_bucket}/{dest_key}")
+        except ClientError as e:
+            error_code = e.response['Error']['Code']
+            error_message = e.response['Error']['Message']
+            if error_code == 'RequestTimeout':
+                result = task.build_task_batch_response("TemporaryFailure", "Retry request to Amazon S3 due to timeout.")
+            else:
+                result = task.build_task_batch_response("PermanentFailure", f"{error_code}: {error_message}")
+        except Exception as e:
+            result = task.build_task_batch_response("PermanentFailure", str(e))
         finally:
             response.add_result(result)