Spaces:
Sleeping
Sleeping
File size: 1,581 Bytes
be16ac4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | import sqlite3
from pathlib import Path
DB_PATH = Path("data/user_data.db")
def get_connection():
DB_PATH.parent.mkdir(parents=True, exist_ok=True)
conn = sqlite3.connect(DB_PATH)
conn.row_factory = sqlite3.Row
return conn
def initialize_db():
conn = get_connection()
cursor = conn.cursor()
cursor.executescript("""
CREATE TABLE IF NOT EXISTS users (
id TEXT PRIMARY KEY,
name TEXT NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);
CREATE TABLE IF NOT EXISTS sessions (
id TEXT PRIMARY KEY,
user_id TEXT NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
FOREIGN KEY (user_id) REFERENCES users(id)
);
""")
conn.commit()
conn.close()
def add_user(user_id, name):
conn = get_connection()
cursor = conn.cursor()
cursor.execute("""INSERT INTO users (id, name) VALUES (?, ?)
ON CONFLICT(id) DO UPDATE SET name = excluded.name""", (user_id, name))
conn.commit()
conn.close()
def create_session(user_id, session_id):
conn = get_connection()
cursor = conn.cursor()
cursor.execute("INSERT INTO sessions (id, user_id) VALUES (?, ?)", (session_id, user_id))
conn.commit()
conn.close()
def user_exists(user_id):
conn = get_connection()
cursor = conn.cursor()
cursor.execute("SELECT 1 FROM users WHERE id = ?", (user_id,))
exists = cursor.fetchone() is not None
conn.close()
return exists
|