feat: add attribute

Author: onerandomusername

monty/events.py

@@ -34,8 +34,7 @@ def text(self) -> str:
     @functools.cached_property
     def urls(self) -> list[str]:
         """Return the URLs found in the message."""
-        text = self.text
-        return list(extract_urls(text))
+        return list(extract_urls(self.text))
 
     @classmethod
     def from_message_inter(

monty/exts/info/codesnippets.py

@@ -352,9 +352,8 @@ async def _parse_snippets(self, *, content: str | None = None, context: MessageC
         return "\n".join(m[1] for m in sorted(all_snippets))
 
     @commands.Cog.listener(MontyEvent.monty_message_processed.value)
-    async def on_message(self, context: MessageContext) -> None:
+    async def on_message(self, message: disnake.Message, context: MessageContext) -> None:
         """Checks if the message has a snippet link, removes the embed, then sends the snippet contents."""
-        message = context.message
         if not message.guild:
             return
 
@@ -368,7 +367,7 @@ async def on_message(self, context: MessageContext) -> None:
         if not my_perms.send_messages:
             return
 
-        message_to_send = await self._parse_snippets(message.content)
+        message_to_send = await self._parse_snippets(content=context.text)
         destination = message.channel
 
         if 0 < len(message_to_send) <= 2000 and message_to_send.count("\n") <= 27:
@@ -394,7 +393,7 @@ async def send_expanded_links(self, inter: disnake.MessageInteraction) -> None:
 
         custom_id = inter.data.custom_id[len(EXPAND_BUTTON_PREFIX) :]
         link = decode_github_link(custom_id)
-        snippet = await self._parse_snippets(link)
+        snippet = await self._parse_snippets(content=link)
 
         def disable_button() -> disnake.ui.View:
             view = disnake.ui.View.from_message(inter.message)

monty/utils/code.py

@@ -45,6 +45,9 @@ def prepare_input(code: str, *, require_fenced: bool = False) -> str | None:
         if len(blocks) > 1:
             code = "\n".join(block.group("code") for block in blocks)
             info = "several code blocks"
+        elif len(match) > 1:
+            code = "\n".join(block.group("code") for block in match)
+            info = "several code blocks"
         else:
             match = match[0] if len(blocks) == 0 else blocks[0]
             code, block, lang, delim = match.group("code", "block", "lang", "delim")

monty/utils/markdown.py

@@ -17,7 +17,7 @@
 
 
 CODE_BLOCK_RE = re.compile(
-    r"(?P<delim>`{1,2})([^\n]+?)(?P=delim)|```(.+?)```",
+    r"```(.+?)```|(?P<delim>`{1,2})([^\n]+?)(?P=delim)",
     re.DOTALL | re.MULTILINE,
 )