Commited backup

GeekTG/FTG-Modules/screenshot.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+
+# Module author: @GovnoCodules
+
+# requires: pygments
+
+import io
+import logging
+import os
+
+import pygments
+from pygments.formatters import ImageFormatter
+from pygments.lexers import Python3Lexer
+from requests import get
+
+from .. import loader, utils
+
+logger = logging.getLogger(__name__)
+
+
+@loader.tds
+class WebShotMod(loader.Module):
+    """Screenshot module"""
+
+    strings = {"name": "Screenshot"}
+
+    async def client_ready(self, client, db):
+        self.client = client
+
+    def __init__(self):
+        self.name = self.strings["name"]
+
+    @loader.sudo
+    async def webshotcmd(self, message):
+        """Reply to link"""
+        reply = None
+        link = utils.get_args_raw(message)
+        if not link:
+            reply = await message.get_reply_message()
+            if not reply:
+                await message.delete()
+                return
+            link = reply.raw_text
+        await message.edit("<b>Screenshotting...</b>")
+        url = "https://webshot.deam.io/{}/?width=1920&height=1080?type=png"
+        file = get(url.format(link))
+        if not file.ok:
+            await message.edit("<b>Something went wrong...</b>")
+            return
+        file = io.BytesIO(file.content)
+        file.name = "webScreenshot.png"
+        file.seek(0)
+        await message.client.send_file(message.to_id, file, reply_to=reply)
+        await message.delete()
+
+    async def fileshotcmd(self, message):
+        """Reply to file"""
+        await message.edit("<b>Screenshotting...</b>")
+        reply = await message.get_reply_message()
+        if not reply:
+            await message.edit("<b>reply to file.py</b>")
+            return
+        media = reply.media
+        if not media:
+            await message.edit("<b>reply to file.py</b>")
+            return
+        file = await message.client.download_file(media)
+        text = file.decode("utf-8")
+        pygments.highlight(
+            text,
+            Python3Lexer(),
+            ImageFormatter(font_name="DejaVu Sans Mono", line_numbers=True),
+            "fileScreenshot.png",
+        )
+        await message.client.send_file(
+            message.to_id, "fileScreenshot.png", force_document=True
+        )
+        os.remove("fileScreenshot.png")
+        await message.delete()