Backup_Repos/FanFicFare
1
0
Fork 0
mirror of https://github.com/JimmXinu/FanFicFare.git synced 2026-04-25 16:32:15 +02:00
Code Issues Projects Releases Wiki Activity

Skip invalid images, detect img types (#1068)

Browse source 
* Skip invalid images

* Exception handling

---------

Co-authored-by: dbhmw <github.spherical376@passmail.net>
This commit is contained in:
dbhmw 2024-05-11 21:06:33 +00:00 committed by GitHub
parent 8d6676617c
commit eabfd1bef3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
fanficfare/story.py
View file

@ -194,9 +194,20 @@ def no_convert_image(url,data):
# parameter.
ext = url[url.rfind('.')+1:].lower()
if ext not in imagetypes:
logger.info("no_convert_image url:%s - no known extension -- using .jpg"%url)
# doesn't have extension? use jpg.
ext='jpg'
try:
from PIL import Image
from .six import BytesIO
ext = Image.open(BytesIO(data)).format.lower()
logger.info("no_convert_image url:%s - from bits got '%s'" % (url, ext))
except (IOError, TypeError):
raise exceptions.RejectImage("no_convert_image url:%s - not a valid image"%url)
except ImportError:
pass
finally:
if ext not in imagetypes:
logger.info("no_convert_image url:%s - no known extension -- using .jpg"%url)
# doesn't have extension? use jpg.
ext='jpg'
return (data,ext,imagetypes[ext])