mmmmmh

Author: i0bs

bot.py

@@ -0,0 +1,45 @@
+import logging
+from typing import Union
+from discord_slash import SlashCommand, SlashContext, ComponentContext, MenuContext
+from discord_slash.model import ContextMenuType, ButtonStyle
+from discord_slash.utils.manage_components import create_actionrow, create_button
+from discord import Intents
+from discord.ext.commands import Bot
+
+bot = Bot(
+    command_prefix="/",
+    help_command=None,
+    intents=Intents.default()
+)
+slash = SlashCommand(
+    bot,
+    sync_commands=False
+)
+log = logging.Logger(name="errors.log", level=logging.DEBUG)
+
+bot.load_extension("cog")
+
+@bot.event
+async def on_ready():
+    print("we're live!")
+
+@bot.event
+async def on_slash_command_error(ctx, ex):
+    log.debug(ctx)
+    log.debug(ex)
+
+@slash.slash(name="testcmd", guild_ids=[852402668294766612])
+async def testcmd(ctx: SlashContext):
+    await ctx.send("test2!")
+
+@slash.context_menu(ContextMenuType.USER, name="Testing User Space", guild_ids=[852402668294766612])
+async def testuser(ctx: MenuContext):
+    await ctx.send("test!")
+
+@slash.context_menu(ContextMenuType.MESSAGE, name="Testing Name Space", guild_ids=[852402668294766612])
+async def testmsg(ctx: Union[ComponentContext, MenuContext]):
+    print(ctx.target)
+    await ctx.defer()
+    await ctx.send(ctx.target["message"]["content"])
+
+bot.run("Mzc5MzQzMzIyNTQ1NzgyNzg0.WgiY_w.8Mxf0hEjruTUs5SQDpKhOxUPquk", bot=True, reconnect=True)

cog.py

@@ -0,0 +1,16 @@
+from discord_slash.context import MenuContext
+from discord.ext.commands import Cog
+from discord.ext.commands.context import Context
+from discord_slash import cog_ext
+from discord_slash.model import ContextMenuType
+
+class Cog(Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @cog_ext.cog_context_menu(name="Testing My Sep", guild_ids=[852402668294766612], target=ContextMenuType.MESSAGE)
+    async def cog_testing(self, ctx: MenuContext):
+        await ctx.send("test!")
+
+def setup(bot):
+    bot.add_cog(Cog(bot))

discord_slash/client.py

@@ -281,7 +281,6 @@ async def to_dict(self):
         cmds = {"global": [], "guild": {x: [] for x in all_guild_ids}}
         wait = {}  # Before merging to return dict, let's first put commands to temporary dict.
         for x in self.commands:
-
             if x == "context":
                 # handle context menu separately.
                 for _x in self.commands["context"]:  # x is the new reference dict

discord_slash/context.py

@@ -265,28 +265,13 @@ async def send(
             for file in files:
                 file.close()
         if not hidden:
-            try:
-                self.menu_messages = (
-                    self.data["resolved"]["messages"] if "resolved" in self.data.keys() else None
-                )
-            except:  # noqa
-                self.menu_messages = None
-            if self.menu_messages:
-                smsg = model.SlashMessage(
-                    state=self.bot._connection,
-                    data=resp,
-                    channel=self.channel or discord.Object(id=self.channel_id),
-                    _http=self._http,
-                    interaction_token=self._token,
-                )
-            else:
-                smsg = model.SlashMessage(
-                    state=self.bot._connection,
-                    data=resp,
-                    channel=self.channel or discord.Object(id=self.channel_id),
-                    _http=self._http,
-                    interaction_token=self._token,
-                )
+            smsg = model.SlashMessage(
+                state=self.bot._connection,
+                data=resp,
+                channel=self.channel or discord.Object(id=self.channel_id),
+                _http=self._http,
+                interaction_token=self._token,
+            )
             if delete_after:
                 self.bot.loop.create_task(smsg.delete(delay=delete_after))
             if initial_message:
@@ -670,43 +655,38 @@ def __init__(
         logger,
     ):
         super().__init__(_http=_http, _json=_json, _discord=_discord, logger=logger)
-        self.target_id = self.data["target_id"]
         self.context_type = _json["type"]
         self._resolved = self.data["resolved"] if "resolved" in self.data.keys() else None
+        self.target = {"id": self.data["target_id"], "message": None}
 
-        try:
-            self.menu_authors = (
-                self.data["resolved"]["members"] if "resolved" in self.data.keys() else None
-            )
-        except:  # noqa
-            self.menu_authors = None
-
-        self.context_message = (
-            [msg for msg in self.menu_messages][0] if self.menu_messages is not None else []
-        )
-        self.context_author = (
-            [user for user in self.menu_authors][0] if self.menu_authors is not None else []
-        )
-
-        if self.guild and self.context_author:
-            print(self.context_author)
-            self.context_author = discord.Member(
-                data=self.context_author, state=self.bot._connection, guild=self.guild
-            )
-
-        try:
-            if self.menu_messages is None:
-                self.menu_messages = model.SlashMessage(
+        if self._resolved is not None:
+            if self.context_type in [3, "3"]:
+                self.target["message"] = model.SlashMessage(
                     state=self.bot._connection,
                     channel=_discord.get_channel(self.channel_id),
-                    data=self.context_message,
+                    data=[msg for msg in self._resolved["messages"]][0],
                     _http=_http,
                     interaction_token=self._token,
                 )
-            else:
-                raise KeyError
-        except:  # noqa
-            return
+
+            try:
+                if self.guild and self._resolved["members"]:
+                    _member = True
+                    self.target["author"] = discord.Member(
+                        data=[auth for auth in self._resolved["members"]][0],
+                        state=self.bot._connection,
+                        guild=self.guild
+                    )
+            except KeyError:  # noqa
+                pass
+            try:
+                if self._resolved["users"]:
+                    self.target["author"] = discord.User(
+                        data=[auth for auth in self._resolved["users"]][0],
+                        state=self.bot_connection
+                    )
+            except KeyError:  # noqa
+                pass
 
     @property
     def cog(self) -> typing.Optional[commands.Cog]:

discord_slash/model.py

@@ -311,7 +311,7 @@ class CommandObject(CallbackObject):
 
     def __init__(self, name, cmd, _type=1):  # Let's reuse old command formatting.
         super().__init__(cmd["func"])
-        self.name = name.lower()
+        self.name = name.lower() if _type == 1 else name
         self.description = cmd["description"]
         self.allowed_guild_ids = cmd["guild_ids"] or []
         self.options = cmd["api_options"] or []