mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-16 06:24:18 +02:00
112 lines
3.7 KiB
Python
112 lines
3.7 KiB
Python
# 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
|