From c91ae2d6aaffbde168ab85e35ff3ad7ce6dedda2 Mon Sep 17 00:00:00 2001 From: sam-heller Date: Wed, 3 Mar 2021 07:14:53 -0500 Subject: Return 404 Floof embed on invalid status code --- bot/exts/evergreen/status_codes.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) (limited to 'bot/exts/evergreen/status_codes.py') diff --git a/bot/exts/evergreen/status_codes.py b/bot/exts/evergreen/status_codes.py index 231d0254..91eaa988 100644 --- a/bot/exts/evergreen/status_codes.py +++ b/bot/exts/evergreen/status_codes.py @@ -31,15 +31,23 @@ class HTTPStatusCodes(Cog): await self.build_embed(url=HTTP_DOG_URL.format(code), ctx=ctx, code=code) async def build_embed(self, url: str, code: int, ctx: Context, ) -> None: - """Attempt to build and dispatch embed. Append error message instead of something goes wrong.""" + """Attempt to build and dispatch embed. Append error message instead if something goes wrong.""" async with self.bot.http_session.get(url, allow_redirects=False) as response: if 200 <= response.status <= 299: - await ctx.send(embed=discord.Embed(title=STATUS_TEMPLATE.format(code), url=url)) + await ctx.send(embed=discord.Embed( + title=STATUS_TEMPLATE.format(code), + url=url + )) + elif 404 == response.status: + await ctx.send(embed=discord.Embed( + title=ERR_404.format(code), + url=url + )) else: await ctx.send(embed=discord.Embed( title=STATUS_TEMPLATE.format(code), - footer=ERR_404.format(code) if response.status == 404 else ERR_UNKNOWN.format(code)) - ) + footer=ERR_UNKNOWN.format(code) + )) def setup(bot: Bot) -> None: -- cgit v1.2.3