Merge branch 'dbfix' into 'master'

bot.py

@@ -23,7 +23,7 @@ load_dotenv()
 bot_token = os.getenv("BOT_TOKEN")
 log_chat_id = os.getenv("LOG_CHAT_ID")
 main_chat_id = os.getenv("MAIN_CHAT_ID")
-
+db_path = os.getenv("DB_PATH")
 # Вызов функции для создания базы данных или подключения к существующей
 create_db()
 
@@ -156,7 +156,7 @@ async def handle_message(message: Message):
     text = " ".join(normalize_text_to_infinitive(text, chat_id))  # Убедимся, что это строка
 
     # Проверяем, если это первое сообщение пользователя
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
     
     cursor.execute('SELECT messages_checked FROM users WHERE id = ?', (user_id,))
@@ -314,4 +314,4 @@ async def main():
 
 if __name__ == "__main__":
     create_db()  # Создание базы данных и таблиц
-    asyncio.run(main())
+    asyncio.run(main())
\ No newline at end of file

sqlite_tools.py

@@ -2,9 +2,10 @@
 import sqlite3
 import os
 
+db_path = os.getenv("DB_PATH")
+
 def create_db():
-    db_path = os.getenv("DB_PATH")
-    
+
     # Подключаемся к базе данных (если она не существует, она будет создана)
     conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
@@ -43,7 +44,7 @@ def create_db():
 
 # Функция для добавления пользователя в базу данных
 def add_user(user_id, first_name, last_name, username):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
 
     cursor.execute('''
@@ -56,7 +57,7 @@ def add_user(user_id, first_name, last_name, username):
 
 # Функция для проверки, есть ли пользователь в базе данных
 def user_exists(user_id):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
 
     cursor.execute('SELECT id FROM users WHERE id = ?', (user_id,))
@@ -67,7 +68,7 @@ def user_exists(user_id):
 
 # Функция для перемещения пользователя в таблицу забаненных
 def ban_user(user_id, first_name, last_name, username, ban_reason):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
 
     # Добавляем в таблицу забаненных
@@ -83,11 +84,11 @@ def ban_user(user_id, first_name, last_name, username, ban_reason):
     conn.close()
 
 def is_user_banned(user_id):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
 
     cursor.execute('SELECT id FROM banned_users WHERE id = ?', (user_id,))
     result = cursor.fetchone()
 
     conn.close()
-    return result is not None     
+    return result is not None     
\ No newline at end of file