diff --git a/database.py b/database.py index f85bb71..48ed179 100644 --- a/database.py +++ b/database.py @@ -1,8 +1,10 @@ import sqlite3 import json +DB_PATH = "db_data/users.db" + def init_database(): - conn = sqlite3.connect("/app/db_data/users.db") + conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute(""" CREATE TABLE IF NOT EXISTS users ( @@ -17,11 +19,12 @@ def init_database(): max_context INTEGER ) """) + print("Database initialized") conn.commit() conn.close() def get_user(chat_id: str): - conn = sqlite3.connect("users.db") + conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute("SELECT * FROM users WHERE chat_id = ?", (chat_id,)) user = c.fetchone() @@ -44,7 +47,7 @@ def get_user(chat_id: str): return None def add_user(chat_id: str, user_data): - conn = sqlite3.connect("users.db") + conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute(""" INSERT INTO users ( @@ -66,7 +69,7 @@ def add_user(chat_id: str, user_data): conn.close() def update_user(chat_id: str, user_data): - conn = sqlite3.connect("users.db") + conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute(""" UPDATE users @@ -95,7 +98,7 @@ def update_user(chat_id: str, user_data): conn.close() def get_total_usage(): - conn = sqlite3.connect("users.db") + conn = sqlite3.connect(DB_PATH) c = conn.cursor() c.execute(""" SELECT