add delete message fun

rtk-rnjn · Sep 9, 2023 · 8a4a502 · 8a4a502
1 parent 55dbd6e
commit 8a4a502
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cogs/reminder/__init__.py b/cogs/reminder/__init__.py
@@ -76,7 +76,8 @@ async def notify_reminder_msg(self, ctx: Context, *, timestamp: int | float) ->
         )
         embed = discord.Embed(description=_id, color=discord.Color.blurple())
         try:
-            await ctx.author.send(msg, embed=embed, view=ctx.send_view())
+            notify_message = await ctx.author.send(msg, embed=embed, view=ctx.send_view())
+            await self.bot.wait_and_delete(message=notify_message, timestamp=timestamp)
             await ctx.tick()
         except discord.Forbidden:
             await ctx.reply(msg, embed=embed)

diff --git a/core/Parrot.py b/core/Parrot.py
@@ -1568,7 +1568,7 @@ async def wait_and_delete(
         self,
         *,
         delay: int | None = None,
-        timestamp: int | None = None,
+        timestamp: float | None = None,
         channel_id: int | None = None,
         message_id: int | None = None,
         message: discord.Message | None = None,