add cert_expires_at

enrollments-events/app/events/reporting/append_cert.py

@@ -1,5 +1,5 @@
 import os
-from datetime import datetime, timedelta
+from datetime import timedelta
 
 import pytz
 from aws_lambda_powertools import Logger
@@ -8,7 +8,6 @@ from aws_lambda_powertools.utilities.data_classes import (
     event_source,
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
-from glom import glom
 from layercake.dateutils import fromisoformat, now, ttl
 from layercake.dynamodb import DynamoDBPersistenceLayer
 from layercake.funcs import pick
@@ -25,14 +24,11 @@ tz = os.getenv('TZ', 'UTC')
 @logger.inject_lambda_context
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool | None:
     new_image = event.detail['new_image']
-    expires_at = glom(new_image, 'cert.expires_at', default=None)
-
-    if not expires_at:
-        return None
-
     enrollment_id = new_image['id']
     org_id = new_image['org_id']
-    expires_at: datetime = fromisoformat(expires_at).replace(tzinfo=pytz.timezone(tz))  # type: ignore
+    expires_at = fromisoformat(new_image['cert_expires_at']).replace(  # type: ignore
+        tzinfo=pytz.timezone(tz)
+    )
     # The reporting month is the month before the certificate expires
     month_start = (expires_at.replace(day=1) - timedelta(days=1)).replace(day=1)
     now_ = now()
@@ -59,11 +55,10 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool | No
                 item={
                     'id': pk,
                     'sk': f'{sk}#ENROLLMENT#{enrollment_id}',
-                    'enrollment_id': new_image['id'],
                     'user': pick(('id', 'name'), new_image['user']),
                     'course': pick(('id', 'name'), new_image['course']),
                     'enrolled_at': new_image['created_at'],
-                    'expires_at': expires_at,  # type: ignore
+                    'expires_at': expires_at,
                     'completed_at': new_image['completed_at'],
                     'created_at': now_,
                 },