Files
ShiteMusicBot/cogs/utils/RoxUtils.py
T
Roxedus f898dd7582 PEP8
A whole FUCK TON of PEP8 improvements
2019-07-03 19:26:48 +02:00

32 lines
962 B
Python

import aiohttp
from bs4 import BeautifulSoup as bs4
session = aiohttp.ClientSession()
class ThumbNailer(object):
async def get_img(self, url):
async with session.get(url, timeout=30) as response:
assert response.status == 200
html = await response.read()
return await ThumbNailer.__parse_result(html)
async def __parse_result(html):
try:
soup = bs4(html, 'html.parser')
img = soup.find("meta", property="twitter:image")["content"]
return img
except Exception as e:
raise e
async def identify(self, identifier, uri):
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 ThumbNailer.get_img(ThumbNailer, url=uri)
return thumbnail_url
else:
return None