add download sample

This commit is contained in:
2025-10-07 20:07:29 -03:00
parent 4fdf98a5b4
commit 08db9d0191
5 changed files with 87 additions and 8 deletions

View File

@@ -21,7 +21,8 @@ from config import (
)
logger = Logger(__name__)
dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
course_layer = DynamoDBPersistenceLayer(COURSE_TABLE, dynamodb_client)
@event_source(data_class=EventBridgeEvent)
@@ -31,11 +32,10 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
now_ = now()
enrollment_id = new_image['id']
course_id = new_image['course']['id']
cert = dyn.collection.get_item(
cert = course_layer.collection.get_item(
KeyPair(
pk=course_id,
sk=SortKey('0', path_spec='cert', rename_key='cert'),
table_name=COURSE_TABLE,
),
raise_on_error=False,
default=False,
@@ -101,7 +101,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
logger.exception(exc)
raise
return dyn.update_item(
return enrollment_layer.update_item(
key=KeyPair(
pk=enrollment_id,
sk='0',