kopia lustrzana https://codeberg.org/pluja/openai-telegram-bot
Fix database connection
rodzic
2f2d0cc671
commit
585adc1916
13
database.py
13
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
|
||||
|
|
Ładowanie…
Reference in New Issue