RGBCube/VReplBot
1
Fork 0
mirror of https://github.com/RGBCube/VReplBot synced 2025-07-27 00:17:46 +00:00
Code Issues Activity Actions

Fix content too large error

Browse source
This commit is contained in:
RGBCube 2023-01-30 20:54:38 +03:00
parent 69a1a4f198
commit fde71c964f
1 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

21
v_repl_bot/cogs/repl.py
View file

@ -1,9 +1,10 @@
from __future__ import annotations from __future__ import annotations
from io import BytesIO
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
from discord.ext import commands from discord import File
from discord.ext.commands import Cog, command from discord.ext.commands import Cog, command, param
from jishaku.codeblocks import Codeblock, codeblock_converter from jishaku.codeblocks import Codeblock, codeblock_converter
if TYPE_CHECKING: if TYPE_CHECKING:
@ -29,7 +30,7 @@ class REPL(
self, self,
ctx: Context, ctx: Context,
*, *,
code: Codeblock | None = commands.param(converter = codeblock_converter, default = None) code: Codeblock | None = param(converter = codeblock_converter, default = None)
) -> None: ) -> None:
if code is None: if code is None:
await ctx.reply("No code provided.") await ctx.reply("No code provided.")
@ -40,9 +41,17 @@ class REPL(
data = { "code": code.content }, data = { "code": code.content },
) as response: ) as response:
text = await response.text() text = await response.text()
await ctx.reply( text = text.replace("`", "\u200B`\u200B") # Zero-width space.
"```\n" + text.replace("`", "\u200B`\u200B") + "\n```"
) # Zero-width space. if len(text) + 6 > 2000:
await ctx.reply(
"The output was too long to be sent as a message. Here is a file instead:",
file = File(BytesIO(text.encode()), filename = "output.txt")
)
else:
await ctx.reply(
"```" + text + "```"
)
async def setup(bot: ReplBot) -> None: async def setup(bot: ReplBot) -> None: