131 lines
3.7 KiB
Python
131 lines
3.7 KiB
Python
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.exceptions import NotFoundError
|
|
from aws_lambda_powertools.event_handler.openapi.params import Body, Query
|
|
from layercake.dateutils import now
|
|
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
|
|
from middlewares.authentication_middleware import User as Authenticated
|
|
|
|
from ...enrollments.enroll import (
|
|
Context,
|
|
Enrollment,
|
|
Org,
|
|
Seat,
|
|
Subscription,
|
|
enroll_now,
|
|
)
|
|
|
|
logger = Logger(__name__)
|
|
router = Router()
|
|
dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
|
|
|
|
|
|
class ScheduledNotFoundError(NotFoundError): ...
|
|
|
|
|
|
@router.get('/<org_id>/enrollments/scheduled')
|
|
def scheduled(
|
|
org_id: str,
|
|
start_key: Annotated[str | None, Query] = None,
|
|
):
|
|
return dyn.collection.query(
|
|
key=PartitionKey(f'SCHEDULED#ORG#{org_id}'),
|
|
start_key=start_key,
|
|
limit=150,
|
|
)
|
|
|
|
|
|
@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,
|
|
),
|
|
)
|
|
|
|
return JSONResponse(status_code=HTTPStatus.NO_CONTENT)
|
|
|
|
|
|
@router.post('/<org_id>/enrollments/scheduled/proceed')
|
|
def proceed(
|
|
org_id: str,
|
|
scheduled_for: Annotated[FutureDatetime, Body(embed=True)],
|
|
lock_hash: Annotated[str, Body(embed=True)],
|
|
):
|
|
now_ = now()
|
|
pk = f'SCHEDULED#ORG#{org_id}'
|
|
sk = f'{scheduled_for.isoformat()}#{lock_hash}'
|
|
|
|
scheduled = dyn.collection.get_item(
|
|
KeyPair(pk, sk),
|
|
exc_cls=ScheduledNotFoundError,
|
|
)
|
|
org = Org(id=org_id, name=scheduled['org_name'])
|
|
created_by: Authenticated = router.context['user']
|
|
seat: Seat | None = scheduled.get('seat')
|
|
billing_day: int | None = scheduled.get('subscription_billing_day')
|
|
ctx: Context = {
|
|
'created_by': created_by,
|
|
'org': org,
|
|
}
|
|
|
|
if billing_day:
|
|
ctx['subscription'] = Subscription(billing_day=billing_day)
|
|
|
|
try:
|
|
enrollment = enroll_now(
|
|
Enrollment(
|
|
user=scheduled['user'],
|
|
course=scheduled['course'],
|
|
seat=seat,
|
|
),
|
|
ctx,
|
|
)
|
|
|
|
with dyn.transact_writer() as transact:
|
|
transact.put(
|
|
item={
|
|
'id': pk,
|
|
'sk': f'{sk}#EXECUTED',
|
|
'enrollment_id': enrollment.id,
|
|
'user': scheduled['user'],
|
|
'course': scheduled['course'],
|
|
'created_by': {
|
|
'id': created_by.id,
|
|
'name': created_by.name,
|
|
},
|
|
'created_at': now_,
|
|
}
|
|
)
|
|
transact.delete(key=KeyPair(pk, sk))
|
|
transact.delete(key=KeyPair('LOCK#SCHEDULED', lock_hash))
|
|
except Exception:
|
|
raise
|
|
else:
|
|
return JSONResponse(
|
|
status_code=HTTPStatus.CREATED,
|
|
body=enrollment,
|
|
)
|