mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-16 14:34:17 +02:00
132 lines
3.8 KiB
Python
132 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Friendly Telegram (telegram userbot)
|
|
# Copyright (C) 2018-2020 The Authors
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
# если не подписан на t.me/keyzend
|
|
# твоя мама шлюха
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
from .. import loader, utils # pylint: disable=relative-beyond-top-level
|
|
import io
|
|
from PIL import Image, ImageOps
|
|
from telethon.tl.types import DocumentAttributeFilename
|
|
import logging
|
|
import random
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def register(cb):
|
|
cb(GriderMod())
|
|
|
|
|
|
@loader.tds
|
|
class GriderMod(loader.Module):
|
|
"""Гавно залупное"""
|
|
strings = {
|
|
"name": "Griding"
|
|
}
|
|
|
|
async def client_ready(self, client, db):
|
|
self.client = client
|
|
|
|
|
|
@loader.sudo
|
|
async def gridcmd(self, message):
|
|
""".gird <reply to photo>"""
|
|
if message.is_reply:
|
|
reply_message = await message.get_reply_message()
|
|
data = await check_media(reply_message)
|
|
if isinstance(data, bool):
|
|
await utils.answer(message, "<code>Реплай на пикчу или стикер блять!</code>")
|
|
return
|
|
else:
|
|
await utils.answer(message, "`Реплай на пикчу или стикер блять`")
|
|
return
|
|
|
|
await message.edit("Режу ебать")
|
|
file = await self.client.download_media(data, bytes)
|
|
media = await griding(file)
|
|
await message.delete()
|
|
await message.client.send_file(message.to_id, media)
|
|
|
|
|
|
|
|
|
|
@loader.sudo
|
|
async def revgridcmd(self, message):
|
|
""".gird <reply to photo>"""
|
|
if message.is_reply:
|
|
reply_message = await message.get_reply_message()
|
|
data = await check_media(reply_message)
|
|
if isinstance(data, bool):
|
|
await utils.answer(message, "<code>Реплай на пикчу или стикер блять!</code>")
|
|
return
|
|
else:
|
|
await utils.answer(message, "`Реплай на пикчу или стикер блять`")
|
|
return
|
|
|
|
await message.edit("Режу ебать")
|
|
file = await self.client.download_media(data, bytes)
|
|
media = await griding(file)
|
|
media = media[::-1]
|
|
await message.delete()
|
|
await message.client.send_file(message.to_id, media)
|
|
|
|
|
|
|
|
|
|
|
|
async def griding(file):
|
|
img = Image.open(io.BytesIO(file))
|
|
(x, y) = img.size
|
|
cy = 3
|
|
cx = 3
|
|
sx = x//cx
|
|
sy = y//cy
|
|
if (sx*cx, sy*cy) != (x, y):
|
|
img = img.resize((sx*cx, sy*cy))
|
|
(lx, ly) = (0, 0)
|
|
media = []
|
|
for i in range(1, cy+1):
|
|
for o in range(1, cx+1):
|
|
mimg = img.crop((lx, ly, lx+sx, ly+sy))
|
|
bio = io.BytesIO()
|
|
bio.name = 'image.png'
|
|
mimg.save(bio, 'PNG')
|
|
media.append(bio.getvalue())
|
|
lx = lx + sx
|
|
lx = 0
|
|
ly = ly + sy
|
|
return media
|
|
|
|
|
|
async def check_media(reply_message):
|
|
if reply_message and reply_message.media:
|
|
if reply_message.photo:
|
|
data = reply_message.photo
|
|
elif reply_message.document:
|
|
if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in reply_message.media.document.attributes:
|
|
return False
|
|
if reply_message.gif or reply_message.video or reply_message.audio or reply_message.voice:
|
|
return False
|
|
data = reply_message.media.document
|
|
else:
|
|
return False
|
|
else:
|
|
return False
|
|
|
|
if not data or data is None:
|
|
return False
|
|
else:
|
|
return data |