userbot: add ssvideo module

squash change from:
    1. ProjectAlf: add ssvideo module [alfianandaa/ProjectAlf@d62863e]
    2. ssvideo: add download progress [alfianandaa/ProjectAlf@921f840]
    3. ssvideo: add missing check condition [MrMissx/UserButt@480adb7]
    4. ssvideo: fix typo [MrMissx/UserButt@b43ec1e]

Co-authored-by: Alfiananda P.A <[email protected]>
Co-authored-by: Mr.Miss <[email protected]>
Signed-off-by: MoveAngel <[email protected]>

Author: 2 people

requirements.txt

@@ -55,4 +55,5 @@ urbandict
 wikipedia
 PyGithub
 python-dateutil
+vcsi
 youtube_search

userbot/modules/ssvideo.py

@@ -0,0 +1,67 @@
+# Copyright (C) 2020 Alfiananda P.A
+#
+# Licensed under the Raphielscape Public License, Version 1.d (the "License");
+# you may not use this file except in compliance with the License.
+#
+
+import asyncio
+import os
+import time
+
+from telethon.tl.types import DocumentAttributeFilename
+from userbot import CMD_HELP, bot
+from userbot.events import register
+from userbot.utils import progress
+
+
+@register(outgoing=True, pattern=r"^\.ssvideo(?: |$)(.*)")
+async def ssvideo(framecap):
+    if not framecap.reply_to_msg_id:
+        return await framecap.edit("`reply to video!`")
+    reply_message = await framecap.get_reply_message()
+    if not reply_message.media:
+        return await framecap.edit("`reply to a video!`")
+    try:
+        frame = int(framecap.pattern_match.group(1))
+        if frame > 10:
+            return await framecap.edit("`hey..dont put that much`")
+    except BaseException:
+        return await framecap.edit("`Please input number of frame!`")
+    if (reply_message.photo
+            or (DocumentAttributeFilename(file_name="AnimatedSticker.tgs")
+                in reply_message.media.document.attributes)
+            or (DocumentAttributeFilename(file_name="sticker.webp")
+                in reply_message.media.document.attributes)
+            ):
+        return await framecap.edit("`Unsupported files!`")
+    c_time = time.time()
+    await framecap.edit("`Downloading media...`")
+    ss = await bot.download_media(
+        reply_message,
+        "anu.mp4",
+        progress_callback=lambda d, t: asyncio.get_event_loop().create_task(
+            progress(d, t, framecap, c_time, "[DOWNLOAD]")
+        ),
+    )
+    try:
+        await framecap.edit("`Proccessing...`")
+        command = f"vcsi -g {frame}x{frame} {ss} -o ss.png "
+        os.system(command)
+        await framecap.client.send_file(
+            framecap.chat_id,
+            "ss.png",
+            reply_to=framecap.reply_to_msg_id,
+        )
+        await framecap.delete()
+    except BaseException as e:
+        await framecap.edit(f"{e}")
+    os.system("rm -rf *.png *.mp4")
+
+
+CMD_HELP.update(
+    {
+        "ssvideo": ".ssvideo <grid>\
+        \nUsage: Capture video frames by <grid> x <grid>.\
+        \n*max grid is 10."
+    }
+)