feat: implement caching of banned users

should work faster than by using a db request for each update

Author: devcarron

database/database.py

@@ -12,6 +12,7 @@ class Database(object):
 
     _instance = None
     _initialized = False
+    _banned_users = set()
 
     def __new__(cls):
         if not Database._instance:
@@ -37,6 +38,8 @@ def __init__(self):
         self.connection.text_factory = lambda x: str(x, 'utf-8', "ignore")
         self.cursor = self.connection.cursor()
 
+        self.load_banned_users()
+
         self._initialized = True
 
     @staticmethod
@@ -77,6 +80,22 @@ def create_database(database_path):
         connection.commit()
         connection.close()
 
+    def load_banned_users(self):
+        """Loads all banned users from the database into a list"""
+        self.cursor.execute("SELECT user_id FROM users WHERE banned=1;")
+        result = self.cursor.fetchall()
+
+        if not result:
+            return
+
+        for row in result:
+            print(int(row["user_id"]))
+            self._banned_users.add(int(row["user_id"]))
+
+    def get_banned_users(self):
+        """Returns a list of all banned user_ids"""
+        return self._banned_users
+
     def get_user(self, user_id):
         self.cursor.execute("SELECT user_id, first_name, last_name, username, games_played, games_won, games_tie, last_played, banned"
                             " FROM users WHERE user_id=?;", [str(user_id)])
@@ -88,18 +107,21 @@ def get_user(self, user_id):
 
     def is_user_banned(self, user_id):
         """Checks if a user was banned by the admin of the bot from using it"""
-        user = self.get_user(user_id)
-        return user is not None and user[8] == 1
+        # user = self.get_user(user_id)
+        # return user is not None and user[8] == 1
+        return int(user_id) in self._banned_users
 
     def ban_user(self, user_id):
         """Bans a user from using a the bot"""
         self.cursor.execute("UPDATE users SET banned=1 WHERE user_id=?;", [str(user_id)])
         self.connection.commit()
+        self._banned_users.add(int(user_id))
 
     def unban_user(self, user_id):
         """Unbans a user from using a the bot"""
         self.cursor.execute("UPDATE users SET banned=0 WHERE user_id=?;", [str(user_id)])
         self.connection.commit()
+        self._banned_users.remove(int(user_id))
 
     def get_recent_players(self):
         one_day_in_secs = 60 * 60 * 24