Add support for bulk deletion

1 files changed, 9 insertions, 1 deletions

diff --git a/metricity/bot.py b/metricity/bot.py
index 3295ed7..e3e11fc 100644
--- a/metricity/bot.py
+++ b/metricity/bot.py
@@ -7,7 +7,7 @@ from typing import Any, Generator, List
 from asyncpg.exceptions import UniqueViolationError
 from discord import (
     CategoryChannel, Game, Guild, Intents,
-    Member, Message as DiscordMessage, RawMessageDeleteEvent,
+    Member, Message as DiscordMessage, RawBulkMessageDeleteEvent, RawMessageDeleteEvent,
     VoiceChannel
 )
 from discord.abc import Messageable
@@ -330,6 +330,14 @@ async def on_raw_message_delete(message: RawMessageDeleteEvent) -> None:
         await message.update(is_deleted=True).apply()
 
 
[email protected]
+async def on_raw_bulk_message_delete(messages: RawBulkMessageDeleteEvent) -> None:
+    """If messages are deleted in bulk and we have a record of them set the is_deleted flag."""
+    for message_id in messages.message_ids:
+        if message := await Message.get(str(message_id)):
+            await message.update(is_deleted=True).apply()
+
+
 @bot.command()
 async def opt_in(ctx: Context) -> None:
     """Opt-in to the server analytics system."""