issue certs

enrollments-events/app/events/issue_cert.py

@@ -9,7 +9,7 @@ from aws_lambda_powertools.utilities.data_classes import (
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 from layercake.dateutils import fromisoformat, now
-from layercake.dynamodb import DynamoDBPersistenceLayer, SortKey, TransactKey
+from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey, TransactKey
 
 from boto3clients import dynamodb_client, s3_client
 from config import BUCKET_NAME, ENROLLMENT_TABLE, PAPERFORGE_API
@@ -23,6 +23,7 @@ enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     now_ = now()
+    enrollment_id = new_image['id']
     course = enrollment_layer.collection.get_items(
         TransactKey(new_image['id'])
         + SortKey('METADATA#COURSE', path_spec='cert', rename_key='cert')
@@ -55,10 +56,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     r = requests.post(PAPERFORGE_API, data=json_data)
     r.raise_for_status()
 
-    object_key = 'issuedcerts/{completed_at}_{id}.pdf'.format(
-        completed_at=completed_at.strftime('%Y-%m-%d'),
-        id=new_image['id'],
-    )
+    object_key = f'issuedcerts/{enrollment_id}.pdf'
     s3_uri = f's3://{BUCKET_NAME}/{object_key}'
 
     try:
@@ -72,18 +70,22 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         logger.exception(exc)
         raise
 
-    enrollment_layer.put_item(
-        item={
-            'id': new_image['id'],
-            'sk': 'ISSUED_CERT',
-            's3_uri': s3_uri,
-            'created_at': now_,
-        }
-    )
-
     logger.info(f'PDF uploaded successfully to {s3_uri}')
 
-    return True
+    return enrollment_layer.update_item(
+        key=KeyPair(
+            pk=enrollment_id,
+            sk='0',
+        ),
+        update_expr='SET issued_cert = :issued_cert',
+        expr_attr_values={
+            ':issued_cert': {
+                's3_uri': s3_uri,
+                'issued_at': now_,
+            },
+        },
+        cond_expr='attribute_exists(sk)',
+    )
 
 
 def _cpffmt(s: str) -> str: