Fixed cog not working properly

Author: eunwoo1104

discord_slash/client.py

@@ -65,6 +65,7 @@ def get_cog_commands(self, cog: commands.Cog):
         res = [x for x in func_list if
                isinstance(x, model.CogCommandObject) or isinstance(x, model.CogSubcommandObject)]
         for x in res:
+            x.cog = cog
             if isinstance(x, model.CogCommandObject):
                 self.commands[x.name] = x
             else:

discord_slash/model.py

@@ -173,31 +173,45 @@ async def delete(self, message_id: typing.Union[int, str] = "@original"):
 class CommandObject:
     def __init__(self, name, cmd, *subcommands): # Let's reuse old command formatting.
         self.name = name
-        self.invoke = cmd["func"]
+        self.func = cmd["func"]
         self.description = cmd["description"]
         self.auto_convert = cmd["auto_convert"]
         self.allowed_guild_ids = cmd["guild_ids"]
         self.options = cmd["api_options"]
         self.has_subcommands = cmd["has_subcommands"]
         self.subcommands = subcommands
 
+    def invoke(self, *args):
+        return self.func(*args)
+
 
 class SubcommandObject:
     def __init__(self, sub, base, name, sub_group=None):
         self.base = base
         self.subcommand_group = sub_group
         self.name = name
-        self.invoke = sub["func"]
+        self.func = sub["func"]
         self.description = sub["description"]
         self.auto_convert = sub["auto_convert"]
         self.allowed_guild_ids = sub["guild_ids"]
 
+    def invoke(self, *args):
+        return self.func(*args)
+
 
 class CogCommandObject(CommandObject):
     def __init__(self, *args):
         super().__init__(*args)
+        self.cog = None # Manually set this later.
+
+    def invoke(self, *args):
+        return self.func(self.cog, *args)
 
 
 class CogSubcommandObject(SubcommandObject):
     def __init__(self, *args):
         super().__init__(*args)
+        self.cog = None # Manually set this later.
+
+    def invoke(self, *args):
+        return self.func(self.cog, *args)