52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
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:
|
|
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],
|
|
}
|
|
)
|