Код с добавленной бд

spam_keywords.py

@@ -22,7 +22,9 @@ SPAM_KEYWORDS = [
     "спецпредложение","эксклюзивный предложение","мой тг канал", "уникальное предложение", "только сегодня", "лютая жиза", "мерч", "продукция", "коллекция",
 
     # Работа
-    "удалённая работа", "человек команда", "удаленный деятельность", "удаленно", "гибкий график", "дистанционный деятельность", "третуется человек", "искать человек", "достойный доход", "удалённая работа", "бакс", "доллар", "нужен человек", "нужен помощник", "вакансии", "удалёнка", "удаленка","условия труда", "писать лс", "личка", "писать в лс", "личное сообщение", "лс", "партнер", "сотрудничество", "прибыль", "бабки", "пасивный заработок","без вложений", "подробности лс",
+    "удалённая работа", "человек команда", "удаленный деятельность", "удаленно", "гибкий график", "дистанционный деятельность", "третуется человек", "искать человек",
+    "достойный доход", "удалённая работа", "бакс", "доллар", "нужен человек", "нужен помощник", "вакансии", "удалёнка", "удаленка","условия труда", "писать лс", "личка",
+    "писать в лс", "личное сообщение", "лс", "л.с.", "подробный лс", "подробный л.с.", "партнер", "сотрудничество", "прибыль", "бабки", "пасивный заработок","без вложений", "подробности лс",
 
     # Мошенничество
     "лотерея", "быстрый выигрыш", "выигрыш", "приз", "деньги", "халява", "прибыль", "заработок", "зп", "подработка", "легкий заработок",

sqlite_tools.py

+# Функция для создания базы данных и таблиц
+import sqlite3
+
+
+def create_db():
+    db_path = 'users.db'
+    
+    # Подключаемся к базе данных (если она не существует, она будет создана)
+    conn = sqlite3.connect(db_path)
+    cursor = conn.cursor()
+
+    # Проверка, существует ли таблица (в случае первого запуска создадим её)
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='users';")
+    result = cursor.fetchone()
+
+    if not result:
+        # Если таблица users не существует, создаём её
+        cursor.execute('''
+            CREATE TABLE users (
+                id INTEGER PRIMARY KEY,
+                first_name TEXT,
+                last_name TEXT,
+                username TEXT,
+                messages_checked INTEGER DEFAULT 0
+            )
+        ''')
+
+        # Создание таблицы для забаненных пользователей
+        cursor.execute('''
+            CREATE TABLE banned_users (
+                id INTEGER PRIMARY KEY,
+                first_name TEXT,
+                last_name TEXT,
+                username TEXT,
+                ban_reason TEXT
+            )
+        ''')
+
+        conn.commit()
+
+    conn.close()
+
+
+# Функция для добавления пользователя в базу данных
+def add_user(user_id, first_name, last_name, username):
+    conn = sqlite3.connect('users.db')
+    cursor = conn.cursor()
+
+    cursor.execute('''
+        INSERT INTO users (id, first_name, last_name, username)
+        VALUES (?, ?, ?, ?)
+    ''', (user_id, first_name, last_name, username))
+
+    conn.commit()
+    conn.close()
+
+# Функция для проверки, есть ли пользователь в базе данных
+def user_exists(user_id):
+    conn = sqlite3.connect('users.db')
+    cursor = conn.cursor()
+
+    cursor.execute('SELECT id FROM users WHERE id = ?', (user_id,))
+    result = cursor.fetchone()
+
+    conn.close()
+    return result is not None
+
+# Функция для перемещения пользователя в таблицу забаненных
+def ban_user(user_id, first_name, last_name, username, ban_reason):
+    conn = sqlite3.connect('users.db')
+    cursor = conn.cursor()
+
+    # Добавляем в таблицу забаненных
+    cursor.execute('''
+        INSERT INTO banned_users (id, first_name, last_name, username, ban_reason)
+        VALUES (?, ?, ?, ?, ?)
+    ''', (user_id, first_name, last_name, username, ban_reason))
+
+    # Удаляем из таблицы активных пользователей
+    cursor.execute('DELETE FROM users WHERE id = ?', (user_id,))
+
+    conn.commit()
+    conn.close()
+
+def is_user_banned(user_id):
+    conn = sqlite3.connect('users.db')
+    cursor = conn.cursor()
+
+    cursor.execute('SELECT id FROM banned_users WHERE id = ?', (user_id,))
+    result = cursor.fetchone()
+
+    conn.close()
+    return result is not None     
\ No newline at end of file