tuxbot-bot/utils/functions/extra.py

from discord.ext import commands
from utils.functions import Config


class CommandsPlus(commands.Command):
    def __init__(self, func, **kwargs):
        super().__init__(func, **kwargs)
        self.category = kwargs.get("category", 'other')


class GroupPlus(commands.Group):
    def __init__(self, func, **kwargs):
        super().__init__(func, **kwargs)
        self.category = kwargs.get("category", 'other')


class ContextPlus(commands.Context):
    async def send(self, content=None, **kwargs):
        config = Config('./configs/config.cfg')

        content = content.replace(config.get("bot", "Token"), 'Whoops! leaked token')
        content = content.replace(config.get("webhook", "Token"), 'Whoops! leaked token')

        return await super().send(content, **kwargs)


def command_extra(*args, **kwargs):
    return commands.command(*args, **kwargs, cls=CommandsPlus)


def group_extra(*args, **kwargs):
    return commands.group(*args, **kwargs, cls=GroupPlus)