Fix database connection

main
pluja 2023-04-19 20:58:59 +02:00
rodzic 2f2d0cc671
commit 585adc1916
1 zmienionych plików z 8 dodań i 5 usunięć

Wyświetl plik

@ -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