42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from discord.ext import commands
|
|
import aiohttp
|
|
import re
|
|
|
|
|
|
class YouTube:
|
|
"""Le YouTube Cog"""
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.session = aiohttp.ClientSession()
|
|
|
|
async def _youtube_results(self, query: str):
|
|
try:
|
|
headers = {"user-agent": "Red-cog/2.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 = "https://www.youtube.com/watch?v={}".format(track)
|
|
if url not in url_list:
|
|
url_list.append(url)
|
|
|
|
except Exception as e:
|
|
url_list = ["Something went terribly wrong! [{}]".format(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 self.bot.say(result[0])
|
|
else:
|
|
await self.bot.say("Nothing found. Try again later.")
|
|
|
|
|
|
def setup(bot):
|
|
n = YouTube(bot)
|
|
bot.add_cog(n)
|