add enrollments to order

api.saladeaula.digital/app/routes/orders/__init__.py

@@ -14,7 +14,7 @@ from pydantic import UUID4
 
 from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
-from config import ORDER_TABLE
+from config import ENROLLMENT_TABLE, ORDER_TABLE
 from exceptions import ConflictError, OrderConflictError, OrderNotFoundError
 from middlewares.authentication_middleware import User as Authenticated
 
@@ -48,14 +48,24 @@ def get_order(order_id: str):
     if not order:
         raise OrderNotFoundError('Order not found')
 
+    org_id = order.get('org_id')
     attempts = dyn.collection.query(KeyPair(order_id, 'TRANSACTION#ATTEMPT#'))
     enrollments = dyn.collection.query(KeyPair(order_id, 'ENROLLMENT#'))
+    seats = (
+        dyn.collection.query(
+            key=KeyPair(f'SEAT#ORG#{org_id}', f'ORDER#{order_id}'),
+            table_name=ENROLLMENT_TABLE,
+        )
+        if org_id
+        else {'items': []}
+    )
 
     return (
         order
         | {
             'payment_attempts': attempts['items'],
             'enrollments': enrollments['items'],
+            'seats': seats['items'],
         }
         # Post-migration (orders): remove the following lines
         | ({'created_at': order['create_date']} if 'create_date' in order else {})