Update to record is_replay

Author: ConnorKirk

aws_lambda_powertools/utilities/idempotency/base.py

@@ -123,17 +123,17 @@ def __init__(
 
         self.persistence_store = persistence_store
 
-    def handle(self, durable_mode: str | None = None) -> Any:
+    def handle(self, is_replay: bool = False) -> Any:
         """
         Main entry point for handling idempotent execution of a function.
 
         Parameters
         ----------
-        durable_mode : str | None, optional
-            Mode for handling in-progress executions. Options:
-            - "REPLAY_MODE": Allow replay of functions that are already in progress
-            - "EXECUTION_MODE": Standard durable execution mode
-            - None: Standard idempotency behavior (raises IdempotencyAlreadyInProgressError)
+        is_replay : bool, optional
+            Whether this is a replay of a function that is already in progress.
+            If True, allows replay of functions that are already in progress.
+            If False, uses standard idempotency behavior (raises IdempotencyAlreadyInProgressError).
+            Defaults to False.
 
         Returns
         -------
@@ -146,12 +146,12 @@ def handle(self, durable_mode: str | None = None) -> Any:
         # In most cases we can retry successfully on this exception.
         for i in range(MAX_RETRIES + 1):  # pragma: no cover
             try:
-                return self._process_idempotency(durable_mode)
+                return self._process_idempotency(is_replay)
             except IdempotencyInconsistentStateError:
                 if i == MAX_RETRIES:
                     raise  # Bubble up when exceeded max tries
 
-    def _process_idempotency(self, durable_mode: str | None):
+    def _process_idempotency(self, is_replay: bool):
         try:
             # We call save_inprogress first as an optimization for the most common case where no idempotent record
             # already exists. If it succeeds, there's no need to call get_record.
@@ -167,7 +167,7 @@ def _process_idempotency(self, durable_mode: str | None):
             # We give preference to ReturnValuesOnConditionCheckFailure because it is a faster and more cost-effective
             # way of retrieving the existing record after a failed conditional write operation.
             record = exc.old_data_record or self._get_idempotency_record()
-            if durable_mode == "REPLAY_MODE":
+            if is_replay:
                 return self._get_function_response()
             # If a record is found, handle it for status
             if record:

aws_lambda_powertools/utilities/idempotency/idempotency.py

@@ -95,11 +95,11 @@ def handler(event, context):
     if hasattr(context, "state"):
         # Extract lambda_context from DurableContext for idempotency tracking
         config.register_lambda_context(context.lambda_context)
-        durable_mode = "REPLAY_MODE" if len(context.state.operations) > 1 else "EXECUTION_MODE"
+        is_replay = len(context.state.operations) > 1
     else:
         # Standard LambdaContext
         config.register_lambda_context(context)
-        durable_mode = None
+        is_replay = False
 
     args = event, context
     idempotency_handler = IdempotencyHandler(
@@ -112,7 +112,7 @@ def handler(event, context):
         function_kwargs=kwargs,
     )
 
-    return idempotency_handler.handle(durable_mode=durable_mode)
+    return idempotency_handler.handle(is_replay=is_replay)
 
 
 def idempotent_function(