Commited backup

KeyZenD/modules/webshot.py

@@ -0,0 +1,48 @@
+from .. import loader, utils
+import logging
+from requests import get
+import io
+
+
+logger = logging.getLogger(__name__)
+
+def register(cb):
+	cb(WebShotMod())
+
+
+@loader.tds
+class WebShotMod(loader.Module):
+	"""link to screen"""
+	strings = {
+		"name": "WebShot"
+	}
+
+	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 = 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>S c r e e n s h o t i n g . . .</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 = "webshot.png"
+		file.seek(0)
+		await message.client.send_file(message.to_id, file, reply_to=reply)
+		await message.delete()