fix enrollment

http-api/app/app.py

@@ -1,20 +1,36 @@
+import json
 import os
+from datetime import date
+from functools import partial
 from typing import Any
 
 from aws_lambda_powertools import Logger, Tracer
 from aws_lambda_powertools.event_handler.api_gateway import (
     APIGatewayHttpResolver,
     CORSConfig,
-    Response,
-    content_types,
 )
 from aws_lambda_powertools.event_handler.exceptions import ServiceError
 from aws_lambda_powertools.logging import correlation_paths
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
+from api_gateway import JSONResponse
 from middlewares import AuthenticationMiddleware
 from routes import courses, enrollments, lookup, orders, orgs, settings, users, webhooks
 
+
+class JSONEncoder(json.JSONEncoder):
+    def default(self, o):
+        if isinstance(o, date):
+            return o.isoformat()
+
+        return super().default(o)
+
+
+def foo(obj):
+    print(obj)
+    return json.dumps(obj, separators=(',', ':'), cls=JSONEncoder)
+
+
 tracer = Tracer()
 logger = Logger(__name__)
 cors = CORSConfig(
@@ -27,6 +43,7 @@ app = APIGatewayHttpResolver(
     enable_validation=True,
     cors=cors,
     debug='AWS_SAM_LOCAL' in os.environ,
+    serializer=partial(json.dumps, separators=(',', ':'), cls=JSONEncoder),
 )
 app.use(middlewares=[AuthenticationMiddleware()])
 app.include_router(courses.router, prefix='/courses')
@@ -47,12 +64,11 @@ app.include_router(lookup.router, prefix='/lookup')
 
 @app.exception_handler(ServiceError)
 def exc_error(exc: ServiceError):
-    return Response(
+    return JSONResponse(
         body={
-            'msg': exc.msg,
-            'err': type(exc).__name__,
+            'type': type(exc).__name__,
+            'message': str(exc),
         },
-        content_type=content_types.APPLICATION_JSON,
         status_code=exc.status_code,
     )
 

http-api/app/routes/enrollments/enroll.py

@@ -51,18 +51,9 @@ def enroll_(payload: Payload):
     context = {'tenant': router.context['tenant']}
 
     with processor(payload.items, handler, context):
-        processor.process()
+        processed_messages = processor.process()
 
-    print(processor.exceptions)
-
-    return JSONResponse(
-        HTTPStatus.OK,
-        {
-            'successes': processor.successes,
-            'failures': processor.failures,
-            'exceptions': [str(exc) for exc in processor.exceptions],
-        },
-    )
+    return JSONResponse(HTTPStatus.OK, processed_messages)
 
 
 def handler(record: Item, context: dict):

http-api/app/rules/enrollment.py

@@ -63,6 +63,7 @@ def enroll(
     user = enrollment.user
     course = enrollment.course
     tenant_id = tenant['id']
+    lock_hash = md5_hash('%s%s' % (user.id, course.id))
 
     with persistence_layer.transact_writer() as transact:
         transact.put(
@@ -117,19 +118,24 @@ def enroll(
             },
         )
 
+        class DeduplicationConflictError(Exception):
+            def __init__(self, *args):
+                super().__init__('Enrollment already exists')
+
         # Prevents the user from enrolling in the same course again until
         # the deduplication window expires or is removed
+        transact.condition(
+            key=KeyPair('lock', lock_hash),
+            cond_expr='attribute_not_exists(sk)',
+            exc_cls=DeduplicationConflictError,
+        )
+
         if deduplication_window:
-            lock_hash = md5_hash('%s%s' % (user.id, course.id))
             offset_days = deduplication_window['offset_days']
             ttl_expiration = ttl(
                 start_dt=now_ + timedelta(days=course.access_period - offset_days)
             )
 
-            class DeduplicationConflictError(Exception):
-                def __init__(self, *args):
-                    super().__init__('Enrollment already exists')
-
             transact.put(
                 item={
                     'id': 'lock',
@@ -138,8 +144,6 @@ def enroll(
                     'create_date': now_,
                     'ttl': ttl_expiration,
                 },
-                cond_expr='attribute_not_exists(sk)',
-                exc_cls=DeduplicationConflictError,
             )
             transact.put(
                 item={