lain/maubot-heic2jpeg
1
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #2 from ajkessel/main

Browse source 
Element (sometimes) uploads heifs as files rather than images, need to check for both.

I'm merging it without testing, but I might test in future if I have a mac available.
This commit is contained in:
ᎠᎡ. Ѕϵrgϵ Ѵictor 2024-02-25 09:34:07 +07:00 • committed by GitHub
commit 64b36a2447
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
hateheif.py
View file

@ -122,7 +122,7 @@ class HateHeifBot(Plugin):
self.rooms = self.config['rooms'] if self.config['rooms'] else None
@command.passive("", msgtypes=(MessageType.IMAGE,))
@command.passive("", msgtypes=(MessageType.IMAGE,MessageType.FILE))
async def hate_heif_message(
self,
evt: MessageEvent,
@ -137,7 +137,7 @@ class HateHeifBot(Plugin):
return
# Double check if it is an image message
if evt.content.msgtype != MessageType.IMAGE:
if evt.content.msgtype != MessageType.IMAGE and evt.content.msgtype != MessageType.FILE:
return
content: MediaMessageEventContent = evt.content
@ -165,6 +165,8 @@ class HateHeifBot(Plugin):
img = img_out.getvalue()
img_tst = Image.open(BytesIO(img))
self.log.debug(f"Created image parameters: {img_tst.format} {img_tst.size} {img_tst.mode}")
content.info.width=img_tst.size[0]
content.info.height=img_tst.size[1]
if is_enc:
img_enc = attachments.encrypt_attachment(img)