From cb29505f08bf9b90f961e5fba2c4ef337b82704d Mon Sep 17 00:00:00 2001
From: deathbybandaid <github@deathbybandaid.net>
Date: Thu, 30 Jun 2022 09:43:23 -0400
Subject: [PATCH] test

---
 .../SBCore/users/__init__.py                  | 26 ++++++++++++-------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/sopel_SpiceBot_Core_1/SBCore/users/__init__.py b/sopel_SpiceBot_Core_1/SBCore/users/__init__.py
index 42b7910..52e7036 100644
--- a/sopel_SpiceBot_Core_1/SBCore/users/__init__.py
+++ b/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"""