32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
import os
|
|
from pathlib import Path
|
|
import sqlite3
|
|
|
|
tam_data_dir = os.getenv("TAM_DATA_DIR", "./data")
|
|
|
|
def session():
|
|
data_dir = Path(tam_data_dir)
|
|
if not data_dir.is_dir():
|
|
data_dir.mkdir(parents=True)
|
|
db_file = data_dir / "tam-remote.db"
|
|
conn = sqlite3.connect(db_file)
|
|
cur = conn.cursor()
|
|
return conn, cur
|
|
|
|
def init_db():
|
|
conn, cur = session()
|
|
cur.execute("CREATE TABLE IF NOT EXISTS auth_keys (auth_key TEXT PRIMARY KEY, description TEXT)")
|
|
cur.execute("CREATE TABLE IF NOT EXISTS prefixes (prefix TEXT PRIMARY KEY, color TEXT, weight INTEGER)")
|
|
cur.execute("""CREATE TABLE IF NOT EXISTS tickets (prefix TEXT, t_id INTEGER, first_name TEXT, last_name TEXT,
|
|
phone_number TEXT, pref TEXT, PRIMARY KEY (prefix, t_id))""")
|
|
cur.execute("""CREATE TABLE IF NOT EXISTS baskets (prefix TEXT, b_id INTEGER, description TEXT, donors TEXT,
|
|
winning_ticket INTEGER, PRIMARY KEY (prefix, b_id))""")
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
class RepoTemplate:
|
|
def __init__(self):
|
|
self.conn, self.cur = session()
|
|
def __del__(self):
|
|
self.conn.close()
|