add purge to canceled and failed enrollment

This commit is contained in:
2026-01-22 12:17:54 -03:00
parent a01e4329f0
commit 5fac7888a8
11 changed files with 129 additions and 108 deletions

View File

@@ -1,8 +1,5 @@
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 Body
from layercake.dateutils import now
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
@@ -17,10 +14,7 @@ dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
@router.patch('/<enrollment_id>/cancel')
def cancel(
enrollment_id: str,
lock_hash: Annotated[str | None, Body(embed=True)] = None,
):
def cancel(enrollment_id: str):
now_ = now()
canceled_by: Authenticated = router.context['user']
@@ -50,33 +44,7 @@ def cancel(
}
)
transact.delete(
key=KeyPair(
pk=enrollment_id,
sk='CANCEL_POLICY',
),
key=KeyPair(enrollment_id, 'CANCEL_POLICY'),
cond_expr='attribute_exists(sk)',
exc_cls=CancelPolicyConflictError,
)
# Remove reminders and policies that no longer apply
transact.delete(
key=KeyPair(
pk=enrollment_id,
sk='SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS',
)
)
transact.delete(
key=KeyPair(
pk=enrollment_id,
sk='SCHEDULE#REMINDER_ACCESS_PERIOD_BEFORE_30_DAYS',
)
)
transact.delete(
key=KeyPair(
pk=enrollment_id,
sk='METADATA#PARENT_SLOT',
)
)
if lock_hash:
transact.delete(key=KeyPair(enrollment_id, 'LOCK'))
transact.delete(key=KeyPair('LOCK', lock_hash))