update key

order-events/app/events/set_as_expired.py

@@ -6,7 +6,6 @@ from aws_lambda_powertools.utilities.data_classes import (
 from aws_lambda_powertools.utilities.typing import LambdaContext
 from layercake.dateutils import now
 from layercake.dynamodb import (
-    ComposeKey,
     DynamoDBPersistenceLayer,
     KeyPair,
 )
@@ -24,6 +23,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     """Set the order status to `EXPIRED` 24 hours after it becomes overdue."""
     new_image = event.detail['new_image']
     now_ = now()
+    sk = new_image['sk']
 
     try:
         order_layer.update_item(
@@ -45,7 +45,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         order_layer.put_item(
             item={
                 'id': new_image['id'],
-                'sk': ComposeKey('failed', prefix=new_image['sk']),
+                'sk': f'{sk}#failed',
                 'created_at': now_,
             }
         )
@@ -56,7 +56,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         order_layer.put_item(
             item={
                 'id': new_image['id'],
-                'sk': ComposeKey('completed', prefix=new_image['sk']),
+                'sk': f'{sk}#executed',
                 'created_at': now_,
             }
         )