iartamonov/mautrix-telegram
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Try not to crash if WebP is not supported

Browse Source
This commit is contained in:
Tulir Asokan
2018-01-27 22:01:44 +02:00
parent 58d572162d
commit 46d8128931
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mautrix_telegram/portal.py
+8 -6
View File
@@ -304,18 +304,20 @@ class Portal:
@staticmethod @staticmethod
def convert_webp(file, to="png"): def convert_webp(file, to="png"):
image = Image.open(BytesIO(file)).convert("RGBA") try:
file = BytesIO() image = Image.open(BytesIO(file)).convert("RGBA")
image.save(file, to) new_file = BytesIO()
return file.getvalue() image.save(new_file, to)
return f"image/{to}", new_file.getvalue()
except:
return "image/webp", file
def handle_telegram_document(self, source, sender, media): def handle_telegram_document(self, source, sender, media):
file = source.download_file(media.document) file = source.download_file(media.document)
mime_type = magic.from_buffer(file, mime=True) mime_type = magic.from_buffer(file, mime=True)
dont_change_mime = False dont_change_mime = False
if mime_type == "image/webp": if mime_type == "image/webp":
file = self.convert_webp(file, to="png") mime_type, file = self.convert_webp(file, to="png")
mime_type = "image/png"
dont_change_mime = True dont_change_mime = True
uploaded = sender.intent.upload_file(file, mime_type) uploaded = sender.intent.upload_file(file, mime_type)
name = media.caption name = media.caption