Backup_Repos/leech
1
0
Fork 0
mirror of https://github.com/kemayo/leech synced 2026-04-20 19:40:45 +02:00
Code Issues Projects Releases Wiki Activity

Convert other cover image types to PNG

Browse source
This commit is contained in:
Alex Raubach 2018-09-02 20:24:15 -04:00
parent ff568eef10
commit 53a3cde16d
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
ebook/cover.py
View file

@ -3,6 +3,7 @@ from PIL import Image, ImageDraw, ImageFont
from io import BytesIO
import textwrap
import requests
import logging
def make_cover(title, author, width=600, height=800, fontname="Helvetica", fontsize=40, bgcolor=(120, 20, 20), textcolor=(255, 255, 255), wrapat=30):
@ -32,11 +33,21 @@ def make_cover_from_url(url, title, author):
try:
img = requests.Session().get(url)
cover = BytesIO(img.content)
if Image.open(cover).format != "PNG":
cover = _convert_to_png(cover)
except:
#logger.info("Encountered an error downloading cover, reverting to default cover")
cover = make_cover(title, author)
return cover
def _convert_to_png(image_bytestream):
img = Image.open(image_bytestream)
png_image = BytesIO()
img.save(png_image, format="PNG")
png_image.seek(0)
return png_image
def _safe_font(preferred, *args, **kwargs):
for font in (preferred, "Helvetica", "FreeSans", "Arial"):