81 lines
3 KiB
Python
81 lines
3 KiB
Python
"""Custom JARVIS client."""
|
|
import json
|
|
import traceback
|
|
from datetime import datetime
|
|
|
|
from aiohttp import ClientSession
|
|
from dis_snek import Context, Snake, listen
|
|
|
|
DEFAULT_GUILD = 862402786116763668
|
|
DEFAULT_ERROR_CHANNEL = 943395824560394250
|
|
DEFAULT_URL = "https://paste.zevs.me/documents"
|
|
|
|
ERROR_MSG = """
|
|
Command Information:
|
|
Name: {invoked_name}
|
|
Args:
|
|
{arg_str}
|
|
|
|
Callback:
|
|
Args:
|
|
{callback_args}
|
|
Kwargs:
|
|
{callback_kwargs}
|
|
"""
|
|
|
|
|
|
class Jarvis(Snake):
|
|
@listen()
|
|
async def on_ready(self) -> None:
|
|
"""Lepton on_ready override."""
|
|
print("Logged in as {}".format(self.user)) # noqa: T001
|
|
print("Connected to {} guild(s)".format(len(self.guilds))) # noqa: T001
|
|
print( # noqa: T001
|
|
"https://discord.com/api/oauth2/authorize?client_id="
|
|
"{}&permissions=8&scope=bot%20applications.commands".format(self.user.id)
|
|
)
|
|
|
|
async def on_command_error(
|
|
self, ctx: Context, error: Exception, *args: list, **kwargs: dict
|
|
) -> None:
|
|
"""Lepton on_command_error override."""
|
|
guild = await self.fetch_guild(DEFAULT_GUILD)
|
|
channel = await guild.fetch_channel(DEFAULT_ERROR_CHANNEL)
|
|
error_time = datetime.utcnow().strftime("%d-%m-%Y %H:%M-%S.%f UTC")
|
|
timestamp = int(datetime.now().timestamp())
|
|
timestamp = f"<t:{timestamp}:T>"
|
|
arg_str = (
|
|
"\n".join(f" {k}: {v}" for k, v in ctx.kwargs.items()) if ctx.kwargs else " None"
|
|
)
|
|
callback_args = "\n".join(f" - {i}" for i in args) if args else " None"
|
|
callback_kwargs = (
|
|
"\n".join(f" {k}: {v}" for k, v in kwargs.items()) if kwargs else " None"
|
|
)
|
|
full_message = ERROR_MSG.format(
|
|
error_time=error_time,
|
|
invoked_name=ctx.invoked_name,
|
|
arg_str=arg_str,
|
|
callback_args=callback_args,
|
|
callback_kwargs=callback_kwargs,
|
|
)
|
|
if len(full_message) >= 1900:
|
|
error_message = " ".join(traceback.format_exception(error))
|
|
full_message += "Exception: |\n " + error_message
|
|
async with ClientSession() as session:
|
|
resp = await session.post(DEFAULT_URL, data=full_message)
|
|
data = await resp.read()
|
|
data = json.loads(data.decode("UTF8"))
|
|
|
|
await channel.send(
|
|
f"JARVIS encountered an error at {timestamp}. Log too big to send over Discord."
|
|
f"\nPlease see log at https://paste.zevs.me/{data['key']}"
|
|
)
|
|
else:
|
|
error_message = "".join(traceback.format_exception(error))
|
|
await channel.send(
|
|
f"JARVIS encountered an error at {timestamp}:"
|
|
f"\n```yaml\n{full_message}\n```"
|
|
f"\nException:\n```py\n{error_message}\n```"
|
|
)
|
|
await ctx.send("Whoops! Encountered an error. The error has been logged.", ephemeral=True)
|
|
return await super().on_command_error(ctx, error, *args, **kwargs)
|