add purge to canceled and failed enrollment
This commit is contained in:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user