Commited backup

KeyZenD/modules/datamosh.py

@@ -0,0 +1,88 @@
+import asyncio
+import logging
+import subprocess, os
+import random
+from .. import loader, utils
+
+logger = logging.getLogger(__name__)
+
+
+@loader.tds
+class DataMoshMod(loader.Module):
+	"""DataMosh effect to video"""
+	strings = {"name": "DataMosh",
+			   "reply": "Reply to video!",
+			   "error": "ERROR! TRY AGAIN!!",
+			   "processing": "DataDataMoshMosh!"}
+
+	@loader.unrestricted
+	async def datamoshcmd(self, message):
+		""". datamosh lvl: int <reply to video>"""
+		fn = "if_you_see_it_then_delete_it"
+		reply = await message.get_reply_message()
+		if not reply:
+			await message.edit("".join([ random.choice(html).format(ch) for ch in self.strings("reply", message)]))
+			return
+		if not reply.video:
+			await message.edit("".join([ random.choice(html).format(ch) for ch in self.strings("reply", message)]))
+			return
+		else:
+			await reply.download_media(fn+"1.mp4")
+		
+		lvl = 1
+		fp = False
+		args = utils.get_args(message)
+		if args:
+			if len(args) == 1:
+				if args[0].isdigit():
+					lvl = int(args[0])
+					if lvl <= 0:
+						lvl = 1
+				else:
+					fp = True
+			if len(args) > 1:
+				fp = True
+				if args[0].isdigit():
+					lvl = int(args[0])
+					if lvl <= 0:
+						lvl = 1
+				elif args[1].isdigit():
+					fp = True
+					lvl = int(args[1])
+					if lvl <= 0:
+						lvl = 1
+
+		await message.edit("".join([ random.choice(html).format(ch) for ch in self.strings("processing", message)]))
+		subprocess.call(f'ffmpeg -loglevel quiet -y -i {fn}1.mp4 -crf 0 -bf 0 {fn}1.avi', shell=True)
+		try:
+			_f = open(fn+'1.avi', 'rb')
+			f_ = open(fn+'2.avi', 'wb')
+		except FileNotFoundError:
+			await message.edit("".join([ random.choice(html).format(ch) for ch in self.strings("error", message)]))
+			os.system(f"rm -f {fn}*")
+			return
+			
+		frs = _f.read().split(b'00dc')
+		fi = b'\x00\x01\xb0'
+		cf = 0
+		for _, fr in enumerate(frs):
+			if fp == False:
+				f_.write(fr + b'00dc')
+				cf += 1
+				if fr[5:8] == fi:
+					fp = True
+			else:
+				if fr[5:8] != fi:
+					cf += 1
+					for i in range(lvl):
+						f_.write(fr + b'00dc')
+		f_.close()
+		_f.close()
+		
+		subprocess.call(f'ffmpeg -loglevel quiet -y -i {fn}2.avi {fn}2.mp4', shell=True)
+		await message.client.send_file(message.to_id, file=fn+"2.mp4", video_note=bool(reply.video_note))
+		os.system(f"rm -f {fn}*")
+		await message.delete()
+		
+html = ["<b>{}<b>", "<code>{}</code>", "<i>{}</i>", "<del>{}</del>", "<u>{}</u>", '<a href="https://bruh.moment">{}</a>']
+