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

19
v_repl_bot/cogs/repl.py
View file

@ -1,9 +1,10 @@
from __future__ import annotations
from io import BytesIO
from typing import TYPE_CHECKING
from discord.ext import commands
from discord.ext.commands import Cog, command
from discord import File
from discord.ext.commands import Cog, command, param
from jishaku.codeblocks import Codeblock, codeblock_converter
if TYPE_CHECKING:
@ -29,7 +30,7 @@ class REPL(
self,
ctx: Context,
*,
code: Codeblock | None = commands.param(converter = codeblock_converter, default = None)
code: Codeblock | None = param(converter = codeblock_converter, default = None)
) -> None:
if code is None:
await ctx.reply("No code provided.")
@ -40,9 +41,17 @@ class REPL(
data = { "code": code.content },
) as response:
text = await response.text()
text = text.replace("`", "\u200B`\u200B") # Zero-width space.
if len(text) + 6 > 2000:
await ctx.reply(
"```\n" + text.replace("`", "\u200B`\u200B") + "\n```"
) # Zero-width space.
"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: