add report submission

api.saladeaula.digital/app/routes/enrollments/enroll.py

@@ -8,9 +8,10 @@ import pytz
 from aws_lambda_powertools import Logger
 from aws_lambda_powertools.event_handler.api_gateway import Router
 from aws_lambda_powertools.event_handler.exceptions import (
-    NotFoundError,
+    ServiceError,
 )
 from aws_lambda_powertools.event_handler.openapi.params import Body
+from aws_lambda_powertools.shared.functions import extract_event_from_common_models
 from layercake.batch import BatchProcessor
 from layercake.dateutils import now, ttl
 from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
@@ -19,9 +20,8 @@ from layercake.strutils import md5_hash
 from pydantic import UUID4, BaseModel, EmailStr, Field, FutureDate
 from typing_extensions import TypedDict
 
-from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
-from config import DEDUP_WINDOW_OFFSET_DAYS, ENROLLMENT_TABLE, TZ
+from config import DEDUP_WINDOW_OFFSET_DAYS, ENROLLMENT_TABLE, TZ, USER_TABLE
 from exceptions import ConflictError
 from middlewares.authentication_middleware import User as Authenticated
 
@@ -31,7 +31,9 @@ dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
 processor = BatchProcessor()
 
 
-class SubscriptionNotFoundError(NotFoundError): ...
+class SubscriptionRequiredError(ServiceError):
+    def __init__(self, msg: str | dict):
+        super().__init__(HTTPStatus.NOT_ACCEPTABLE, msg)
 
 
 class DeduplicationConflictError(ConflictError): ...
@@ -74,30 +76,32 @@ class Org(BaseModel):
 
 @router.post('/')
 def enroll(
-    org_id: Annotated[UUID4 | str, Body(embed=True)],
+    org_id: Annotated[str | UUID4, Body(embed=True)],
     enrollments: Annotated[tuple[Enrollment, ...], Body(embed=True)],
 ):
+    now_ = now()
     org = dyn.collection.get_items(
         KeyPair(
             pk=str(org_id),
             sk='0',
+            table_name=USER_TABLE,
         )
         + KeyPair(
             pk=str(org_id),
             sk='METADATA#SUBSCRIPTION_TERMS',
             rename_key='terms',
+            table_name=USER_TABLE,
         )
         + KeyPair(
             pk='SUBSCRIPTION',
             sk=f'ORG#{org_id}',
             rename_key='subscribed',
+            table_name=USER_TABLE,
         )
     )
 
     if 'subscribed' not in org:
-        return JSONResponse(
-            status_code=HTTPStatus.NOT_ACCEPTABLE,
-        )
+        raise SubscriptionRequiredError('Organization not subscribed')
 
     ctx = {
         'org': Org.model_validate(org),
@@ -105,20 +109,38 @@ def enroll(
         'terms': SubscriptionTerms.model_validate(org['terms']),
     }
 
-    now = [e for e in enrollments if not e.scheduled_for]
+    immediate = [e for e in enrollments if not e.scheduled_for]
     later = [e for e in enrollments if e.scheduled_for]
 
-    with processor(now, enroll_now, ctx) as batch:
+    with processor(immediate, enroll_now, ctx) as batch:
         now_out = batch.process()
 
     with processor(later, enroll_later, ctx) as batch:
         later_out = batch.process()
 
-    return {
-        'enrolled': now_out,
-        'scheduled': later_out,
+    def fmt(r):
+        return {
+            'status': r.status.value,
+            'input_record': extract_event_from_common_models(r.input_record),
+            'output': extract_event_from_common_models(r.output),
+            'cause': extract_event_from_common_models(r.cause),
+        }
+
+    item = {
+        'id': f'SUBMISSION#ORG#{org_id}',
+        'sk': now_,
+        'enrolled': list(map(fmt, now_out)) if now_out else None,
+        'scheduled': list(map(fmt, later_out)) if later_out else None,
+        'ttl': ttl(start_dt=now_, days=30 * 6),
     }
 
+    try:
+        dyn.put_item(item=item)
+    except Exception as exc:
+        logger.exception(exc)
+    finally:
+        return item
+
 
 Context = TypedDict(
     'Context',
@@ -155,6 +177,9 @@ def enroll_now(enrollment: Enrollment, context: Context):
             item={
                 'id': enrollment.id,
                 'sk': '0',
+                'score': None,
+                'progress': 0,
+                'status': 'PENDING',
                 'user': user.model_dump(),
                 'course': course.model_dump(),
                 'access_expires_at': access_expires_at,
@@ -230,7 +255,8 @@ def enroll_now(enrollment: Enrollment, context: Context):
                     'created_at': now_,
                 },
             )
-    return True
+
+    return enrollment
 
 
 def enroll_later(enrollment: Enrollment, context: Context):