Dm safe

blurple/io/reaction.py

@@ -1,6 +1,5 @@
-import discord
-
 import blurple.io as io
+import discord
 
 
 class ReactionAddBasic(io.Reply):
@@ -24,8 +23,8 @@ class ReactionRemoveBasic(ReactionAddBasic):
     event = "raw_reaction_remove"
 
 
-class ReactionAddReply(ReactionAddBasic):
-    """ Ask for the user's reaction reply.
+class DirectMessageReactionAddReply(ReactionAddBasic):
+    """ Ask for the user's reaction reply. Safe to use in Direct Messages.
 
         :Example Usage:
         .. code-block:: python
@@ -45,6 +44,15 @@ def reply_check(self, payload: discord.RawReactionActionEvent):
         return payload.user_id == self.ctx.author.id and \
                payload.message_id == self.message.id
 
+
+class ReactionAddReply(DirectMessageReactionAddReply):
+    """ Ask for the user's reaction reply.
+
+        :Example Usage:
+        .. code-block:: python
+
+            reply = await io.ReactionAddBasic(ctx, validate=["✅", "❎"]).result()
+    """
     async def on_reply_attempt(self, payload: discord.RawReactionActionEvent):
         """Specialized to remove the user's reaction."""
         await self.message.remove_reaction(payload.emoji, self.ctx.bot.get_user(payload.user_id))

blurple/io/reply.py

@@ -1,13 +1,14 @@
 from __future__ import annotations
-import re
+
+import asyncio
 import inspect
+import re
 import typing as t
 from abc import ABC
-import discord
-from discord.ext import commands
-import asyncio
 
 import blurple.ui as ui
+import discord
+from discord.ext import commands
 
 
 class Reply(ABC):
@@ -131,8 +132,8 @@ async def _validate_reply(cls, reply, valid: t.Union[str, t.Container, t.Callabl
             return content in valid
         if callable(valid):
             if inspect.iscoroutinefunction(object):
-                return await valid()
-            return valid()
+                return await valid(reply)
+            return valid(reply)
 
     @staticmethod
     def _get_reply_content(reply):