40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from aws_lambda_powertools import Logger
|
|
from aws_lambda_powertools.utilities.data_classes import (
|
|
EventBridgeEvent,
|
|
event_source,
|
|
)
|
|
from aws_lambda_powertools.utilities.typing import LambdaContext
|
|
from layercake.dynamodb import (
|
|
DynamoDBPersistenceLayer,
|
|
PartitionKey,
|
|
)
|
|
|
|
from boto3clients import dynamodb_client
|
|
from config import ORDER_TABLE
|
|
|
|
logger = Logger(__name__)
|
|
dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
|
|
|
|
|
|
@event_source(data_class=EventBridgeEvent)
|
|
@logger.inject_lambda_context
|
|
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
|
|
keys = event.detail['keys']
|
|
order_id = keys['id']
|
|
|
|
r = dyn.collection.query(PartitionKey(order_id), limit=150)
|
|
logger.info('Records found', total_items=len(r['items']), records=r['items'])
|
|
|
|
with dyn.batch_writer() as batch:
|
|
for pair in r['items']:
|
|
batch.delete_item(
|
|
Key={
|
|
'id': {'S': pair['id']},
|
|
'sk': {'S': pair['sk']},
|
|
}
|
|
)
|
|
|
|
logger.info('Deleted all records')
|
|
|
|
return True
|