added puu.sh filename support

This commit is contained in:
iou1name 2018-09-19 13:01:19 -04:00
parent c22ee5ee37
commit 8d171ffdbe

View File

@ -8,10 +8,28 @@ from html.parser import HTMLParser
import requests
from module import hook
from module import hook, url_callback
HEADERS = {"User-Agent": "bix nood gimme the title", "Range": "bytes=0-4096"}
@url_callback('puu.sh/')
def get_puush_fname(bot, url):
"""
Callback puu.sh links. Gets the filename and displays it.
"""
try:
res = requests.head(url)
except requests.exceptions.RequestException:
return
fname = res.headers.get('Content-Disposition', '')
fname = re.search(r'\"(.*)\"', fname).group(1)
if not fname:
return
bot.msg(f"[ \x0310puu.sh \x03] - \x0304{fname}")
@hook(True)
def title_auto(bot, trigger):
"""