47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
|
#!/usr/bin/env python3.12
|
||
|
|
|
||
|
|
import importlib
|
||
|
|
from fastapi import FastAPI
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
class ValidYTSearchRequest(BaseModel):
|
||
|
|
"""
|
||
|
|
- **t**: title to search
|
||
|
|
"""
|
||
|
|
|
||
|
|
t: str = "rick astley - never gonna give you up"
|
||
|
|
|
||
|
|
|
||
|
|
class YT(FastAPI):
|
||
|
|
"""YT Endpoints"""
|
||
|
|
def __init__(self, app: FastAPI, util, constants): # 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 = {
|
||
|
|
"yt_video_search": self.yt_video_search_handler,
|
||
|
|
#tbd
|
||
|
|
}
|
||
|
|
|
||
|
|
for endpoint, handler in self.endpoints.items():
|
||
|
|
app.add_api_route(f"/{endpoint}/", handler, methods=["POST"])
|
||
|
|
|
||
|
|
async def yt_video_search_handler(self, data: ValidYTSearchRequest):
|
||
|
|
"""
|
||
|
|
/yt_video_search/
|
||
|
|
Search for YT Video by Title (closest match returned)
|
||
|
|
"""
|
||
|
|
|
||
|
|
title = data.t
|
||
|
|
yts_res = await self.ytsearch.search(title)
|
||
|
|
yt_video_id = yts_res[0].get('id', False)
|
||
|
|
|
||
|
|
return {
|
||
|
|
'video_id': yt_video_id,
|
||
|
|
'extras': yts_res[0]
|
||
|
|
}
|
||
|
|
|
||
|
|
|