fix enroll and reenroll relationship

enrollments-events/app/events/allocate_slots.py

@@ -81,6 +81,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
 class Course:
     id: str
     name: str
+    access_period: int
 
 
 def _get_courses(ids: set) -> tuple[Course, ...]:
@@ -93,6 +94,7 @@ def _get_courses(ids: set) -> tuple[Course, ...]:
         Course(
             id=idx,
             name=obj['name'],
+            access_period=obj['access_period'],
         )
         for idx, obj in result.items()
     )

enrollments-events/app/events/enroll.py

@@ -18,7 +18,11 @@ from layercake.dynamodb import (
 
 from boto3clients import dynamodb_client
 from config import COURSE_TABLE, ENROLLMENT_TABLE, ORDER_TABLE
-from enrollment import LinkedEntity, enroll
+from enrollment import (
+    Kind,
+    LinkedEntity,
+    enroll,
+)
 from schemas import Course, Enrollment, User
 
 logger = Logger(__name__)
@@ -89,7 +93,15 @@ def _handler(record: Course, context: dict) -> Enrollment:
         enrollment,
         persistence_layer=enrollment_layer,
         deduplication_window={'offset_days': 90},
-        linked_entities=frozenset({LinkedEntity(context['order_id'], 'ORDER')}),
+        linked_entities=frozenset(
+            {
+                LinkedEntity(
+                    id=context['order_id'],
+                    kind=Kind.ORDER,
+                    table_name=ORDER_TABLE,
+                ),
+            }
+        ),
     )
 
     return enrollment

enrollments-events/app/events/issue_cert.py

@@ -56,6 +56,9 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     )
 
     try:
+        if 's3_uri' not in cert:
+            raise ValueError('Template URI is missing')
+
         # Send template URI and data to Paperforge API to generate a PDF
         r = requests.post(
             PAPERFORGE_API,
@@ -79,10 +82,11 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                     ),
                 },
             ),
+            timeout=5,
         )
         r.raise_for_status()
 
-        object_key = f'issuedcerts/{enrollment_id}.pdf'
+        object_key = f'certs/{enrollment_id}.pdf'
         s3_uri = f's3://{BUCKET_NAME}/{object_key}'
 
         s3_client.put_object(
@@ -93,10 +97,10 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         )
 
         logger.debug(f'PDF uploaded successfully to {s3_uri}')
-    except KeyError:
+    except ValueError as exc:
         # PDF generation fails if template URI is missing
         s3_uri = None
-        logger.debug('Template URI is missing')
+        logger.exception(exc)
     except requests.exceptions.RequestException as exc:
         logger.exception(exc)
         raise

enrollments-events/app/events/reenroll_if_failed.py

@@ -10,7 +10,7 @@ from layercake.dynamodb import DynamoDBPersistenceLayer, SortKey, TransactKey
 
 from boto3clients import dynamodb_client
 from config import ENROLLMENT_TABLE
-from enrollment import LinkedEntity, enroll
+from enrollment import Kind, LinkedEntity, enroll
 from schemas import Course, Enrollment, User
 
 logger = Logger(__name__)
@@ -52,7 +52,12 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
             'offset_days': metadata['dedup_window_offset_days'],
         },
         linked_entities=frozenset(
-            {LinkedEntity(new_image['id'], 'ENROLLMENT')},
+            {
+                LinkedEntity(
+                    id=new_image['id'],
+                    kind=Kind.ENROLLMENT,
+                ),
+            },
         ),
         persistence_layer=dyn,
     )

enrollments-events/app/events/set_access_expired.py

@@ -29,8 +29,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                     pk=old_image['id'],
                     sk='0',
                 ),
-                update_expr='SET access_expired = :true, \
-                                 updated_at = :now',
+                update_expr='SET access_expired = :true, updated_at = :now',
                 expr_attr_values={
                     ':true': True,
                     ':now': now_,