LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-10 19:46:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix photo resizing not working for images with alpha channels

Browse Source
This commit is contained in:
Lonami Exo 2019-02-14 12:10:34 +01:00
parent d1c755809d
commit 41bfb8ae52
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
telethon/client/uploads.py
View File

@ -30,7 +30,9 @@ class _CacheType:
return self._cls == other
def _resize_photo_if_needed(file, is_image, width=1280, height=1280):
def _resize_photo_if_needed(
file, is_image, width=64, height=64, background=(255, 255, 255)):
# https://github.com/telegramdesktop/tdesktop/blob/12905f0dcb9d513378e7db11989455a1b764ef75/Telegram/SourceFiles/boxes/photo_crop_box.cpp#L254
if (not is_image
or PIL is None
@ -47,8 +49,16 @@ def _resize_photo_if_needed(file, is_image, width=1280, height=1280):
return file
image.thumbnail((width, height), PIL.Image.ANTIALIAS)
# We could save the resized image with the original format, but
# JPEG often compresses better -> smaller size -> faster upload
# We need to mask away the alpha channel ([3]), since otherwise
# IOError is raised when trying to save alpha channels in JPEG.
result = PIL.Image.new('RGB', image.size, background)
result.paste(image, mask=image.split()[3])
buffer = io.BytesIO()
image.save(buffer, 'JPEG')
result.save(buffer, 'JPEG')
buffer.seek(0)
return buffer