Lacie/xp/database.py

21 lines
517 B
Python

import sqlite3
import os
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
def get_db(lifetime=True):
db_name = "lifetime.db" if lifetime else "annual.db"
db_path = os.path.join(BASE_DIR, db_name)
conn = sqlite3.connect(db_path)
cur = conn.cursor()
cur.execute("""
CREATE TABLE IF NOT EXISTS xp (
user_id TEXT PRIMARY KEY,
xp INTEGER DEFAULT 0,
level INTEGER DEFAULT 0,
last_message INTEGER DEFAULT 0
)
""")
conn.commit()
return conn, cur