feat: implement banning of users

Important: This doesn't handle banned users correctly yet - it's just the commands for banning users

Author: devcarron

blackjackbot/commands/admin/commands.py

@@ -3,6 +3,8 @@
 import logging
 import re
 
+from telegram import ParseMode
+
 from blackjackbot.commands.admin import notify_admins
 from blackjackbot.commands.util.decorators import admin_method
 from blackjackbot.errors import NoActiveGameException
@@ -13,6 +15,48 @@
 logger = logging.getLogger(__name__)
 
 
+@admin_method
+def ban_user_cmd(update, context):
+    """Bans a user from using the bot"""
+    # Try to get user_id from command
+    if len(context.args) != 1:
+        update.message.reply_text("Please provide a valid user_id - `/ban <user_id>`", parse_mode=ParseMode.MARKDOWN_V2)
+        return
+
+    match = re.search(r"^\d+$", context.args[0])
+    if not match:
+        logger.error(f"The user_id did not match: {context.args}")
+        return
+    user_id = match.group(0)
+
+    db = Database()
+    db.ban_user(user_id=user_id)
+
+    logger.info(f"Admin '{update.effective_user.id}' banned user '{user_id}'!")
+    notify_admins(f"Admin '{update.effective_user.id}' banned user '{user_id}'!")
+
+
+@admin_method
+def unban_user_cmd(update, context):
+    """Unbans a user from using the bot"""
+    # Try to get user_id from command
+    if len(context.args) != 1:
+        update.message.reply_text("Please provide a valid user_id - `/unban <user_id>`", parse_mode=ParseMode.MARKDOWN_V2)
+        return
+
+    match = re.search(r"^\d+$", context.args[0])
+    if not match:
+        logger.error(f"The user_id did not match: {context.args}")
+        return
+    user_id = match.group(0)
+
+    db = Database()
+    db.unban_user(user_id=user_id)
+
+    logger.info(f"Admin '{update.effective_user.id}' unbanned user '{user_id}'!")
+    notify_admins(f"Admin '{update.effective_user.id}' unbanned user '{user_id}'!")
+
+
 @admin_method
 def kill_game_cmd(update, context):
     """Kills the game for a certain chat/group"""

database/database.py

@@ -3,6 +3,7 @@
 import os
 import sqlite3
 from time import time
+
 from util import Cache
 
 
@@ -63,6 +64,7 @@ def create_database(database_path):
                        "'games_won' INTEGER DEFAULT 0,"
                        "'games_tie' INTEGER DEFAULT 0,"
                        "'last_played' INTEGER DEFAULT 0,"
+                       "'banned' INTEGER DEFAULT 0,"
                        "PRIMARY KEY('user_id'));")
 
         cursor.execute("CREATE TABLE IF NOT EXISTS 'chats'"
@@ -73,14 +75,29 @@ def create_database(database_path):
         connection.close()
 
     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"
+        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)])
 
         result = self.cursor.fetchone()
         if not result or len(result) == 0:
             return None
         return result
 
+    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
+
+    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()
+
+    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()
+
     def get_recent_players(self):
         one_day_in_secs = 60 * 60 * 24
         current_time = int(time())