From 67b2049d12888c1f5bf4ce0d64655d469843b8e0 Mon Sep 17 00:00:00 2001 From: Chris Lovering Date: Thu, 26 May 2022 09:57:56 +0100 Subject: Allow latex command in data science and ai channel --- bot/exts/fun/latex.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'bot/exts/fun/latex.py') diff --git a/bot/exts/fun/latex.py b/bot/exts/fun/latex.py index d43ec8c4..f0f2f8c0 100644 --- a/bot/exts/fun/latex.py +++ b/bot/exts/fun/latex.py @@ -10,6 +10,8 @@ from PIL import Image from discord.ext import commands from bot.bot import Bot +from bot.constants import Channels, WHITELISTED_CHANNELS +from bot.utils.decorators import whitelist_override FORMATTED_CODE_REGEX = re.compile( r"(?P(?P```)|``?)" # code delimiter: 1-3 backticks; (?P=block) only matches if it's a block @@ -31,6 +33,10 @@ TEMPLATE = string.Template(Path("bot/resources/fun/latex_template.txt").read_tex PAD = 10 +LATEX_ALLOWED_CHANNNELS = WHITELISTED_CHANNELS + ( + Channels.data_science_and_ai, +) + def _prepare_input(text: str) -> str: """Extract latex from a codeblock, if it is in one.""" @@ -97,6 +103,7 @@ class Latex(commands.Cog): @commands.command() @commands.max_concurrency(1, commands.BucketType.guild, wait=True) + @whitelist_override(channels=LATEX_ALLOWED_CHANNNELS) async def latex(self, ctx: commands.Context, *, query: str) -> None: """Renders the text in latex and sends the image.""" query = _prepare_input(query) -- cgit v1.2.3