Changed admin check

classes/holo_user.py

@@ -5,6 +5,7 @@ import discord
 import discord.member
 
 from modules.database import col_users, col_warnings
+from modules.utils import config_get
 
 logger = logging.getLogger(__name__)
 
@@ -98,6 +99,45 @@ class HoloUser:
         )
         logger.info(f"Set attribute {key} of user {self.id} to {value}")
 
+    async def is_moderator(
+        self, member: Union[discord.User, discord.Member, discord.member.Member]
+    ) -> bool:
+        """Check if user is moderator or council member
+
+        ### Args:
+            * `member` (Union[discord.User, discord.Member, discord.member.Member]): Member object
+
+        ### Returns:
+            `bool`: `True` if member is a moderator or member of council and `False` if not
+        """
+        if isinstance(member, discord.User):
+            return False
+        moderator_role = await config_get("moderator", "roles")
+        council_role = await config_get("council", "roles")
+        for role in member.roles:
+            if role.id == moderator_role or role.id == council_role:
+                return True
+        return False
+
+    async def is_council(
+        self, member: Union[discord.User, discord.Member, discord.member.Member]
+    ) -> bool:
+        """Check if user is a member of council
+
+        ### Args:
+            * `member` (Union[discord.User, discord.Member, discord.member.Member]): Member object
+
+        ### Returns:
+            `bool`: `True` if member is a member of council and `False` if not
+        """
+        if isinstance(member, discord.User):
+            return False
+        council_role = await config_get("council", "roles")
+        for role in member.roles:
+            if role.id == council_role:
+                return True
+        return False
+
     # def purge(self) -> None:
     #     """Completely remove data from database. Will not remove transactions logs and warnings."""
     #     col_users.delete_one(filter={"_id": self.db_id})