api/endpoints/yt.py

import importlib
from fastapi import FastAPI, Depends
from fastapi.responses import JSONResponse
from fastapi_throttle import RateLimiter
from typing import Optional, Union
from .constructors import ValidYTSearchRequest


class YT(FastAPI):
    """
    YT Endpoints
    """

    def __init__(self, app: FastAPI, util, constants) -> None:
        self.app: FastAPI = app
        self.util = util
        self.constants = constants
        self.ytsearch = importlib.import_module("youtube_search_async").YoutubeSearch()

        self.endpoints: dict = {
            "yt/search": self.yt_video_search_handler,
        }

        for endpoint, handler in self.endpoints.items():
            app.add_api_route(
                f"/{endpoint}",
                handler,
                methods=["POST"],
                include_in_schema=True,
                dependencies=[Depends(RateLimiter(times=2, seconds=2))],
            )

    async def yt_video_search_handler(self, data: ValidYTSearchRequest) -> JSONResponse:
        """Search for YT Video by Title (closest match returned)"""

        title: str = data.t
        yts_res: Optional[list[dict]] = await self.ytsearch.search(title)
        if not yts_res:
            return JSONResponse(
                status_code=404,
                content={
                    "err": True,
                    "errorText": "No result.",
                },
            )
        yt_video_id: Union[str, bool] = yts_res[0].get("id", False)

        return JSONResponse(
            content={
                "video_id": yt_video_id,
                "extras": yts_res[0],
            }
        )