update for huggingscience server

app.py

@@ -31,12 +31,29 @@ bot = commands.Bot(command_prefix='!', intents=intents, max_messages=1000000)
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.DEBUG)
 
+# ============================================================
+# CONFIGURATION — edit these for your server
+# ============================================================
+GUILD_ID = 1417920887549857874                      # your server's ID
+LOG_CHANNEL_ID = 1418169465555910699                # channel where mod events are posted
+ADMIN_USER_ID = 847651116863062047                  # ban-appeal contact + owner of !role_buttons
+EXEMPT_ROLE_IDS = {                                 # roles exempt from spam detection / autoban
+    1418158261735264327,
+    1419966815559225344,
+    1418179704065884243,
+    1418886227830112297,
+    1418510579894845472,
+}
+ALERT_ROLE_ID = 1418510579894845472                 # role pinged on spam alerts / bans
+ADMIN_CHANNEL_ID = None                             # set to a channel ID to suppress spam warnings there
+SELF_ASSIGN_ROLE_IDS = []                           # role IDs for !role_buttons (empty disables)
+# ============================================================
+
 #rate_limiter = RateLimiter(max_calls=10, period=60)  # needs testing
 message_cache = {}
 
 AUTO_BAN_ALERT_THRESHOLD = 7
-AUTO_BAN_EXEMPT_ROLE_IDS = {897381378172264449, 897376942817419265}
-# moderator = 1078351789843292311
+AUTO_BAN_EXEMPT_ROLE_IDS = EXEMPT_ROLE_IDS
 
 def is_exempt(member: discord.Member) -> bool:
     return any(getattr(r, "id", None) in AUTO_BAN_EXEMPT_ROLE_IDS for r in getattr(member, "roles", []))
@@ -52,8 +69,8 @@ async def on_message(message):
         global number_of_messages
         if message.author != bot.user:
             message_cache[message.id] = message
-            lunarflu = bot.get_user(811235357663297546) #811235357663297546
-            
+            admin_user = bot.get_user(ADMIN_USER_ID)
+
             """Backup"""
 
             number_of_messages = number_of_messages + 1
@@ -69,20 +86,22 @@ async def on_message(message):
             if message.embeds:
                 for embed in message.embeds:
                     backup_message += f"\nEmbed Title: {embed.title}\nEmbed Description: {embed.description}"
-            dm_message = await lunarflu.send(backup_message)
+            if admin_user is not None:
+                dm_message = await admin_user.send(backup_message)
 
             """Antispam"""
             #Detecting certain unwanted strings
             try:
-                forbidden_patterns = [r"@everyone", 
-                                      r"@here", 
+                forbidden_patterns = [r"@everyone",
+                                      r"@here",
                                       r"(https?:\/\/|http?:\/\/)?(www.)?(discord.(gg|io|me|li)|discordapp.com\/invite|discord.com\/invite)\/[^\s\/]+?(?=\b)"]
                 if any(re.search(pattern, message.content, re.IGNORECASE) for pattern in forbidden_patterns):
-                    ignored_role_ids = [897381378172264449, 897376942817419265] #admins, huggingfolks
+                    ignored_role_ids = list(EXEMPT_ROLE_IDS)
                     if any(role.id in ignored_role_ids for role in message.author.roles):
-                        if message.author != lunarflu:
+                        if message.author.id != ADMIN_USER_ID:
                             return
-                    dm_unwanted = await lunarflu.send(f" {lunarflu.mention} [experimental] SUSPICIOUS MESSAGE: {message_link} | {message.author}: {message.content}")
+                    if admin_user is not None:
+                        dm_unwanted = await admin_user.send(f" {admin_user.mention} [experimental] SUSPICIOUS MESSAGE: {message_link} | {message.author}: {message.content}")
             except Exception as e:
                 print(f"Antispam->Detecting certain unwanted strings Error: {e}")
 
@@ -127,7 +146,7 @@ async def on_message(message):
                         # warning for 4+
                         channel = message.channel
                         if spam_count == false_positive_threshold:
-                            if channel.id != 996580741121065091: # admin channel excluded due to how automod messages are categorized by the discord bot
+                            if ADMIN_CHANNEL_ID is None or channel.id != ADMIN_CHANNEL_ID:
                                 await channel.send(f"{message.author.mention}, you may be posting too quickly! Please slow down a bit 🤗")
                         
                         var1 = message.created_at
@@ -135,11 +154,7 @@ async def on_message(message):
                         print(f"seconds since last message by {message.author}: {(var1 - var2).total_seconds()}")   
                         print(f"spam_count: {spam_count}")
 
-                        test_server = os.environ.get('TEST_SERVER')
-                        if test_server == 'True':
-                            alert = "<@&1106995261487710411>" # test @alerts role
-                        if test_server == 'False':
-                            alert = "<@&1108342563628404747>" # normal @alerts role
+                        alert = f"<@&{ALERT_ROLE_ID}>"
 
                         await bot.log_channel.send(
                             f"[EXPERIMENTAL ALERT] {message.author} may be posting too quickly! \n"
@@ -162,13 +177,7 @@ async def on_message(message):
                                 print("Autoban: could not resolve Member, skipping.")
                             else:
                                 # Prepare alert ping safely
-                                test_server = os.environ.get('TEST_SERVER')
-                                if test_server == 'True':
-                                    alert = "<@&1106995261487710411>"  # test alerts
-                                elif test_server == 'False':
-                                    alert = "<@&1108342563628404747>"  # prod alerts
-                                else:
-                                    alert = ""  # no ping fallback
+                                alert = f"<@&{ALERT_ROLE_ID}>"
                         
                                 # Skip bots and exempt roles
                                 if member.bot or is_exempt(member):
@@ -194,7 +203,7 @@ async def on_message(message):
                                             try:
                                                 await member.send(
                                                     f"You have been automatically banned from **{message.guild.name}** for repeated spam. "
-                                                    f"If this was an error, please contact @lunarflu."
+                                                    f"If this was an error, please contact <@{ADMIN_USER_ID}>."
                                                 )
                                             except Exception as dm_err:
                                                 print(f"Could not DM user before ban: {dm_err}")
@@ -372,7 +381,7 @@ async def on_member_ban(guild, banned_user):
             else:
                 print(f'{entry2.user} banned {entry2.target} (no reason specified)')             
             
-            content = "<@&1108342563628404747>" # @alerts role
+            content = f"<@&{ALERT_ROLE_ID}>"
             embed = Embed(color=Color.red())
             embed.set_author(name=f"{entry2.target}    ID: {entry2.target.id}", icon_url=entry2.target.avatar.url if entry2.target.avatar else bot.user.avatar.url)
             embed.title = "User Banned"
@@ -385,11 +394,10 @@ async def on_member_ban(guild, banned_user):
             embed.add_field(name="Reason", value=ban_reason, inline=False)
             embed.set_footer(text=f"{convert_to_timezone(datetime.utcnow(), zurich_tz)}")
 
-            #user = bot.get_user(811235357663297546)
-            await bot.log_channel.send(content=content, embed=embed)  
+            await bot.log_channel.send(content=content, embed=embed)
             
             try:
-                dm_message = await banned_user.send(f"You've been banned from the Hugging Face Discord. To appeal, reach out to <@811235357663297546> via DM")
+                dm_message = await banned_user.send(f"You've been banned from {guild.name}. To appeal, reach out to <@{ADMIN_USER_ID}> via DM")
             except Exception as e:
                 print(f"Could not send DM to banned user: {e}")
 
@@ -407,7 +415,7 @@ async def on_member_unban(guild, unbanned_user):
                 moderator = entry.user    
 
                 created_and_age = f"{unbanned_user.created_at}"
-                content = "<@&1108342563628404747>" # @alerts role
+                content = f"<@&{ALERT_ROLE_ID}>"
                 embed = Embed(color=Color.red())
                 embed.set_author(name=f"{unbanned_user}    ID: {unbanned_user.id}", icon_url=unbanned_user.avatar.url if unbanned_user.avatar else bot.user.avatar.url)
                 embed.title = "User Unbanned"
@@ -416,10 +424,8 @@ async def on_member_unban(guild, unbanned_user):
                 embed.add_field(name="Moderator", value=moderator.mention, inline=True)
                 embed.add_field(name="Nickname", value=moderator.nick, inline=True)
                 embed.set_footer(text=f"{convert_to_timezone(datetime.utcnow(), zurich_tz)}")
-                
-                #user = bot.get_user(811235357663297546)
-                #dm_message = await user.send(content=content, embed=embed)
-                await bot.log_channel.send(content=content, embed=embed)   
+
+                await bot.log_channel.send(content=content, embed=embed)
               
     except Exception as e:
         print(f"on_member_unban Error: {e}")  
@@ -431,7 +437,7 @@ async def on_member_unban(guild, unbanned_user):
 async def on_member_join(member):
     try:
         await asyncio.sleep(5)
-        guild = bot.get_guild(879548962464493619)     
+        guild = bot.get_guild(GUILD_ID)
   
         embed = Embed(color=Color.blue())
         avatar_url = member.avatar.url if member.avatar else bot.user.avatar.url
@@ -524,25 +530,6 @@ async def on_guild_role_delete(role):
         print(f"on_guild_role_delete Error: {e}")
 
     
-@bot.event
-async def on_guild_role_update(before, after):
-    try:
-        # editing roles, could expand this 
-        if before.name != after.name:
-            embed = Embed(description=f'Role {before.mention} was renamed to {after.name}', color=Color.orange())
-            await bot.log_channel.send(embed=embed)
-    
-        if before.permissions.administrator != after.permissions.administrator:
-            # changes involving the administrator permission / sensitive permissions (can help to prevent mistakes)
-            content = "<@&1108342563628404747>" # @alerts role
-            embed = Embed(description=f'Role {after.mention} had its administrator permission {"enabled" if after.permissions.administrator else "disabled"}', color=Color.red())
-            await bot.log_channel.send(content=content, embed=embed)  
-    except Exception as e:
-        print(f"on_guild_role_update Error: {e}")  
-
-
-
-
 
 
 @bot.event
@@ -651,14 +638,16 @@ class PersistentRoleSelectionView(View):
 # Command that sends the role buttons message.
 @bot.command(name="role_buttons")
 async def role_buttons(ctx):
-    # Only allow lunarflu (ID 811235357663297546) to use this command.
-    if ctx.author.id != 811235357663297546:
+    # Only the configured admin user is allowed to invoke this command.
+    if ctx.author.id != ADMIN_USER_ID:
         await ctx.send("You are not authorized to use this command.", delete_after=10)
         return
 
-    # List the role IDs you want to include (replace these with your actual role IDs).
-    role_ids = [1014517792550166630, 1014548568238997616, 1014548769355862036, 1077250031180071023, 1093982736961785877, 1359533406102487281, 1174686719166124168, 1361696873764618381]
-    roles = [ctx.guild.get_role(rid) for rid in role_ids if ctx.guild.get_role(rid) is not None]
+    if not SELF_ASSIGN_ROLE_IDS:
+        await ctx.send("No self-assignable roles configured. Set SELF_ASSIGN_ROLE_IDS in app.py.", delete_after=15)
+        return
+
+    roles = [ctx.guild.get_role(rid) for rid in SELF_ASSIGN_ROLE_IDS if ctx.guild.get_role(rid) is not None]
 
     # Create the persistent view.
     view = PersistentRoleSelectionView(roles)
@@ -691,24 +680,27 @@ async def on_ready():
     await asyncio.sleep(5)
     print('Logged on as', bot.user)
     await asyncio.sleep(5)
-    bot.log_channel = bot.get_channel(1036960509586587689) # admin-logs
+    bot.log_channel = bot.get_channel(LOG_CHANNEL_ID)
     await asyncio.sleep(5)
     print(bot.log_channel)
-    guild = bot.get_guild(879548962464493619)
-    
+    guild = bot.get_guild(GUILD_ID)
+
+    if guild and SELF_ASSIGN_ROLE_IDS:
+        roles = [guild.get_role(rid) for rid in SELF_ASSIGN_ROLE_IDS if guild.get_role(rid) is not None]
+        if roles:
+            persistent_view = PersistentRoleSelectionView(roles)
+            bot.add_view(persistent_view)  # This makes the view persistent across restarts.
+
     if guild:
-        role_ids = [1014517792550166630, 1014548568238997616, 1014548769355862036, 1077250031180071023, 1093982736961785877, 1359533406102487281, 1174686719166124168, 1361696873764618381]
-        roles = [guild.get_role(rid) for rid in role_ids if guild.get_role(rid) is not None]
-        persistent_view = PersistentRoleSelectionView(roles)
-        bot.add_view(persistent_view)  # This makes the view persistent across restarts.
-    
-    for channel in guild.text_channels: # helps with more accurate logging across restarts
-        try:
-            message_cache.update({m.id: m async for m in channel.history(limit=10000)})
-            print(f"Finished caching messages for channel: {channel.name}")
-        except Exception as e:
-            print(f"An error occurred while fetching messages from {channel.name}: {e}")
-        await asyncio.sleep(0.1)
+        for channel in guild.text_channels:  # helps with more accurate logging across restarts
+            try:
+                message_cache.update({m.id: m async for m in channel.history(limit=10000)})
+                print(f"Finished caching messages for channel: {channel.name}")
+            except Exception as e:
+                print(f"An error occurred while fetching messages from {channel.name}: {e}")
+            await asyncio.sleep(0.1)
+    else:
+        print(f"on_ready: could not find guild with ID {GUILD_ID}. Is the bot in the server?")