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 sqlite3
|
||||||
import json
|
import json
|
||||||
|
|
||||||
|
DB_PATH = "db_data/users.db"
|
||||||
|
|
||||||
def init_database():
|
def init_database():
|
||||||
conn = sqlite3.connect("/app/db_data/users.db")
|
conn = sqlite3.connect(DB_PATH)
|
||||||
c = conn.cursor()
|
c = conn.cursor()
|
||||||
c.execute("""
|
c.execute("""
|
||||||
CREATE TABLE IF NOT EXISTS users (
|
CREATE TABLE IF NOT EXISTS users (
|
||||||
|
@ -17,11 +19,12 @@ def init_database():
|
||||||
max_context INTEGER
|
max_context INTEGER
|
||||||
)
|
)
|
||||||
""")
|
""")
|
||||||
|
print("Database initialized")
|
||||||
conn.commit()
|
conn.commit()
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
def get_user(chat_id: str):
|
def get_user(chat_id: str):
|
||||||
conn = sqlite3.connect("users.db")
|
conn = sqlite3.connect(DB_PATH)
|
||||||
c = conn.cursor()
|
c = conn.cursor()
|
||||||
c.execute("SELECT * FROM users WHERE chat_id = ?", (chat_id,))
|
c.execute("SELECT * FROM users WHERE chat_id = ?", (chat_id,))
|
||||||
user = c.fetchone()
|
user = c.fetchone()
|
||||||
|
@ -44,7 +47,7 @@ def get_user(chat_id: str):
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def add_user(chat_id: str, user_data):
|
def add_user(chat_id: str, user_data):
|
||||||
conn = sqlite3.connect("users.db")
|
conn = sqlite3.connect(DB_PATH)
|
||||||
c = conn.cursor()
|
c = conn.cursor()
|
||||||
c.execute("""
|
c.execute("""
|
||||||
INSERT INTO users (
|
INSERT INTO users (
|
||||||
|
@ -66,7 +69,7 @@ def add_user(chat_id: str, user_data):
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
def update_user(chat_id: str, user_data):
|
def update_user(chat_id: str, user_data):
|
||||||
conn = sqlite3.connect("users.db")
|
conn = sqlite3.connect(DB_PATH)
|
||||||
c = conn.cursor()
|
c = conn.cursor()
|
||||||
c.execute("""
|
c.execute("""
|
||||||
UPDATE users
|
UPDATE users
|
||||||
|
@ -95,7 +98,7 @@ def update_user(chat_id: str, user_data):
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
def get_total_usage():
|
def get_total_usage():
|
||||||
conn = sqlite3.connect("users.db")
|
conn = sqlite3.connect(DB_PATH)
|
||||||
c = conn.cursor()
|
c = conn.cursor()
|
||||||
c.execute("""
|
c.execute("""
|
||||||
SELECT
|
SELECT
|
||||||
|
|
Ładowanie…
Reference in New Issue