Files
saladeaula.digital/konviva-events/app/app.py

71 lines
2.2 KiB
Python

from decimal import Decimal
from http import HTTPStatus
from typing import Annotated, Any
from aws_lambda_powertools import Logger, Tracer
from aws_lambda_powertools.event_handler.api_gateway import (
APIGatewayHttpResolver,
Response,
)
from aws_lambda_powertools.event_handler.openapi.params import Body
from aws_lambda_powertools.logging import correlation_paths
from aws_lambda_powertools.utilities.typing import LambdaContext
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey
from boto3clients import dynamodb_client
from config import ENROLLMENT_TABLE
from enrollment import EnrollmentNotFoundError, set_score, update_progress
logger = Logger(__name__)
tracer = Tracer()
app = APIGatewayHttpResolver(enable_validation=True)
dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
@app.post('/')
@tracer.capture_method
def postback(
status: Annotated[str, Body(embed=True)],
konviva_id: Annotated[str, Body(alias='ID_MATRICULA', embed=True)],
score: Annotated[Decimal, Body(alias='APROVEITAMENTO', embed=True)],
progress: Annotated[Decimal, Body(alias='ANDAMENTO', embed=True)],
):
score = score.quantize(Decimal('0.00'))
progress = progress.quantize(Decimal('0.00'))
enrollment_id = dyn.collection.get_item(
KeyPair(
# Post-migration: uncomment the following line
# pk='KONVIVA',
pk='konviva',
sk=SortKey(konviva_id, path_spec='enrollment_id'),
),
exc_cls=EnrollmentNotFoundError,
)
# Make sure webhooks send the correct event names
match status:
case 'IN_PROGRESS':
update_progress(
enrollment_id,
progress,
dynamodb_persistence_layer=dyn,
)
case 'COMPLETED':
set_score(
enrollment_id,
score,
progress,
dynamodb_persistence_layer=dyn,
)
return Response(status_code=HTTPStatus.NO_CONTENT)
@logger.inject_lambda_context(correlation_id_path=correlation_paths.API_GATEWAY_HTTP)
@tracer.capture_lambda_handler
def lambda_handler(
event: dict[str, Any],
context: LambdaContext,
) -> dict[str, Any]:
return app.resolve(event, context)