api/endpoints/yt.py

#!/usr/bin/env python3.12

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

class YT(FastAPI):
    """
    YT Endpoints
    """   
    def __init__(self, app: FastAPI, util,
                 constants) -> None:  # pylint: disable=super-init-not-called
        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)
    
    async def yt_video_search_handler(self, data: ValidYTSearchRequest) -> JSONResponse:
        """
        Search for YT Video by Title (closest match returned)
        - **t**: Title to search
        """
        
        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],
        })