tuxbot-bot/cogs/utils/extra.py

22 lines
557 B
Python

from discord.ext import commands
class commandsPlus(commands.Command):
def __init__(self, func, **kwargs):
super().__init__(func, **kwargs)
self.category = kwargs.pop("category")
def commandExtra(*args, **kwargs):
return commands.command(*args, **kwargs, cls=commandsPlus)
class GroupPlus(commands.Group):
def __init__(self, func, **kwargs):
super().__init__(func, **kwargs)
self.category = kwargs.pop("category")
def groupExtra(*args, **kwargs):
return commands.group(*args, **kwargs, cls=GroupPlus)