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,))
...@@ -314,4 +314,4 @@ async def main(): ...@@ -314,4 +314,4 @@ async def main():
if __name__ == "__main__": if __name__ == "__main__":
create_db() # Создание базы данных и таблиц create_db() # Создание базы данных и таблиц
asyncio.run(main()) asyncio.run(main())
\ No newline at end of file
...@@ -2,9 +2,10 @@ ...@@ -2,9 +2,10 @@
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)
cursor = conn.cursor() cursor = conn.cursor()
...@@ -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,11 +84,11 @@ def ban_user(user_id, first_name, last_name, username, ban_reason): ...@@ -83,11 +84,11 @@ 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,))
result = cursor.fetchone() result = cursor.fetchone()
conn.close() conn.close()
return result is not None return result is not None
\ No newline at end of file
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