32 lines
980 B
Python
32 lines
980 B
Python
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)
|