Source code for rootski.main.endpoints.word

from typing import Optional, Union

import rootski.services.database.dynamo.models as dynamo
from fastapi import APIRouter, Request
from loguru import logger
from rootski.schemas.core import Services
from rootski.services.database.dynamo import models as dynamo
from rootski.services.database.dynamo.actions.word import WordNotFoundError, get_word_by_id
from rootski.services.database.dynamo.db_service import DBService as DynamoDBService
from rootski.services.database.dynamo.models2schemas.word import dynamo_to_pydantic__word
from starlette.status import HTTP_404_NOT_FOUND
from urllib3 import HTTPResponse

from rootski import schemas

router = APIRouter()

TWordResponse = Union[
    schemas.AdjectiveResponse, schemas.NounResponse, schemas.VerbResponse, schemas.WordResponse
]


@router.get("/word/{word_id}/{word_type}", response_model=TWordResponse)
async def get_word_data(word_id: int, word_type: str, request: Request):
    """
    Return all data necessary to populate the word page for the given word
    except for the breakdown (see the note)

    NOTE: the "breakdown" field is not returned by this endpoint any more.
    That data should be fetched using GET /breakdown
    """
    app_services: Services = request.app.state.services
    dynamo_service: DynamoDBService = app_services.dynamo

    logger.info(f"Getting word data for word {word_id} of type {word_type}")

    word: Optional[dynamo.Word] = None
    try:
        word: dynamo.Word = get_word_by_id(word_id=word_id, db=dynamo_service)
    except WordNotFoundError:
        raise HTTPResponse(status=HTTP_404_NOT_FOUND)

    response: TWordResponse = dynamo_to_pydantic__word(word=word)

    return response