From 585adc19163d234a7b11a664bee20da6d22fa700 Mon Sep 17 00:00:00 2001
From: pluja <codeberg@r3d.red>
Date: Wed, 19 Apr 2023 20:58:59 +0200
Subject: [PATCH] Fix database connection

---
 database.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

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