Commit a80471cb authored by Ivan Mazhukin's avatar Ivan Mazhukin

fix db connection via DB_PATH env

parent e2cf6ee0
......@@ -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,))
......
......@@ -2,8 +2,9 @@
import sqlite3
import os
db_path = os.getenv("DB_PATH")
def create_db():
db_path = os.getenv("DB_PATH")
# Подключаемся к базе данных (если она не существует, она будет создана)
conn = sqlite3.connect(db_path)
......@@ -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,7 +84,7 @@ 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,))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment