91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
from http import HTTPStatus
|
|
from os import rename
|
|
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
|
|
from glom import T, glom
|
|
from layercake.dateutils import now
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey, TransactKey
|
|
|
|
from api_gateway import JSONResponse
|
|
from boto3clients import dynamodb_client
|
|
from config import ENROLLMENT_TABLE
|
|
|
|
from .cancel import router as cancel
|
|
from .dedup_window import router as dedup_window
|
|
from .download_cert import router as download_cert
|
|
from .enroll import router as enroll
|
|
|
|
__all__ = ['cancel', 'dedup_window', 'download_cert', 'enroll']
|
|
|
|
logger = Logger(__name__)
|
|
router = Router()
|
|
dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
|
|
|
|
|
|
@router.get('/<enrollment_id>')
|
|
def get_enrollment(enrollment_id: str):
|
|
return dyn.collection.get_items(
|
|
TransactKey(enrollment_id)
|
|
+ SortKey('0')
|
|
+ SortKey('ORG')
|
|
+ SortKey('LOCK')
|
|
+ SortKey('SCORM_COMMIT#LAST', rename_key='last_commit'),
|
|
)
|
|
|
|
|
|
@router.get('/<enrollment_id>/scorm')
|
|
def get_scorm(enrollment_id: str):
|
|
enrollment = dyn.collection.get_items(
|
|
TransactKey(enrollment_id) + SortKey('0') + SortKey('SCORM_COMMIT#LAST')
|
|
)
|
|
|
|
if not enrollment:
|
|
raise NotFoundError('Enrollment not found')
|
|
|
|
spec = {'id': 'course.id', 'scormset': 'course.scormset'}
|
|
course_id, scormset_id = glom(enrollment, spec).values()
|
|
scormset = dyn.collection.get_item(
|
|
KeyPair(course_id, f'SCORMSET#{scormset_id}'),
|
|
raise_on_error=False,
|
|
default={},
|
|
)
|
|
|
|
return enrollment | {'scormset': scormset}
|
|
|
|
|
|
@router.post('/<enrollment_id>')
|
|
def post_enrollment(
|
|
enrollment_id: str,
|
|
cmi: Annotated[dict, Body(embed=True)],
|
|
):
|
|
now_ = now()
|
|
|
|
with dyn.transact_writer() as transact:
|
|
transact.condition(
|
|
key=KeyPair(enrollment_id, '0'),
|
|
cond_expr='attribute_exists(sk)',
|
|
exc_cls=NotFoundError,
|
|
)
|
|
transact.put(
|
|
item={
|
|
'id': enrollment_id,
|
|
'sk': 'SCORM_COMMIT#LAST',
|
|
'cmi': cmi,
|
|
'created_at': now_,
|
|
}
|
|
)
|
|
transact.put(
|
|
item={
|
|
'id': f'SCORM_COMMIT#{enrollment_id}',
|
|
'sk': now_.isoformat(),
|
|
'cmi': cmi,
|
|
'created_at': now_,
|
|
}
|
|
)
|
|
|
|
return JSONResponse(HTTPStatus.NO_CONTENT)
|