import sqlite3

db_name = "ats_doku.db"

def initialize_db():
    connection = sqlite3.connect(db_name)
    cursor = connection.cursor()

    # --- Tabelle 1: Die Liste für dein Dropdown ---
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS ats (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT UNIQUE
        )
    ''')

    # --- Tabelle 2: Das Logbuch (jetzt mit Ort) ---
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS einsaetze (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT,
            datum TEXT,
            zeit INTEGER
        )
    ''')

    # Optional: Ein paar Test-Mitarbeiter anlegen, falls die Tabelle leer ist
    # (Damit du direkt was im Dropdown siehst)
    cursor.execute("SELECT count(*) FROM mitarbeiter")
    if cursor.fetchone()[0] == 0:
        mitarbeiter_liste = [("Tim Grubmüller",), ("Phil Lang",), ("Max Hämmerle",)]
        cursor.executemany("INSERT INTO mitarbeiter (name) VALUES (?)", mitarbeiter_liste)
        print("Test-Mitarbeiter angelegt.")

    connection.commit()
    connection.close()


def add_data_to_db():
    connection = sqlite3.connect(db_name)
    cursor = connection.cursor(" ")