enroll

konviva-events/app/events/enroll.py

@@ -4,13 +4,37 @@ from aws_lambda_powertools.utilities.data_classes import (
     event_source,
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
+from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
+
+import konviva
+from boto3clients import dynamodb_client
+from config import ENROLLMENT_TABLE
 
 logger = Logger(__name__)
+enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
 
 
 @event_source(data_class=EventBridgeEvent)
 @logger.inject_lambda_context
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
+    user_id = new_image['user_id']
+    class_id = new_image['class_id']
 
-    return True
+    try:
+        enrollment_id = konviva.enroll(
+            user_id,
+            class_id,
+        )
+    except konviva.KonvivaError as err:
+        logger.exception(err)
+        return False
+
+    return enrollment_layer.update_item(
+        key=KeyPair(new_image['id'], 'konviva'),
+        update_expr='SET enrollment_id = :enrollment_id',
+        cond_expr='attribute_exists(sk)',
+        expr_attr_values={
+            ':enrollment_id': enrollment_id,
+        },
+    )