api/endpoints/yt.py

#!/usr/bin/env python3.12

import importlib
from fastapi import FastAPI
from pydantic import BaseModel
from typing import Optional
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 = 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,
            #tbd
            }

        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) -> dict:
        """
        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 {
                'err': True,
                'errorText': 'No result.',
            }
        yt_video_id: str|bool = yts_res[0].get('id', False)

        return {
            'video_id': yt_video_id,
            'extras': yts_res[0]
        }