modules/WSConvertWebp.py

   1 # Mattermost Bot module.
   2 #  Copyright (c) 2016-2022 by Someone <someone@somenet.org> (aka. Jan Vales <jan@jvales.net>)
   3 #  published under MIT-License
   4
   5 import json
   6 import tempfile
   7 from PIL import Image
   8
   9
  10 # pylint: disable=wrong-import-position
  11 from AbstractWSHandler import AbstractWSHandler
  12 class WSConvertWebp(AbstractWSHandler):
  13     NAME = "convertwebp"
  14
  15     def on_WS_EVENT(self, data):
  16         post = json.loads(data["data"]["post"])
  17
  18         if not "metadata" in post or not "files" in post["metadata"]:
  19             return False
  20
  21         with tempfile.TemporaryDirectory() as tmpdirname:
  22             tempfilenames = []
  23             for mmfile in post["metadata"]["files"]:
  24                 if mmfile["mime_type"] == "image/webp":
  25                     res = self.bot.api.get_file(mmfile["id"])
  26                     open(tmpdirname+"/"+mmfile["id"]+".webp", 'wb').write(res.content)
  27                     image = Image.open(tmpdirname+"/"+mmfile["id"]+".webp").convert("RGB")
  28                     image.save(tmpdirname+"/"+mmfile["id"]+".png", "png")
  29                     tempfilenames.append(tmpdirname+"/"+mmfile["id"]+".png")
  30
  31             if tempfilenames:
  32                 root_id = post["id"]
  33                 if post["root_id"] != "":
  34                     root_id = post["root_id"]
  35                 self.bot.api.create_post(post['channel_id'], "Let me convert that for you", props={"from_webhook":"true"}, filepaths=tempfilenames, root_id=root_id)
  36                 return True