indymedia
/
epicyon
mirror of https://gitlab.com/bashrc2/epicyon
7
4
Fork 1
Code Issues 14 Releases Wiki Activity

Get content type

main
Bob Mottram 2021-12-17 09:48:45 +00:00
parent ca11153a36
commit b5975917a6
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
session.py
View File

@ -457,19 +457,28 @@ def downloadImage(session, baseDir: str, url: str,
def getImageBinaryFromUrl(session, url: str, timeoutSec: int, debug: bool):
"""http GET for an image
"""
mimeType = 'image/png'
contentType = None
try:
result = session.get(url, timeout=timeoutSec)
if result.status_code != 200:
print('WARN: getImageFromUrl: ' + url +
' failed with error code ' + str(result.status_code))
mimeType = 'image/png'
if 'image/x-icon' in result.headers['content-length']:
if result.headers.get('content-type'):
contentType = result.headers['content-type']
elif result.headers.get('Content-type'):
contentType = result.headers['Content-type']
elif result.headers.get('Content-Type'):
contentType = result.headers['Content-Type']
if not contentType:
return None, None
if 'image/x-icon' in contentType:
mimeType = 'image/x-icon'
elif 'image/webp' in result.headers['content-length']:
elif 'image/webp' in contentType:
mimeType = 'image/webp'
elif 'image/jpeg' in result.headers['content-length']:
elif 'image/jpeg' in contentType:
mimeType = 'image/jpeg'
elif 'image/gif' in result.headers['content-length']:
elif 'image/gif' in contentType:
mimeType = 'image/gif'
return result.content, mimeType
except requests.exceptions.RequestException as e: