import db_helper as db

DB_PATH = "action_units.db"

def setup_test_db():
    # 1. Verbindung herstellen (erstellt die Datei, falls nicht vorhanden)
    conn, cursor = db.connect_db(DB_PATH)

    # 2. Spalten definieren
    # Wir erstellen eine Spalte für den Zeitstempel und beispielhaft einige AUs.
    # In SQLite können wir später mit deinem Helper weitere Spalten hinzufügen.
    columns = {
        "timestamp": "TEXT",
        "AU01": "REAL",
        "AU02": "REAL",
        "AU04": "REAL",
        "AU05": "REAL",
        "AU06": "REAL",
        "AU07": "REAL",
        "AU09": "REAL",
        "AU10": "REAL",
        "AU11": "REAL",
        "AU12": "REAL",
        "AU14": "REAL",
        "AU15": "REAL",
        "AU17": "REAL",
        "AU20": "REAL",
        "AU23": "REAL",
        "AU24": "REAL",
        "AU25": "REAL",
        "AU26": "REAL",
        "AU28": "REAL",
        "AU43": "REAL",
    }
    
    # Constraints (z.B. Zeitstempel darf nicht leer sein)
    constraints = {
        "timestamp": ["NOT NULL"]
    }
    
    # Primärschlüssel definieren (Kombination aus Zeitstempel und ggf. ID)
    primary_key = {"pk_timestamp": ["timestamp"]}

    try:
        sql = db.create_table(conn, cursor, "actionUnits", columns, constraints, primary_key)
        print("Tabelle erfolgreich erstellt!")
        print(f"SQL-Befehl:\n{sql}")
    except Exception as e:
        print(f"Hinweis: {e}") 
    finally:
        db.disconnect_db(conn, cursor)

if __name__ == "__main__":
    setup_test_db()