WIP

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