Jelajahi Sumber

Fix error when getting allowed role for a non registered server

Lucas Villeneuve 5 tahun lalu
induk
melakukan
80d39b7392
1 mengubah file dengan 6 tambahan dan 3 penghapusan
  1. 6 3
      myanimebot/utils.py

+ 6 - 3
myanimebot/utils.py

@@ -352,12 +352,15 @@ def insert_user_into_db(user_name : str, service : Service, servers : str) -> bo
 def get_allowed_role(server : int) -> int:
     '''Return the allowed role for a given server'''
 
-    cursor = globals.conn.cursor(buffered=True)
+    cursor = globals.conn.cursor(buffered=True, dictionary=True)
     cursor.execute("SELECT admin_group FROM t_servers WHERE server=%s LIMIT 1", [str(server)])
     allowedRole = cursor.fetchone()
     cursor.close()
 
-    return allowedRole[0]
+    if allowedRole is None:
+        return None
+
+    return allowedRole["admin_group"]
 
 def get_role_name(provided_role_id : int, server) -> str :
     ''' Convert a role ID into a displayable name '''
@@ -369,4 +372,4 @@ def get_role_name(provided_role_id : int, server) -> str :
                 role_name = role.name
         if role_name is None:
             role_name = "[DELETED ROLE]"
-    return role_name
+    return role_name