add cancel schedule

api.saladeaula.digital/app/routes/orgs/enrollments/scheduled.py

@@ -1,10 +1,14 @@
+from http import HTTPStatus
 from typing import Annotated
 
 from aws_lambda_powertools import Logger
 from aws_lambda_powertools.event_handler.api_gateway import Router
-from aws_lambda_powertools.event_handler.openapi.params import Query
-from layercake.dynamodb import DynamoDBPersistenceLayer, PartitionKey
+from aws_lambda_powertools.event_handler.exceptions import NotFoundError
+from aws_lambda_powertools.event_handler.openapi.params import Body, Query
+from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, PartitionKey
+from pydantic import FutureDatetime
 
+from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
 from config import ENROLLMENT_TABLE
 
@@ -23,3 +27,33 @@ def scheduled(
         start_key=start_key,
         limit=150,
     )
+
+
+class ScheduledNotFoundError(NotFoundError): ...
+
+
+@router.delete('/<org_id>/enrollments/scheduled')
+def cancel(
+    org_id: str,
+    scheduled_for: Annotated[FutureDatetime, Body(embed=True)],
+    lock_hash: Annotated[str, Body(embed=True)],
+):
+    with dyn.transact_writer() as transact:
+        transact.delete(
+            key=KeyPair(
+                pk=f'SCHEDULED#ORG#{org_id}',
+                sk=f'{scheduled_for.isoformat()}#{lock_hash}',
+            ),
+            cond_expr='attribute_exists(sk)',
+            exc_cls=ScheduledNotFoundError,
+        )
+        transact.delete(
+            key=KeyPair(
+                pk='LOCK#SCHEDULED',
+                sk=lock_hash,
+            ),
+            cond_expr='attribute_exists(sk)',
+            exc_cls=ScheduledNotFoundError,
+        )
+
+    return JSONResponse(status_code=HTTPStatus.NO_CONTENT)