Commited backup

KeyZenD/modules/spinner.py

@@ -0,0 +1,123 @@
+# -*- 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.
+
+#   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(SpinnerMod())
+
+
+@loader.tds
+class SpinnerMod(loader.Module):
+	"""Гавно залупное"""
+	strings = {
+		"name": "Spinner"
+	}
+
+	async def client_ready(self, client, db):
+		self.client = client
+	
+	@loader.sudo
+	async def spincmd(self, message):
+		"""you spin me round..."""
+		args = utils.get_args(message)
+		
+		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
+			
+		image = io.BytesIO()
+		await self.client.download_media(data, image)
+		image = Image.open(image)
+		image.thumbnail((512, 512), Image.ANTIALIAS)
+		img = Image.new("RGB", (512, 512), "black")
+		img.paste(image, ((512-image.width)//2, (512-image.height)//2))
+		image = img
+		way = random.choice([1, -1])
+		frames = []
+		for i in range(1, 60):
+			im = image.rotate(i*6*way)
+			frames.append(im)
+		frames.remove(im)
+
+		image_stream = io.BytesIO()
+		image_stream.name = "spin.gif"
+		im.save(image_stream, "GIF", save_all=True, append_images=frames, duration = 10)
+		image_stream.seek(0)
+		await utils.answer(message, image_stream)
+
+	@loader.sudo
+	async def epilepsycmd(self, message):
+		"""ПРИВЕТ ЭПИЛЕТИКИ АХАХАХХА"""
+		args = utils.get_args(message)
+		
+		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
+			
+		image = io.BytesIO()
+		await self.client.download_media(data, image)
+		image = Image.open(image).convert("RGB")
+		invert = ImageOps.invert(image)
+
+		image_stream = io.BytesIO()
+		image_stream.name = "epilepsy.gif"
+		image.save(image_stream, "GIF", save_all=True, append_images=[invert], duration = 1)
+		image_stream.seek(0)
+		await utils.answer(message, image_stream)
+
+	
+
+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