import sqlite3

DATA = "/var/www/html/data.db"

def get_connection():
    conn = sqlite3.connect(DATA, check_same_thread=False)
    conn.row_factory = sqlite3.Row
    return conn

def data_init():
    conn = get_connection()
    cursor = conn.cursor()

    cursor.execute("""
    CREATE TABLE IF NOT EXISTS tg (
        tg_id INTEGER PRIMARY KEY,
        po_id INTEGER,
        language TEXT,
        ui_message_id INTEGER,
        step TEXT,
        link TEXT,
        rang INTEGER,
        streak INTEGER,
        sub_1 TEXT,
        sub_2 TEXT,
        sub_3 TEXT,
        ref_id INTEGER
    )
    """)
    cursor.execute("""
    CREATE TABLE IF NOT EXISTS po (
        po_id INTEGER PRIMARY KEY,
        tg_id INTEGER,
        reg INTEGER NOT NULL DEFAULT 0,
        ftd INTEGER NOT NULL DEFAULT 0,
        sum_dep INTEGER NOT NULL DEFAULT 0
    )
    """)
    conn.commit()
    conn.close