falso
/
oddluck-limnoria-plugins
mirror of https://github.com/oddluck/limnoria-plugins.git
1
0
Fork 0
Code Issues Releases Wiki Activity

get header and verify image type

This commit is contained in:
Gordon Shumway 2019-03-26 12:26:20 -04:00 committed by GitHub
parent 45341660a1
commit 34f2a6ab95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ASCII/plugin.py
View File

@ -230,7 +230,13 @@ class ASCII(callbacks.Plugin):
filename = "{0}/{1}".format(filepath, url.split('/')[-1])
ua = UserAgent()
header = {'User-Agent':str(ua.random)}
response = requests.get(url, headers=header)
image_formats = ("image/png", "image/jpeg", "image/jpg", "image/gif")
r = requests.head(url, headers=header)
if r.headers["content-type"] in image_formats:
response = requests.get(url, headers=header)
else:
irc.reply("Invalid file type.")
return
if response.status_code == 200:
with open("{0}".format(filename), 'wb') as f:
f.write(response.content)
@ -322,7 +328,13 @@ class ASCII(callbacks.Plugin):
filename = "{0}/{1}".format(filepath, url.split('/')[-1])
ua = UserAgent()
header = {'User-Agent':str(ua.random)}
response = requests.get(url, headers=header)
image_formats = ("image/png", "image/jpeg", "image/jpg", "image/gif")
r = requests.head(url, headers=header)
if r.headers["content-type"] in image_formats:
response = requests.get(url, headers=header)
else:
irc.reply("Invalid file type.")
return
if response.status_code == 200:
with open("{0}".format(filename), 'wb') as f:
f.write(response.content)