add event

order-events/app/events/assign_tenant_cnpj.py

@@ -43,6 +43,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     if len(ids) < 2:
         raise ValueError('IDs not found.')
 
+    logger.info('IDs found', ids=ids)
+
     with order_layer.transact_writer() as transact:
         transact.update(
             key=KeyPair(new_image['id'], '0'),
@@ -63,14 +65,4 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
             },
         )
 
-        # Post-migration: remove the following line
-        transact.put(
-            item={
-                'id': new_image['id'],
-                'sk': 'metadata#tenant',
-                'tenant_id': f'ORG#{ids["org_id"]}',
-                'create_date': now_,
-            }
-        )
-
     return True

order-events/app/events/remove_slots_if_canceled.py

@@ -8,7 +8,6 @@ from layercake.dynamodb import (
     ComposeKey,
     DynamoDBPersistenceLayer,
     KeyPair,
-    SortKey,
 )
 
 from boto3clients import dynamodb_client
@@ -29,27 +28,19 @@ class TenantDoesNotExistError(Exception):
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     order_id = new_image['id']
-    # Post-migration: remove the following line
-    tenant_id = order_layer.collection.get_item(
-        KeyPair(
-            order_id,
-            SortKey('metadata#tenant', path_spec='tenant_id'),
-        ),
-        exc_cls=TenantDoesNotExistError,
-    ).removeprefix('ORG#')
-
-    # Post-migration: uncomment the following line
-    # tenant_id = new_image['tenant']
+    tenant_id = new_image['tenant']
 
     result = enrollment_layer.collection.query(
         KeyPair(
             # Post-migration: uncomment the following line
-            # ComposeKey(tenant_id, prefix='slots'),
+            # ComposeKey(tenant_id, prefix='slots#org'),
             ComposeKey(tenant_id, prefix='vacancies'),
             order_id,
         )
     )
 
+    logger.info('Slots found', slots=result['items'])
+
     with enrollment_layer.batch_writer() as batch:
         for pair in result['items']:
             batch.delete_item(

order-events/app/events/set_as_expired.py

@@ -0,0 +1,46 @@
+from aws_lambda_powertools import Logger
+from aws_lambda_powertools.utilities.data_classes import (
+    EventBridgeEvent,
+    event_source,
+)
+from aws_lambda_powertools.utilities.typing import LambdaContext
+from layercake.dateutils import now
+from layercake.dynamodb import (
+    DynamoDBPersistenceLayer,
+    KeyPair,
+)
+
+from boto3clients import dynamodb_client
+from config import ORDER_TABLE
+
+logger = Logger(__name__)
+order_layer = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
+
+
+@event_source(data_class=EventBridgeEvent)
+@logger.inject_lambda_context
+def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
+    new_image = event.detail['new_image']
+    now_ = now()
+
+    try:
+        order_layer.update_item(
+            key=KeyPair(new_image['id'], '0'),
+            update_expr='SET #status = :expired, updated_at = :updated_at',
+            cond_expr='#status IN (:pending) and payment_method = :payment_method',
+            expr_attr_names={
+                '#status': 'status',
+            },
+            expr_attr_values={
+                ':expired': 'EXPIRED',
+                ':pending': 'PENDING',
+                ':payment_method': 'MANUAL',
+                ':updated_at': now_,
+            },
+        )
+    except Exception:
+        logger.info('Failed to update status to EXPIRED', order_id=new_image['id'])
+        return False
+    else:
+        logger.info('Status set to EXPIRED', order_id=new_image['id'])
+        return True

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

@@ -55,12 +55,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     return True
 
 
-class CourseNotFoundError(Exception):
-    def __init__(self, *args):
-        super().__init__('Course not found')
-
-
-def _get_course(course_id: str) -> dict:
+def _get_course(course_id: str) -> dict | None:
     with sqlite3.connect(
         database=SQLITE_DATABASE, detect_types=sqlite3.PARSE_DECLTYPES
     ) as conn:
@@ -70,4 +65,4 @@ def _get_course(course_id: str) -> dict:
         for row in rows:
             return row['json']
 
-    raise CourseNotFoundError
+    return None

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