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)
