resolve requested changes

vutfitdiscord · Sep 19, 2024 · 8282c88 · 8282c88
1 parent d9917f8
commit 8282c88
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cogs/roles/cog.py b/cogs/roles/cog.py
@@ -255,4 +255,5 @@ async def remove_exclusive_roles(
 
     @commands.Cog.listener()
     async def on_member_join(self, member: disnake.Member):
-        await member.add_roles(member.guild.get_role(Base.config.newbie_role))
+        if member.guild == Base.config.guild_id:  # We're on VUT FIT guild
+            await member.add_roles(member.guild.get_role(Base.config.newbie_role))
diff --git a/features/verification.py b/features/verification.py
@@ -322,5 +322,5 @@ async def clear_host_roles(self, inter: disnake.ApplicationCommandInteraction):
 
         verify = disnake.utils.get(guild.roles, name="Verify")
         zajemce = disnake.utils.get(guild.roles, name="ZajemceoStudium")
-        newbie = member.guild.get_role(config.newbie_role)
+        newbie = disnake.utils.get(id=config.newbie_role)
         await member.remove_roles(host, verify, zajemce, newbie, reason="Verify user")