test

sopel_SpiceBot_Core_1/SBCore/users/__init__.py

@@ -8,21 +8,29 @@ class Users():
     def __init__(self, bot):
         self.bot = None
 
+        self.priv_dict = {
+                            "OWNER": 10,
+                            "ADMIN": 5
+                            }
+
     """Bot Priviledges"""
 
     def has_bot_privilege(self, nick, privilege):
 
-        if privilege.upper() == "ADMIN":
-            if nick in self.list_bot_admin():
-                return True
-            elif nick in self.list_bot_owner():
-                return True
+        if not isinstance(privilege, int):
+            if privilege.upper() in list(self.priv_dict.keys()):
+                privilege = self.priv_dict[privilege.upper()]
+            else:
+                privilege = 0
 
-        elif privilege.upper() == "OWNER":
-            if nick in self.list_bot_owner():
-                return True
+        return self.get_nick_bot_privilege(nick) >= privilege
 
-        return False
+    def get_nick_bot_privilege(self, nick):
+        if self.is_bot_owner(nick):
+            return self.priv_dict["OWNER"]
+        elif self.is_bot_admin(nick):
+            return self.priv_dict["ADMIN"]
+        return 0
 
     """Bot Owner"""