mirror of
https://github.com/UiO-Gaming/ShiteMusicBot.git
synced 2026-11-05 13:56:51 +01:00
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
from time import perf_counter
|
|
|
|
from bs4 import BeautifulSoup as bs4
|
|
|
|
from bot import MusicBot
|
|
|
|
|
|
class Thumbnailer(object):
|
|
def __init__(self, bot: MusicBot):
|
|
self.bot: MusicBot = bot
|
|
self.logger = self.bot.main_logger.bot_logger.getChild("Thumbnailer")
|
|
|
|
async def get_html(self, url: str):
|
|
async with self.bot.session.get(url, timeout=3) as response:
|
|
assert response.status == 200
|
|
return await response.read()
|
|
|
|
async def _soundcloud(self, url: str) -> str:
|
|
perf_start = perf_counter()
|
|
html = await self.get_html(url)
|
|
try:
|
|
soup = bs4(html, 'html.parser')
|
|
soup = soup.find("meta", property="twitter:image")
|
|
img = soup.get("content", default="")
|
|
perf_stop = perf_counter()
|
|
self.logger.debug("Took %s to find thumbnail from Soundcloud" % (perf_stop - perf_start))
|
|
return img
|
|
except Exception as e:
|
|
self.logger.exception("%s" % e)
|
|
return ""
|
|
|
|
async def _bandcamp(self, url: str) -> str:
|
|
perf_start = perf_counter()
|
|
html = await self.get_html(url)
|
|
try:
|
|
soup = bs4(html, 'html.parser')
|
|
soup = soup.find(class_="popupImage")
|
|
img = soup.get("href", default="")
|
|
perf_stop = perf_counter()
|
|
self.logger.debug("Took %s to find thumbnail from Bandcamp" % (perf_stop - perf_start))
|
|
return img
|
|
except Exception as e:
|
|
self.logger.exception("%s" % e)
|
|
return ""
|
|
|
|
async def _vimeo(self, url: str) -> str:
|
|
perf_start = perf_counter()
|
|
html = await self.get_html(url)
|
|
try:
|
|
soup = bs4(html, 'html.parser')
|
|
soup = soup.find("meta", property="og:image")
|
|
img = soup.get("content", default="")
|
|
perf_stop = perf_counter()
|
|
self.logger.debug("Took %s to find thumbnail from Vimeo" % (perf_stop - perf_start))
|
|
return img
|
|
except Exception as e:
|
|
self.logger.exception("%s" % e)
|
|
return ""
|
|
|
|
async def identify(self, identifier: str, uri: str) -> str:
|
|
if "youtube" in uri:
|
|
thumbnail_url = f"https://img.youtube.com/vi/{identifier}/0.jpg"
|
|
return thumbnail_url
|
|
elif "soundcloud" in uri:
|
|
thumbnail_url = await self._soundcloud(url=uri)
|
|
return thumbnail_url
|
|
elif "bandcamp" in uri:
|
|
thumbnail_url = await self._bandcamp(url=uri)
|
|
return thumbnail_url
|
|
elif "vimeo" in uri:
|
|
thumbnail_url = await self._vimeo(url=uri)
|
|
return thumbnail_url
|
|
else:
|
|
return ""
|