fix issue #14

Kilvoctu · Oct 29, 2022 · 81ec359 · 81ec359
1 parent bf7cc3a
commit 81ec359
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/aiya.py b/aiya.py
@@ -51,9 +51,13 @@ async def on_raw_reaction_add(ctx):
             if message.embeds:
                 if message.embeds[0].footer.text == f'{ctx.member.name}#{ctx.member.discriminator}':
                     await message.delete()
+        # this is for deleting generations in DMs
         except:
-            #todo: I need to add a way for bot to delete DM generations
-            pass
+            channel = await self.fetch_user(ctx.user_id)
+            message = await channel.fetch_message(ctx.message_id)
+            if message.embeds:
+                if message.embeds[0].footer.text == f'{ctx.member.name}#{ctx.member.discriminator}':
+                    await message.delete()
 
 @self.event
 async def on_guild_join(guild):