Commited backup

KeyZenD/modules/DNA.py

@@ -0,0 +1,111 @@
+# requires: pillow
+# requires: wand
+from .. import loader, utils
+import io
+from telethon.tl.types import DocumentAttributeFilename
+import logging
+from wand.image import Image
+from PIL import Image as IM
+# https://t.me/KeyZenD
+# https://t.me/SomeScripts
+logger = logging.getLogger(__name__)
+
+def register(cb):
+    cb(DistortNoApiMod())
+
+
+@loader.tds
+class DistortNoApiMod(loader.Module):
+    """distorting images"""
+    strings = {
+        "name": "DistortNoApi"
+    }
+
+    async def client_ready(self, client, db):
+        self.client = client
+
+    @loader.sudo
+    async def distortcmd(self, message):
+        """.distort <reply to photo>
+        .distort im
+        .distort 50
+        .distort 50 im
+        .distort im 50
+        im => кидает стикеры как фото
+        50 => (от 0 до дохуя) процент сжатия"""
+        if message.is_reply:
+            reply_message = await message.get_reply_message()
+            data, mime = await check_media(reply_message)
+            if isinstance(data, bool):
+                await utils.answer(message, "<code>Reply to image or stick!</code>")
+                return
+        else:
+            await utils.answer(message, "<code>Reply to image or stick!</code>")
+            return
+        rescale_rate = 70
+        a = utils.get_args(message)
+        force_file = False
+        if a:
+            if 'im' in a:
+                force_file = True
+                a.remove('im')
+            if len(a) > 0:
+                if a[0].isdigit():
+                    rescale_rate = int(a[0])
+                    if rescale_rate <= 0:
+                        rescale_rate = 70
+
+        await message.edit("<code>D i s t o r t i n g . . .</code>")
+        file = await message.client.download_media(data, bytes)
+        file, img = io.BytesIO(file), io.BytesIO()
+        img.name = 'img.png'
+        IM.open(file).save(img, 'PNG')
+        media = await distort(io.BytesIO(img.getvalue()), rescale_rate)
+        out, im = io.BytesIO(), IM.open(media)
+        if force_file:
+            mime = 'png'
+        out.name = f'out.{mime}'
+        im.save(out, mime.upper())
+        out.seek(0)
+        await message.edit("<code>S e n d i n g . . .</code>")
+        await message.client.send_file(message.to_id, out, reply_to=reply_message.id)
+
+        await message.delete()
+
+async def distort(file, rescale_rate):
+    img = Image(file=file)
+    x, y = img.size[0], img.size[1]
+    popx = int(rescale_rate*(x//100))
+    popy = int(rescale_rate*(y//100))
+    img.liquid_rescale(popx, popy, delta_x=1, rigidity=0)
+    img.resize(x, y)
+    out = io.BytesIO()
+    out.name = f'output.png'
+    img.save(file=out)
+    return io.BytesIO(out.getvalue())
+
+async def check_media(reply_message):
+    mime = None
+    if reply_message and reply_message.media:
+        if reply_message.photo:
+            data = reply_message.photo
+            mime = 'image/jpeg'
+        elif reply_message.document:
+            if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in reply_message.media.document.attributes:
+                return False, mime
+            if reply_message.gif or reply_message.video or reply_message.audio or reply_message.voice:
+                return False, mime
+            data = reply_message.media.document
+            mime = reply_message.media.document.mime_type
+            if 'image/' not in mime:
+                return False, mime
+        else:
+            return False, mime
+    else:
+        return False, mime
+
+    if not data or data is None:
+        return False, mime
+    else:
+        mime = mime.split('/')[1]
+        return data, mime