modules/Circles.py at master · KeyZenD/modules · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
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)