feat: store punished users

bot/events_handlers/member_events.py

@@ -1,4 +1,9 @@
 from bot.utils import generate_random_nickname, is_valid_username, is_numeric_name
+from database import user_has_role
+
+import discord
+
+SPAM_ROLE_ID = 1350511935677927514
 
 
 async def handle_member_join(member):
@@ -15,6 +20,11 @@ async def handle_member_join(member):
         new_nick = generate_random_nickname()
         await member.edit(nick=new_nick)
 
+    if user_has_role(member.id):
+        spam_role = discord.utils.get(member.guild.roles, id=SPAM_ROLE_ID)
+        if spam_role:
+            await member.add_roles(spam_role)
+
 
 async def handle_member_update(before, after):
     name_to_check = after.name

bot/events_handlers/message_events.py

@@ -1,9 +1,12 @@
 from datetime import timedelta
 import re
 import discord
+
 from bot.utils import timeout_member, aware_utcnow
 from bot.config import message_patterns
 
+from database import add_user_to_role
+
 BOT_LOG = 1112049391482703873
 
 CRAZY_USER_ID = 1319364607487512658
@@ -210,7 +213,12 @@ async def handle_message(message, bot):
             spam_role = message.guild.get_role(SPAM_ROLE_ID)
             member = message.guild.get_member(message.author.id)
 
-            await member.add_roles(spam_role)
+            # Check if the member already has the spam role
+            if spam_role not in member.roles:
+                await member.add_roles(spam_role)
+
+                # Add the user to the database
+                add_user_to_role(member.id, SPAM_ROLE_ID)
 
             await message.reply(
                 f"Dink Donk! Time to ping everyone! {spam_role.mention}",