WIP

orders-events/app/events/billing/append_enrollment.py

@@ -90,6 +90,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                 item={
                     'id': pk,
                     'sk': f'{sk}#SCHEDULE#AUTO_CLOSE',
+                    # Post-migration: uncomment the following line
+                    # 'sk': f'{sk}#SCHEDULED#AUTO_CLOSE',
                     'ttl': ttl(
                         start_dt=datetime.combine(end_period, time())
                         + timedelta(days=1)

orders-events/app/events/billing/cancel_enrollment.py

@@ -80,10 +80,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
             cond_expr='attribute_not_exists(sk)',
         )
     except Exception as exc:
-        logger.exception(
-            exc,
-            keypair={'id': pk, 'sk': sk},
-        )
+        logger.exception(exc, keypair={'id': pk, 'sk': sk})
         return False
     else:
         return True

orders-events/app/events/payments/create_invoice.py

@@ -107,8 +107,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                 transact.put(
                     item={
                         'id': order_id,
-                        'sk': 'SCHEDULE#SELF_DESTRUCTION',
-                        'ttl': ttl(start_dt=now_, days=14),
+                        'sk': 'SCHEDULED#SELF_DESTRUCTION',
+                        'ttl': ttl(start_dt=now_, days=7),
                         'created_at': now_,
                     }
                 )

orders-events/app/events/run_self_destruction.py

@@ -23,7 +23,11 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     order_id = keys['id']
 
     r = dyn.collection.query(PartitionKey(order_id), limit=150)
-    logger.info('Records found', total_items=len(r['items']), records=r['items'])
+    logger.info(
+        msg='Records found',
+        total_items=len(r['items']),
+        records=r['items'],
+    )
 
     with dyn.batch_writer() as batch:
         for pair in r['items']:

orders-events/app/events/start_fulfillment.py

@@ -0,0 +1,78 @@
+import pprint
+from dataclasses import asdict, dataclass
+from uuid import uuid4
+
+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.dynamodb import DynamoDBPersistenceLayer, KeyChain, KeyPair
+
+from boto3clients import dynamodb_client
+from config import ENROLLMENT_TABLE
+
+logger = Logger(__name__)
+dyn = DynamoDBPersistenceLayer(ENROLLMENT_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']
+    order_id = new_image['id']
+    enrollments = dyn.collection.query(
+        key=KeyPair(order_id, 'ENROLLMENT#'),
+    ).get('items', [])
+
+    if not enrollments:
+        items = dyn.collection.get_item(
+            KeyPair(order_id, 'ITEMS'),
+            raise_on_error=False,
+            default=[],
+        )
+        pprint.pp(items)
+    # docx = {
+    #     'id': f'SEAT#ORG#{org_id}',
+    #     'sk': f'ORDER#{order_id}#ENROLLMENT#{uuid4()}',
+    #     'course': {},
+    #     'created_at': now_,
+    # }
+
+    pprint.pp(enrollments)
+
+    return True
+
+
+# Se houver matriculas
+#  -> com: scheduled_for
+#     -> tenta agendar, se não joga para vagas
+#  -> tenta matriculas, se falhar, joga para vagas
+
+# se não houver vagas, gera as vagas.
+
+
+@dataclass(frozen=True)
+class Course:
+    id: str
+    name: str
+    access_period: int
+
+
+def _get_courses(ids: set) -> tuple[Course, ...]:
+    pairs = tuple(KeyPair(idx, '0') for idx in ids)
+    r = dyn.collection.get_items(
+        KeyChain(pairs),
+        flatten_top=False,
+    )
+    courses = tuple(
+        Course(
+            id=idx,
+            name=obj['name'],
+            access_period=obj['access_period'],
+        )
+        for idx, obj in r.items()
+    )
+
+    return courses