From 5b54e9a426143ae7ef16a313fca6fde8ac02eec7 Mon Sep 17 00:00:00 2001 From: Gareth Coles Date: Wed, 27 Jun 2018 23:01:56 +0100 Subject: [Snekbox] Improve output --- bot/cogs/snekbox.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/bot/cogs/snekbox.py b/bot/cogs/snekbox.py index 67bdd5a6e..22dbde89e 100644 --- a/bot/cogs/snekbox.py +++ b/bot/cogs/snekbox.py @@ -13,6 +13,13 @@ RMQ_ARGS = { "auto_delete": True } +CODE_TEMPLATE = """ +try: + {CODE} +except Exception as e: + print(e) +""" + class Snekbox: """ @@ -46,6 +53,9 @@ class Snekbox: log.info(f"Received code from {ctx.author.name}#{ctx.author.discriminator} for evaluation:\n{code}") self.jobs[ctx.author.id] = datetime.datetime.now() + code = [f" {line}" for line in code.split("\n")] + code = CODE_TEMPLATE.replace("{CODE}", "\n".join(code)) + try: await self.rmq.send_json( "input", @@ -67,7 +77,7 @@ class Snekbox: output = f"{output[:1900]}... (truncated)" await ctx.send( - f"{ctx.author.mention} Your eval job has completed.\n\n```{output}```" + f"{ctx.author.mention} Your eval job has completed.\n\n```\n{output}\n```" ) del self.jobs[ctx.author.id] -- cgit v1.2.3