51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import aiohttp
|
|
import re
|
|
from redbot.core import commands
|
|
from redbot.core.utils.menus import menu, DEFAULT_CONTROLS
|
|
|
|
|
|
class YouTube(commands.Cog):
|
|
"""Search YouTube for videos."""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.session = aiohttp.ClientSession()
|
|
|
|
async def _youtube_results(self, query: str):
|
|
try:
|
|
headers = {"user-agent": "Red-cog/3.0"}
|
|
async with self.session.get("https://www.youtube.com/results", params={"search_query": query}, headers=headers) as r:
|
|
result = await r.text()
|
|
yt_find = re.findall(r"{\"videoId\":\"(.{11})", result)
|
|
url_list = []
|
|
for track in yt_find:
|
|
url = f"https://www.youtube.com/watch?v={track}"
|
|
if url not in url_list:
|
|
url_list.append(url)
|
|
|
|
except Exception as e:
|
|
url_list = [f"Something went terribly wrong! [{e}]"]
|
|
|
|
return url_list
|
|
|
|
@commands.command()
|
|
async def youtube(self, ctx, *, query: str):
|
|
"""Search on Youtube."""
|
|
result = await self._youtube_results(query)
|
|
if result:
|
|
await ctx.send(result[0])
|
|
else:
|
|
await ctx.send("Nothing found. Try again later.")
|
|
|
|
@commands.command()
|
|
async def ytsearch(self, ctx, *, query: str):
|
|
"""Search on Youtube, multiple results."""
|
|
result = await self._youtube_results(query)
|
|
if result:
|
|
await menu(ctx, result, DEFAULT_CONTROLS)
|
|
else:
|
|
await ctx.send("Nothing found. Try again later.")
|
|
|
|
def cog_unload(self):
|
|
self.bot.loop.create_task(self.session.close())
|