judev1 / fbot Goto Github PK

class Errorhandler(commands.Cog):

    def __init__(self, bot):
        self.bot = bot
        self.errorlogs = bot.get_channel(743392645228920882)
        self.message_cooldown = commands.CooldownMapping.from_cooldown(
            1.0, 3.0, commands.BucketType.user)

    async def create_embed(self, ctx, error):
        embed = discord.Embed(title=f"Error Caught!",
                              color=discord.Colour.red(), description=f"{error}")
        embed.set_thumbnail(
            url=self.bot.user.avatar_url_as(static_format="png"))
        bucket = self.message_cooldown.get_bucket(ctx.message)
        retry_after = bucket.update_rate_limit()
        if retry_after:
            return
        else:
            try:
                await ctx.send(embed=embed, delete_after=5)
                await ctx.message.add_reaction('\u274C')
            except:
                await ctx.send(f"`{error}`\n***Enable Embed permissions please.***", delete_after=5)
                await ctx.message.add_reaction('\u274C')
                return

   @commands.Cog.listener()
    async def on_command_error(self, ctx, error):
        error = getattr(error, "original", error)
        if hasattr(ctx.command, "on_error"):
            return
        else:
            pass

        if isinstance(error, errors.MissingRequiredArgument) or isinstance(error, errors.BadArgument):
            helper = str(ctx.invoked_subcommand) if ctx.invoked_subcommand else str(
                ctx.command)
            await ctx.send_help(helper)
            await self.create_embed(ctx, error)

        elif isinstance(error, errors.CommandInvokeError):
            error = default.traceback_maker(error.original)
            if "2000 or fewer" in str(error) and len(ctx.message.clean_content) > 1900:
                return await ctx.send(
                    "You attempted to make the command display more than 2,000 characters...\n"
                    "Both error and command will be ignored."
                )
            await ctx.send(f"There was an error processing the command\n{error}")
            

        elif isinstance(error, errors.BotMissingPermissions):
            await self.create_embed(ctx, error)