improve report

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

@@ -51,6 +51,9 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     old_image = event.detail['old_image']
     # Key pattern `CERT#REPORTING#ORG#{org_id}`
     *_, org_id = old_image['id'].split('#')
+    # Key pattern `MONTH#{month}#SCHEDULE#SEND_REPORT_EMAIL`
+    _, month, *_ = old_image['sk'].split('#')
+
     event_name = old_image['sk']
     target_month = old_image['target_month']
     pretty_month = _monthfmt(datetime.strptime(target_month, '%Y-%m').date())
@@ -95,6 +98,28 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     emailmsg.attach(attachment)
 
     try:
+        with enrollment_layer.transact_writer() as transact:
+            transact.update(
+                key=KeyPair(
+                    pk=old_image['id'],
+                    sk=f'MONTH#{month}',
+                ),
+                update_expr='SET #status = :status, updated_at = :updated_at',
+                expr_attr_names={'#status': 'status'},
+                expr_attr_values={
+                    ':status': 'CLOSED',
+                    ':updated_at': now_,
+                },
+                cond_expr='attribute_exists(sk)',
+            )
+            transact.put(
+                item={
+                    'id': old_image['id'],
+                    'sk': f'{event_name}#EXECUTED',
+                    'created_at': now_,
+                }
+            )
+
         sesv2_client.send_email(
             Content={
                 'Raw': {
@@ -102,14 +127,6 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                 },
             }
         )
-        enrollment_layer.put_item(
-            item={
-                'id': old_image['id'],
-                'sk': f'{event_name}#EXECUTED',
-                'created_at': now_,
-            }
-        )
-        logger.info('Email sent')
     except Exception as exc:
         logger.exception(exc)
         enrollment_layer.put_item(
@@ -121,6 +138,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         )
         return False
     else:
+        logger.info('Email sent')
         return True