diff --git a/order-events/app/events/set_as_expired.py b/order-events/app/events/set_as_expired.py index 4432676..59aaf43 100644 --- a/order-events/app/events/set_as_expired.py +++ b/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_, } )