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() ...@@ -23,7 +23,7 @@ load_dotenv()
bot_token = os.getenv("BOT_TOKEN") bot_token = os.getenv("BOT_TOKEN")
log_chat_id = os.getenv("LOG_CHAT_ID") log_chat_id = os.getenv("LOG_CHAT_ID")
main_chat_id = os.getenv("MAIN_CHAT_ID") main_chat_id = os.getenv("MAIN_CHAT_ID")
db_path = os.getenv("DB_PATH")
# Вызов функции для создания базы данных или подключения к существующей # Вызов функции для создания базы данных или подключения к существующей
create_db() create_db()
...@@ -156,7 +156,7 @@ async def handle_message(message: Message): ...@@ -156,7 +156,7 @@ async def handle_message(message: Message):
text = " ".join(normalize_text_to_infinitive(text, chat_id)) # Убедимся, что это строка text = " ".join(normalize_text_to_infinitive(text, chat_id)) # Убедимся, что это строка
# Проверяем, если это первое сообщение пользователя # Проверяем, если это первое сообщение пользователя
conn = sqlite3.connect('users.db') conn = sqlite3.connect(db_path)
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute('SELECT messages_checked FROM users WHERE id = ?', (user_id,)) cursor.execute('SELECT messages_checked FROM users WHERE id = ?', (user_id,))
......
...@@ -2,8 +2,9 @@ ...@@ -2,8 +2,9 @@
import sqlite3 import sqlite3
import os import os
db_path = os.getenv("DB_PATH")
def create_db(): def create_db():
db_path = os.getenv("DB_PATH")
# Подключаемся к базе данных (если она не существует, она будет создана) # Подключаемся к базе данных (если она не существует, она будет создана)
conn = sqlite3.connect(db_path) conn = sqlite3.connect(db_path)
...@@ -43,7 +44,7 @@ def create_db(): ...@@ -43,7 +44,7 @@ def create_db():
# Функция для добавления пользователя в базу данных # Функция для добавления пользователя в базу данных
def add_user(user_id, first_name, last_name, username): def add_user(user_id, first_name, last_name, username):
conn = sqlite3.connect('users.db') conn = sqlite3.connect(db_path)
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute(''' cursor.execute('''
...@@ -56,7 +57,7 @@ def add_user(user_id, first_name, last_name, username): ...@@ -56,7 +57,7 @@ def add_user(user_id, first_name, last_name, username):
# Функция для проверки, есть ли пользователь в базе данных # Функция для проверки, есть ли пользователь в базе данных
def user_exists(user_id): def user_exists(user_id):
conn = sqlite3.connect('users.db') conn = sqlite3.connect(db_path)
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute('SELECT id FROM users WHERE id = ?', (user_id,)) cursor.execute('SELECT id FROM users WHERE id = ?', (user_id,))
...@@ -67,7 +68,7 @@ def user_exists(user_id): ...@@ -67,7 +68,7 @@ def user_exists(user_id):
# Функция для перемещения пользователя в таблицу забаненных # Функция для перемещения пользователя в таблицу забаненных
def ban_user(user_id, first_name, last_name, username, ban_reason): 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() cursor = conn.cursor()
# Добавляем в таблицу забаненных # Добавляем в таблицу забаненных
...@@ -83,7 +84,7 @@ def ban_user(user_id, first_name, last_name, username, ban_reason): ...@@ -83,7 +84,7 @@ def ban_user(user_id, first_name, last_name, username, ban_reason):
conn.close() conn.close()
def is_user_banned(user_id): def is_user_banned(user_id):
conn = sqlite3.connect('users.db') conn = sqlite3.connect(db_path)
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute('SELECT id FROM banned_users WHERE id = ?', (user_id,)) 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