matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Implement _FakeSlashMessage.from_interaction

Browse Source
This commit is contained in:
Gnome 2021-09-01 11:51:45 +01:00
parent 0aa8c5bef3
commit 5a49e1bf68
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
discord/ext/commands/bot.py
View File

@ -79,6 +79,13 @@ class _FakeSlashMessage(discord.PartialMessage):
author: Union[discord.User, discord.Member] author: Union[discord.User, discord.Member]
tts = False tts = False
@classmethod
def from_interaction(cls, interaction: discord.Interaction) -> discord.Message:
self: discord.Message = cls(channel=interaction.channel, id=interaction.id) # type: ignore
assert interaction.user is not None
self.author = interaction.user
return self
def when_mentioned(bot: Union[Bot, AutoShardedBot], msg: Message) -> List[str]: def when_mentioned(bot: Union[Bot, AutoShardedBot], msg: Message) -> List[str]:
"""A callable that implements a command prefix equivalent to being mentioned. """A callable that implements a command prefix equivalent to being mentioned.
@ -1088,6 +1095,9 @@ class BotBase(GroupMixin):
else: else:
return # cannot do anything without stable channel return # cannot do anything without stable channel
interaction.channel = channel # type: ignore
del channel
# Fetch out subcommands from the options # Fetch out subcommands from the options
command_name = interaction.data['name'] command_name = interaction.data['name']
command_options = interaction.data.get('options') or [] command_options = interaction.data.get('options') or []
@ -1101,10 +1111,8 @@ class BotBase(GroupMixin):
if command is None: if command is None:
raise errors.CommandNotFound(f'Command "{command_name}" is not found') raise errors.CommandNotFound(f'Command "{command_name}" is not found')
message: discord.Message = _FakeSlashMessage(id=interaction.id, channel=channel) # type: ignore
message.author = interaction.user
# Fetch a valid prefix, so process_commands can function # Fetch a valid prefix, so process_commands can function
message = _FakeSlashMessage.from_interaction(interaction)
prefix = await self.get_prefix(message) prefix = await self.get_prefix(message)
if isinstance(prefix, list): if isinstance(prefix, list):
prefix = prefix[0] prefix = prefix[0]