close bolling

enrollments-events/app/events/stopgap/set_subscription_covered.py

@@ -22,6 +22,7 @@ enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
 @logger.inject_lambda_context
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
+    now_ = now()
     data = user_layer.get_item(
         # Post-migration: uncomment the following line
         # KeyPair(new_image['org_id'], 'METADATA#BILLING_TERMS'),
@@ -32,16 +33,27 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         return False
 
     try:
-        enrollment_layer.put_item(
-            item={
-                'id': new_image['id'],
-                'sk': 'METADATA#BILLING_TERMS',
-                'org_id': new_image['tenant_id'],
-                'billing_day': data['billing_day'],
-                'created_at': now(),
-            },
-            cond_expr='attribute_not_exists(sk)',
-        )
+        with enrollment_layer.transact_writer() as transact:
+            transact.update(
+                key=KeyPair(new_image['id'], '0'),
+                update_expr='SET subscription_covered = :subscription_covered, \
+                            updated_at = :updated_at',
+                expr_attr_values={
+                    ':subscription_covered': True,
+                    ':updated_at': now_,
+                },
+                cond_expr='attribute_exists(sk)',
+            )
+            transact.put(
+                item={
+                    'id': new_image['id'],
+                    'sk': 'METADATA#SUBSCRIPTION_COVERED',
+                    'org_id': new_image['tenant_id'],
+                    'billing_day': data['billing_day'],
+                    'created_at': now(),
+                },
+                cond_expr='attribute_not_exists(sk)',
+            )
     except Exception:
         return False
     else: