api.py

"""Module for API calls."""

from fastapi import FastAPI, Query
from akshara import varnakaarya as vk

app = FastAPI()


@app.get("/vinyaasa")
async def api_vinyaasa(
    word: str = Query(..., description="The word to compute the vinyaasa for")
):
    """Return the vinyaasa of a word."""
    try:
        vinyaasa = vk.get_vinyaasa(word)
        status = "success"
    except AssertionError:
        vinyaasa = None
        status = "failure"

    return {"vinyaasa": vinyaasa, "status": status}


@app.get("/akshara")
async def api_akshara(
    word: str = Query(..., description="The word to compute the akshara for")
):
    """Return the akshara of a word."""

    try:
        akshara = vk.get_akshara(word)
        status = "success"
    except AssertionError:
        return {"akshara": None, "status": "failure"}

    return {"akshara": akshara, "status": status}


@app.get("/shabda")
async def api_shabda(
    letters: str = Query(
        ..., description="Comma-separated letters to compute the shabda for"
    )
):
    """Return the shabda of a list of letters."""
    try:
        letter_list = letters.split(",")
        shabda = vk.get_shabda(letter_list)
        status = "success"
    except AssertionError:
        shabda = None
        status = "failure"

    return {"word": shabda, "status": status}