diff options
| -rw-r--r-- | bot/constants.py | 1 | ||||
| -rw-r--r-- | bot/exts/fun/latex.py | 7 |
2 files changed, 8 insertions, 0 deletions
diff --git a/bot/constants.py b/bot/constants.py index da81a089..bc0b5931 100644 --- a/bot/constants.py +++ b/bot/constants.py @@ -112,6 +112,7 @@ class Channels(NamedTuple): bot_commands = 267659945086812160 community_meta = 267659945086812160 organisation = 551789653284356126 + data_science_and_ai = 366673247892275221 devlog = int(environ.get("CHANNEL_DEVLOG", 622895325144940554)) dev_contrib = 635950537262759947 mod_meta = 775412552795947058 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<delim>(?P<block>```)|``?)" # 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) |