Commited backup

hikariatama/ftg/latex.py

@@ -0,0 +1,85 @@
+#             █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
+#             █▀█ █ █ █ █▀█ █▀▄ █
+#              © Copyright 2022
+#           https://t.me/hikariatama
+#
+# 🔒      Licensed under the GNU AGPLv3
+# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
+
+# scope: hikka_min 1.2.10
+
+# meta pic: https://img.icons8.com/fluency/452/texshop.png
+# meta banner: https://mods.hikariatama.ru/badges/latex.jpg
+# meta developer: @hikarimods
+
+import io
+import logging
+
+import matplotlib.pyplot as plt
+from telethon.tl.types import Message
+
+from .. import loader, utils
+
+logger = logging.getLogger(__name__)
+
+
+@loader.tds
+class LaTeXMod(loader.Module):
+    """Renders mathematical formulas in LaTeX pngs"""
+
+    strings = {
+        "name": "LaTeX",
+        "no_args": "🚫 <b>Specify a formula to render</b>",
+        "cant_render": "🚫 <b>Can't render formula</b>",
+    }
+
+    strings_ru = {
+        "no_args": "🚫 <b>Укажи формулу для рендера</b>",
+        "cant_render": "🚫 <b>В формуле обнаружена ошибка</b>",
+    }
+
+    async def latexcmd(self, message: Message):
+        """<formula> - Create LaTeX render"""
+        args = utils.get_args_raw(message)
+        if not args:
+            await utils.answer(message, self.strings("no_args"))
+            return
+
+        try:
+            tex = f"${args}$"
+
+            fig = plt.figure()
+            ax = fig.add_axes([0, 0, 1, 1])
+            ax.set_axis_off()
+
+            t = ax.text(
+                0.5,
+                0.5,
+                tex,
+                horizontalalignment="center",
+                verticalalignment="center",
+                fontsize=25,
+                color="black",
+            )
+
+            ax.figure.canvas.draw()
+
+            bbox = t.get_window_extent()
+            fig.set_size_inches(bbox.width / 80, bbox.height / 80)
+            buf = io.BytesIO()
+            plt.savefig(buf)
+            buf.seek(0)
+        except Exception:
+            logger.exception("Can't render formula")
+            await utils.answer(message, self.strings("cant_render"))
+            return
+
+        await self._client.send_file(
+            message.peer_id,
+            buf.getvalue(),
+            reply_to=message.reply_to_msg_id,
+            caption=f"🧮 <b>LaTeX</b>: <code>{args}</code>",
+        )
+
+        if message.out:
+            await message.delete()