diff options
| author | Rapptz <[email protected]> | 2017-07-02 01:43:33 -0400 |
|---|---|---|
| committer | Rapptz <[email protected]> | 2017-07-02 01:43:45 -0400 |
| commit | 8caa088d259b8640ed6429bc21bdebebf8443c74 (patch) | |
| tree | 6acd019a2b47408ed4b233e8c6add34f2a050630 /discord/message.py | |
| parent | Fix Message.remove_reaction and Message.add_reaction taking in Reaction (diff) | |
| download | discord.py-8caa088d259b8640ed6429bc21bdebebf8443c74.tar.xz discord.py-8caa088d259b8640ed6429bc21bdebebf8443c74.zip | |
Allow PartialReactionEmoji in add_reaction and remove_reaction.
Diffstat (limited to 'discord/message.py')
| -rw-r--r-- | discord/message.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/discord/message.py b/discord/message.py index d321663f..3efea9fb 100644 --- a/discord/message.py +++ b/discord/message.py @@ -29,7 +29,7 @@ import re from . import utils from .reaction import Reaction -from .emoji import Emoji +from .emoji import Emoji, PartialReactionEmoji from .calls import CallMessage from .enums import MessageType, try_enum from .errors import InvalidArgument, ClientException, HTTPException, NotFound @@ -593,7 +593,7 @@ class Message: Parameters ------------ - emoji: Union[:class:`Emoji`, :class:`Reaction`, str] + emoji: Union[:class:`Emoji`, :class:`Reaction`, :class:`PartialReactionEmoji`, str] The emoji to react with. Raises @@ -613,6 +613,8 @@ class Message: if isinstance(emoji, Emoji): emoji = '%s:%s' % (emoji.name, emoji.id) + elif isinstance(emoji, PartialReactionEmoji): + emoji = emoji._as_reaction() elif isinstance(emoji, str): pass # this is okay else: @@ -636,7 +638,7 @@ class Message: Parameters ------------ - emoji: Union[:class:`Emoji`, :class:`Reaction`, str] + emoji: Union[:class:`Emoji`, :class:`Reaction`, :class:`PartialReactionEmoji`, str] The emoji to remove. member: :class:`abc.Snowflake` The member for which to remove the reaction. @@ -658,6 +660,8 @@ class Message: if isinstance(emoji, Emoji): emoji = '%s:%s' % (emoji.name, emoji.id) + elif isinstance(emoji, PartialReactionEmoji): + emoji = emoji._as_reaction() elif isinstance(emoji, str): pass # this is okay else: |