27 lines
859 B
Python
27 lines
859 B
Python
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 layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
|
|
|
|
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_item(
|
|
KeyPair(enrollment_id, '0'),
|
|
exc_cls=NotFoundError,
|
|
)
|