Commited backup

KeyZenD/modules/pypng.py

@@ -0,0 +1,43 @@
+from .. import loader, utils  # pylint: disable=relative-beyond-top-level
+import logging
+import pygments
+from pygments.lexers import Python3Lexer
+from pygments.formatters import ImageFormatter
+import os
+
+
+logger = logging.getLogger(__name__)
+
+def register(cb):
+	cb(py2pngMod())
+
+
+@loader.tds
+class py2pngMod(loader.Module):
+	"""Uploader"""
+	strings = {
+		"name": "pypng"
+	}
+
+	async def client_ready(self, client, db):
+		self.client = client
+	
+	
+	@loader.sudo
+	async def pypngcmd(self, message):
+		"""reply to text code or py file"""
+		await message.edit("<b>Py to PNG</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), 'out.png')
+		await message.client.send_file(message.to_id, 'out.png', force_document=True)
+		os.remove("out.png")
+		await message.delete()