diff --git a/modules/url.py b/modules/url.py index d0127f3..6a7b429 100755 --- a/modules/url.py +++ b/modules/url.py @@ -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): """