47 lines
1.0 KiB
Python
47 lines
1.0 KiB
Python
import json
|
|
from typing import Literal
|
|
|
|
from aws_lambda_powertools.event_handler.api_gateway import Router
|
|
from elasticsearch import Elasticsearch
|
|
from pydantic import BaseModel
|
|
|
|
import elastic
|
|
from http_models import SearchResponse
|
|
from settings import ELASTIC_CONN, ENROLLMENT_TABLE
|
|
|
|
router = Router()
|
|
elastic_client = Elasticsearch(**ELASTIC_CONN)
|
|
|
|
|
|
@router.get('/')
|
|
def get_enrollments() -> SearchResponse:
|
|
event = router.current_event
|
|
query = event.get_query_string_value('query', '{}')
|
|
page_size = event.get_query_string_value('page_size', '25')
|
|
|
|
return elastic.search( # type: ignore
|
|
index=ENROLLMENT_TABLE,
|
|
page_size=int(page_size),
|
|
query=json.loads(query),
|
|
elastic_client=elastic_client,
|
|
)
|
|
|
|
|
|
@router.get('/<id>')
|
|
def get_enrollment(id: str):
|
|
return {}
|
|
|
|
|
|
class CancelPayload(BaseModel):
|
|
status: Literal['CANCELED'] = 'CANCELED'
|
|
|
|
|
|
@router.patch('/<id>')
|
|
def cancel(id: str, payload: CancelPayload):
|
|
return {}
|
|
|
|
|
|
@router.post('/')
|
|
def enroll():
|
|
return {}
|