add event

order-events/app/events/stopgap/remove_slots.py

@@ -8,8 +8,6 @@ from layercake.dynamodb import (
     ComposeKey,
     DynamoDBPersistenceLayer,
     KeyPair,
-    SortKey,
-    TransactKey,
 )
 
 from boto3clients import dynamodb_client
@@ -27,21 +25,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     """Remove slots if the tenant has a `metadata#billing_policy` and
     the total is greater than zero."""
     new_image = event.detail['new_image']
-    order_id = new_image['id']
-    data = order_layer.collection.get_items(
-        TransactKey(order_id)
-        + SortKey('0')
-        + KeyPair(
-            pk=order_id,
-            sk=SortKey(
-                sk='metadata#tenant',
-                path_spec='tenant_id',
-                remove_prefix='metadata#',
-            ),
-            rename_key='tenant_id',
-        )
-    )
-    tenant_id = data['tenant_id'].removeprefix('ORG#')
+    data = order_layer.get_item(KeyPair(new_image['id'], '0'))
+    tenant_id = data['tenant']
 
     policy = user_layer.collection.get_item(
         KeyPair(pk=tenant_id, sk='metadata#billing_policy'),
@@ -49,16 +34,22 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         default=False,
     )
 
-    # Skip if missing billing policy or order is zero/negative
+    # Skip if billing policy is missing or order is less than or equal to zero
     if not policy or data['total'] <= 0:
+        logger.info('Missing billing policy.')
         return False
 
+    logger.info(f'Billing policy from Tenant ID "{tenant_id}" found', policy=policy)
+
     result = enrollment_layer.collection.query(
         KeyPair(
             ComposeKey(tenant_id, prefix='vacancies'),
-            order_id,
+            new_image['id'],
         )
     )
+
+    logger.info('Slots found', slots=result['items'])
+
     with enrollment_layer.batch_writer() as batch:
         for pair in result['items']:
             batch.delete_item(
@@ -68,4 +59,6 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                 }
             )
 
+        logger.info('Deleted all slots')
+
     return True