from aaw.db_components.component_db_connection import run_query
from aaw.db_auth import generate_lerncode

def add_lerncode(material_id, original_text_id):
    # Prüfen, ob bereits ein Lerncode existiert
    check_sql = "SELECT id FROM material_lerncodes WHERE material_id = %s AND original_text_id = %s"
    existing = run_query(check_sql, (material_id, original_text_id), fetch="one")
    
    if existing:
        # Löschen des vorhandenen Lerncodes
        delete_sql = "DELETE FROM material_lerncodes WHERE material_id = %s AND original_text_id = %s"
        run_query(delete_sql, (material_id, original_text_id), commit=True)
        print("Bestehender Lerncode gelöscht.")
    
    # Neuen Lerncode generieren
    new_lerncode = generate_lerncode()
    print("Neuer Lerncode:", new_lerncode)
    
    # Lerncode in die Datenbank einfügen
    insert_sql = "INSERT INTO material_lerncodes (material_id, original_text_id, lerncode) VALUES (%s, %s, %s)"
    try:
        run_query(insert_sql, (material_id, original_text_id, new_lerncode), commit=True)
        print("Lerncode erfolgreich gespeichert!")
    except Exception as e:
        print("Fehler beim Speichern des Lerncodes:", str(e))

# Material ID 2 und Original Text ID 7
add_lerncode(2, 7)