mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-17 14:54:18 +02:00
Added and updated repositories 2025-07-11 08:27:20
This commit is contained in:
@@ -1,105 +1,105 @@
|
||||
from .. import loader, utils # pylint: disable=relative-beyond-top-level
|
||||
from PIL import Image, ImageDraw, ImageOps, ImageFilter
|
||||
import io
|
||||
from telethon.tl.types import DocumentAttributeFilename
|
||||
import logging
|
||||
from moviepy.editor import VideoFileClip
|
||||
import os
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
def register(cb):
|
||||
cb(CirclesMod())
|
||||
|
||||
|
||||
@loader.tds
|
||||
class CirclesMod(loader.Module):
|
||||
"""округляет всё"""
|
||||
strings = {
|
||||
"name": "Circles"
|
||||
}
|
||||
def __init__(self):
|
||||
self.name = self.strings['name']
|
||||
|
||||
async def client_ready(self, client, db):
|
||||
self.client = client
|
||||
|
||||
|
||||
|
||||
@loader.sudo
|
||||
async def roundcmd(self, message):
|
||||
""".round <Reply to image/sticker or video/gif>"""
|
||||
reply = None
|
||||
if message.is_reply:
|
||||
reply = await message.get_reply_message()
|
||||
data = await check_media(reply)
|
||||
if isinstance(data, bool):
|
||||
await utils.answer(message, "<b>Reply to image/sticker or video/gif!</b>")
|
||||
return
|
||||
else:
|
||||
await utils.answer(message, "<b>Reply to image/sticker or video/gif!</b>")
|
||||
return
|
||||
data, type = data
|
||||
if type == "img":
|
||||
await message.edit("<b>Processing image</b>📷")
|
||||
img = io.BytesIO()
|
||||
bytes = await message.client.download_file(data, img)
|
||||
im = Image.open(img)
|
||||
w, h = im.size
|
||||
img = Image.new("RGBA", (w,h), (0,0,0,0))
|
||||
img.paste(im, (0, 0))
|
||||
m = min(w, h)
|
||||
img = img.crop(((w-m)//2, (h-m)//2, (w+m)//2, (h+m)//2))
|
||||
w, h = img.size
|
||||
mask = Image.new('L', (w, h), 0)
|
||||
draw = ImageDraw.Draw(mask)
|
||||
draw.ellipse((10, 10, w-10, h-10), fill=255)
|
||||
mask = mask.filter(ImageFilter.GaussianBlur(2))
|
||||
img = ImageOps.fit(img, (w, h))
|
||||
img.putalpha(mask)
|
||||
im = io.BytesIO()
|
||||
im.name = "img.webp"
|
||||
img.save(im)
|
||||
im.seek(0)
|
||||
await message.client.send_file(message.to_id, im, reply_to=reply)
|
||||
else:
|
||||
await message.edit("<b>Processing video</b>🎥")
|
||||
await message.client.download_file(data, "video.mp4")
|
||||
video = VideoFileClip("video.mp4")
|
||||
video.reader.close()
|
||||
w, h = video.size
|
||||
m = min(w, h)
|
||||
box = [(w-m)//2, (h-m)//2, (w+m)//2, (h+m)//2]
|
||||
video = video.crop(*box)
|
||||
await message.edit("<b>Saving video</b>📼")
|
||||
video.write_videofile("result.mp4")
|
||||
await message.client.send_file(message.to_id, "result.mp4", video_note=True, reply_to=reply)
|
||||
os.remove("video.mp4")
|
||||
os.remove("result.mp4")
|
||||
await message.delete()
|
||||
|
||||
|
||||
|
||||
async def check_media(reply):
|
||||
type = "img"
|
||||
if reply and reply.media:
|
||||
if reply.photo:
|
||||
data = reply.photo
|
||||
elif reply.document:
|
||||
if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in reply.media.document.attributes:
|
||||
return False
|
||||
if reply.gif or reply.video:
|
||||
type = "vid"
|
||||
if reply.audio or reply.voice:
|
||||
return False
|
||||
data = reply.media.document
|
||||
else:
|
||||
return False
|
||||
else:
|
||||
return False
|
||||
|
||||
if not data or data is None:
|
||||
return False
|
||||
else:
|
||||
return (data, type)
|
||||
from .. import loader, utils # pylint: disable=relative-beyond-top-level
|
||||
from PIL import Image, ImageDraw, ImageOps, ImageFilter
|
||||
import io
|
||||
from telethon.tl.types import DocumentAttributeFilename
|
||||
import logging
|
||||
from moviepy.editor import VideoFileClip
|
||||
import os
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
def register(cb):
|
||||
cb(CirclesMod())
|
||||
|
||||
|
||||
@loader.tds
|
||||
class CirclesMod(loader.Module):
|
||||
"""округляет всё"""
|
||||
strings = {
|
||||
"name": "Circles"
|
||||
}
|
||||
def __init__(self):
|
||||
self.name = self.strings['name']
|
||||
|
||||
async def client_ready(self, client, db):
|
||||
self.client = client
|
||||
|
||||
|
||||
|
||||
@loader.sudo
|
||||
async def roundcmd(self, message):
|
||||
""".round <Reply to image/sticker or video/gif>"""
|
||||
reply = None
|
||||
if message.is_reply:
|
||||
reply = await message.get_reply_message()
|
||||
data = await check_media(reply)
|
||||
if isinstance(data, bool):
|
||||
await utils.answer(message, "<b>Reply to image/sticker or video/gif!</b>")
|
||||
return
|
||||
else:
|
||||
await utils.answer(message, "<b>Reply to image/sticker or video/gif!</b>")
|
||||
return
|
||||
data, type = data
|
||||
if type == "img":
|
||||
await message.edit("<b>Processing image</b>📷")
|
||||
img = io.BytesIO()
|
||||
bytes = await message.client.download_file(data, img)
|
||||
im = Image.open(img)
|
||||
w, h = im.size
|
||||
img = Image.new("RGBA", (w,h), (0,0,0,0))
|
||||
img.paste(im, (0, 0))
|
||||
m = min(w, h)
|
||||
img = img.crop(((w-m)//2, (h-m)//2, (w+m)//2, (h+m)//2))
|
||||
w, h = img.size
|
||||
mask = Image.new('L', (w, h), 0)
|
||||
draw = ImageDraw.Draw(mask)
|
||||
draw.ellipse((10, 10, w-10, h-10), fill=255)
|
||||
mask = mask.filter(ImageFilter.GaussianBlur(2))
|
||||
img = ImageOps.fit(img, (w, h))
|
||||
img.putalpha(mask)
|
||||
im = io.BytesIO()
|
||||
im.name = "img.webp"
|
||||
img.save(im)
|
||||
im.seek(0)
|
||||
await message.client.send_file(message.to_id, im, reply_to=reply)
|
||||
else:
|
||||
await message.edit("<b>Processing video</b>🎥")
|
||||
await message.client.download_file(data, "video.mp4")
|
||||
video = VideoFileClip("video.mp4")
|
||||
video.reader.close()
|
||||
w, h = video.size
|
||||
m = min(w, h)
|
||||
box = [(w-m)//2, (h-m)//2, (w+m)//2, (h+m)//2]
|
||||
video = video.crop(*box)
|
||||
await message.edit("<b>Saving video</b>📼")
|
||||
video.write_videofile("result.mp4")
|
||||
await message.client.send_file(message.to_id, "result.mp4", video_note=True, reply_to=reply)
|
||||
os.remove("video.mp4")
|
||||
os.remove("result.mp4")
|
||||
await message.delete()
|
||||
|
||||
|
||||
|
||||
async def check_media(reply):
|
||||
type = "img"
|
||||
if reply and reply.media:
|
||||
if reply.photo:
|
||||
data = reply.photo
|
||||
elif reply.document:
|
||||
if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in reply.media.document.attributes:
|
||||
return False
|
||||
if reply.gif or reply.video:
|
||||
type = "vid"
|
||||
if reply.audio or reply.voice:
|
||||
return False
|
||||
data = reply.media.document
|
||||
else:
|
||||
return False
|
||||
else:
|
||||
return False
|
||||
|
||||
if not data or data is None:
|
||||
return False
|
||||
else:
|
||||
return (data, type)
|
||||
|
||||
Reference in New Issue
Block a user