Brancheneinstufung2/list_generator.py

import csv
from datetime import datetime
import collections
import os.path
import json 

from google.oauth2 import service_account
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError

# --- Konfiguration ---
GRUPPEN_CSV_FILENAME = 'Namensliste.csv' 
FAMILIEN_CSV_FILENAME = 'familien.csv'   

TARGET_FOLDER_ID = "18DNQaH9zbcBzwhckJI-4Uah-WXTXg6bg" 
LAST_INPUT_FILE = "listen_generator_last_input.json" 

DEFAULT_EINRICHTUNG_ODER_EVENT = "Meine Veranstaltung" 
DEFAULT_FOTODATUM_ODER_EVENTDATUM = "TT.MM.JJJJ - TT.MM.JJJJ" 
DEFAULT_AUSGABEMODUS = "e"
DEFAULT_EINRICHTUNGSTYP = "k" # k für Kindergarten, s für Schule

# Suffix nur für Kindergärten
GRUPPENNAME_SUFFIX = "gruppe" 

SCOPES = [
    'https://www.googleapis.com/auth/documents',
    'https://www.googleapis.com/auth/drive.file' 
]
SERVICE_ACCOUNT_FILE = 'service_account.json'
# --- Ende Konfiguration ---

def get_last_inputs():
    """Lädt die zuletzt verwendeten Eingaben (Typ, Einrichtung/Event, Datum, Modus)."""
    try:
        with open(LAST_INPUT_FILE, 'r') as f:
            data = json.load(f)
            return (data.get("einrichtungstyp", DEFAULT_EINRICHTUNGSTYP),
                    data.get("einrichtung_event", DEFAULT_EINRICHTUNG_ODER_EVENT), 
                    data.get("datum_info", DEFAULT_FOTODATUM_ODER_EVENTDATUM),
                    data.get("ausgabemodus", DEFAULT_AUSGABEMODUS))
    except (FileNotFoundError, json.JSONDecodeError):
        return DEFAULT_EINRICHTUNGSTYP, DEFAULT_EINRICHTUNG_ODER_EVENT, DEFAULT_FOTODATUM_ODER_EVENTDATUM, DEFAULT_AUSGABEMODUS

def save_last_inputs(einrichtungstyp, einrichtung_event, datum_info, ausgabemodus):
    """Speichert die aktuellen Eingaben."""
    try:
        with open(LAST_INPUT_FILE, 'w') as f:
            json.dump({"einrichtungstyp": einrichtungstyp,
                       "einrichtung_event": einrichtung_event, 
                       "datum_info": datum_info, 
                       "ausgabemodus": ausgabemodus}, f)
    except IOError:
        print("Fehler beim Speichern der letzten Eingaben.")

def get_services_with_service_account():
    # ... (unverändert) ...
    creds = None; docs_service = None; drive_service = None
    try: creds = service_account.Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
    except Exception as e: print(f"Fehler Credentials: {e}"); return None, None 
    try: docs_service = build('docs', 'v1', credentials=creds); print("Docs API Service erstellt.")
    except Exception as e: print(f"Fehler Docs Service: {e}")
    try:
        if any(s in SCOPES for s in ['https://www.googleapis.com/auth/drive.file', 'https://www.googleapis.com/auth/drive']):
            drive_service = build('drive', 'v3', credentials=creds); print("Drive API Service erstellt.")
        else: print("WARNUNG: Kein Drive Scope für Drive Service.")
    except Exception as e: print(f"Fehler Drive Service: {e}")
    return docs_service, drive_service

# Die Hilfsfunktion parse_datetime_string bleibt unverändert
def parse_datetime_string(datetime_str):
    try:
        dt_obj = datetime.strptime(datetime_str, "%d.%m.%Y %H:%M")
        return dt_obj, dt_obj.strftime("%d.%m.%Y"), dt_obj.strftime("%H:%M")
    except ValueError:
        return None, None, None

# --- Funktion für Modus "intern" und "extern" (Kita/Schule) ---
def generate_kita_schule_content(docs_service, document_id_to_fill, ausgabemodus_kuerzel, einrichtungstyp_kuerzel):
    # ... (CSV-Lesen bleibt gleich) ...
    kinder_nach_gruppen = collections.defaultdict(list)
    try:
        with open(GRUPPEN_CSV_FILENAME, mode='r', encoding='utf-8-sig', newline='') as csvfile:
            reader = csv.DictReader(csvfile, delimiter=';')
            for row in reader:
                vorname = row.get('Vorname Kind','').strip() or row.get('Vorname','').strip()
                nachname = row.get('Nachname Kind','').strip() or row.get('Nachname','').strip()
                gruppe_original = row.get('Gruppe','').strip() or row.get('Klasse','').strip() # Flexibel auch für "Klasse" Spalte
                einzelfotos_wert = row.get('Einzelfotos', 'Nein').strip()
                gruppenfotos_wert = row.get('Gruppenfotos', 'Nein').strip()
                if not vorname or not nachname or not gruppe_original: continue
                einzelfotos_display = "✓" if einzelfotos_wert.lower() == 'ja' else "Nein"
                gruppenfotos_display = "✓" if gruppenfotos_wert.lower() == 'ja' else "Nein"
                kinder_nach_gruppen[gruppe_original].append({
                    'Nachname': nachname, 'Vorname': vorname,
                    'Einzelfotos': einzelfotos_display, 'Gruppenfotos': gruppenfotos_display })
    except FileNotFoundError: print(f"FEHLER: '{GRUPPEN_CSV_FILENAME}' nicht gefunden."); return False
    except Exception as e: print(f"FEHLER CSV-Lesen ('{GRUPPEN_CSV_FILENAME}'): {e}"); return False
    if not kinder_nach_gruppen: print(f"Keine Daten aus '{GRUPPEN_CSV_FILENAME}'."); return False
    
    for gk_key in kinder_nach_gruppen:
        kinder_nach_gruppen[gk_key].sort(key=lambda x: (x['Nachname'].lower(), x['Vorname'].lower()))
    sorted_gruppen_namen = sorted(kinder_nach_gruppen.keys())
    stand_zeit = datetime.now().strftime("%d.%m.%Y %H:%M Uhr") 
    requests = []
    col_width_nachname = 22; col_width_vorname = 22; col_width_e = 3; col_width_g = 3

    # --- ANPASSUNG: Texte basierend auf Einrichtungstyp ---
    gruppen_spalten_header = "Klasse" if einrichtungstyp_kuerzel == 's' else "Gruppe"
    erinnerungstext1 = "Dies ist die Liste der bereits angemeldeten Schüler. Bitte die noch fehlenden"
    erinnerungstext2 = "Schüler an die Anmeldung erinnern."
    if einrichtungstyp_kuerzel == 'k': # Kindergarten
        erinnerungstext1 = "Dies ist die Liste der bereits angemeldeten Kinder. Bitte die Eltern der noch fehlenden"
        erinnerungstext2 = "Kinder an die Anmeldung erinnern."
    # --- ENDE ANPASSUNG ---

    for i, gruppe_original in enumerate(sorted_gruppen_namen):
        kinder_liste = kinder_nach_gruppen[gruppe_original]; anzahl_kinder = len(kinder_liste)
        
        # --- ANPASSUNG: Suffix nur für Kindergarten ---
        gruppe_display_name = gruppe_original
        if einrichtungstyp_kuerzel == 'k':
            gruppe_display_name += GRUPPENNAME_SUFFIX
        # --- ENDE ANPASSUNG ---
        
        group_block_lines = ["", ""] 
        kopf_nn = "Nachname".ljust(col_width_nachname); kopf_vn = "Vorname".ljust(col_width_vorname)
        if ausgabemodus_kuerzel == "i":
            kopf_e = "E".ljust(col_width_e); kopf_g = "G".ljust(col_width_g)
            group_block_lines.append(f"{kopf_nn}\t{kopf_vn}\t{kopf_e}\t{kopf_g}\t{gruppen_spalten_header}")
        else:
            group_block_lines.append(f"{kopf_nn}\t{kopf_vn}\t{gruppen_spalten_header}")
        for kind in kinder_liste:
            nn_pad = kind['Nachname'].ljust(col_width_nachname); vn_pad = kind['Vorname'].ljust(col_width_vorname)
            if ausgabemodus_kuerzel == "i":
                e_disp = kind['Einzelfotos'].ljust(col_width_e); g_disp = kind['Gruppenfotos'].ljust(col_width_g)
                group_block_lines.append(f"{nn_pad}\t{vn_pad}\t{e_disp}\t{g_disp}\t{gruppe_display_name}")
            else:
                group_block_lines.append(f"{nn_pad}\t{vn_pad}\t{gruppe_display_name}")
        
        group_block_lines.extend(["", f"{anzahl_kinder} angemeldete Kinder", "", 
                                  erinnerungstext1, # Angepasster Text
                                  erinnerungstext2, # Angepasster Text
                                  "", f"Stand {stand_zeit}", ""])
        
        full_group_block_text = "\n".join(group_block_lines) + "\n" 
        requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': full_group_block_text}})
        if i < len(sorted_gruppen_namen) - 1:
            requests.append({'insertPageBreak': {'endOfSegmentLocation': {}}})
    if requests:
        try:
            print(f"Sende Batch Update (Modus: {ausgabemodus_kuerzel}) für Doc ID '{document_id_to_fill}'...")
            docs_service.documents().batchUpdate(documentId=document_id_to_fill, body={'requests': requests}).execute()
            print(f"Dokument erfolgreich im Modus '{ausgabemodus_kuerzel}' befüllt.")
            return True
        except HttpError as err: print(f"Fehler beim Befüllen: {err}"); return False
    return True

# --- Funktion für Modus "familie" (bleibt unverändert) ---
def generate_familien_content(docs_service, document_id_to_fill):
    # ... (Code für diese Funktion bleibt exakt wie in der letzten Version) ...
    # (Ich füge ihn hier zur Vollständigkeit ein, aber er wurde nicht geändert)
    termine_nach_tagen = collections.defaultdict(list)
    try:
        with open(FAMILIEN_CSV_FILENAME, mode='r', encoding='utf-8-sig', newline='') as csvfile: 
            reader = csv.DictReader(csvfile, delimiter=';')
            for row_idx, row in enumerate(reader):
                vorname_kind = row.get('Invitee First Name','').strip()
                nachname_kind = row.get('Invitee Last Name','').strip()
                start_datetime_str = row.get('Start Date & Time', '').strip()
                end_datetime_str = row.get('End Date & Time', '').strip()
                if not (vorname_kind and nachname_kind and start_datetime_str and end_datetime_str): continue
                start_dt_obj, datum_str, startzeit_str = parse_datetime_string(start_datetime_str)
                _, _, endzeit_str_val = parse_datetime_string(end_datetime_str)
                if not start_dt_obj: continue
                kinder_anzahl_str = row.get('Response 1', 'k.A.').strip()
                pub_antwort = row.get('Response 2', '').strip()
                pub_display = "X" if pub_antwort.lower() == 'ja, gerne' else ""
                termine_nach_tagen[datum_str].append({
                    'Vorname': vorname_kind, 'Nachname': nachname_kind, 'StartzeitObj': start_dt_obj, 
                    'Startzeit': startzeit_str, 'Endzeit': endzeit_str_val, 'Kinder': kinder_anzahl_str, 
                    'Pub': pub_display, 'Erledigt': "☐" })
    except FileNotFoundError: print(f"FEHLER: '{FAMILIEN_CSV_FILENAME}' nicht gefunden."); return False
    except Exception as e: print(f"FEHLER CSV-Lesen ('{FAMILIEN_CSV_FILENAME}'): {e}"); return False
    if not termine_nach_tagen: print(f"Keine Daten aus '{FAMILIEN_CSV_FILENAME}'."); return False
    sorted_tage = sorted(termine_nach_tagen.keys(), key=lambda d: datetime.strptime(d, "%d.%m.%Y"))
    requests = []
    col_w_vn=20; col_w_nn=20; col_w_zeit=7; col_w_kinder=10; col_w_pub=5; col_w_erl=8
    for tag_idx, tag_datum_str in enumerate(sorted_tage):
        tages_termine = sorted(termine_nach_tagen[tag_datum_str], key=lambda t: t['StartzeitObj'])
        if tag_idx > 0: requests.append({'insertPageBreak': {'endOfSegmentLocation': {}}})
        tages_block_lines = [f"Termine am {tag_datum_str}", "", ""]
        kopf_vn="Vorname".ljust(col_w_vn); kopf_nn="Nachname".ljust(col_w_nn); kopf_zeit="Uhrzeit".ljust(col_w_zeit)
        kopf_kinder="# Kinder".ljust(col_w_kinder); kopf_pub="Pub".ljust(col_w_pub); kopf_erl="Erledigt".ljust(col_w_erl)
        tages_block_lines.append(f"{kopf_vn}\t{kopf_nn}\t{kopf_zeit}\t{kopf_kinder}\t{kopf_pub}\t{kopf_erl}")
        letzte_endzeit_obj = None
        for termin_idx, termin in enumerate(tages_termine):
            if letzte_endzeit_obj and termin['StartzeitObj'] > letzte_endzeit_obj:
                leere_zeile_parts = [" ".ljust(w) for w in [col_w_vn,col_w_nn,col_w_zeit,col_w_kinder,col_w_pub,col_w_erl]]
                tages_block_lines.append("\t".join(leere_zeile_parts))
            vn_pad=termin['Vorname'].ljust(col_w_vn); nn_pad=termin['Nachname'].ljust(col_w_nn); zeit_pad=termin['Startzeit'].ljust(col_w_zeit)
            kind_pad=termin['Kinder'].ljust(col_w_kinder); pub_pad=termin['Pub'].ljust(col_w_pub); erl_pad=termin['Erledigt'].ljust(col_w_erl)
            tages_block_lines.append(f"{vn_pad}\t{nn_pad}\t{zeit_pad}\t{kind_pad}\t{pub_pad}\t{erl_pad}")
            _, _, termin_endzeit_str_fuer_obj = parse_datetime_string(f"{tag_datum_str} {termin['Endzeit']}")
            if termin_endzeit_str_fuer_obj: letzte_endzeit_obj = datetime.strptime(f"{tag_datum_str} {termin['Endzeit']}", "%d.%m.%Y %H:%M")
            else: letzte_endzeit_obj = termin['StartzeitObj'] + timedelta(minutes=6)
        full_tages_block_text = "\n".join(tages_block_lines) + "\n\n"
        requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': full_tages_block_text}})
    if requests:
        try:
            print(f"Sende Batch Update (Familien Modus) für Doc ID '{document_id_to_fill}'...")
            docs_service.documents().batchUpdate(documentId=document_id_to_fill, body={'requests': requests}).execute()
            print("Dokument erfolgreich im Familien Modus befüllt.")
            return True
        except HttpError as err: print(f"Fehler beim Befüllen (Familien Modus): {err}"); return False
    return True

# --- Main execution block ---
if __name__ == "__main__":
    last_einrichtungstyp, last_einrichtung_event, last_datum_info, last_ausgabemodus = get_last_inputs()

    print("\nBitte geben Sie die Details ein:")
    
    # --- ANPASSUNG: Neue Abfrage für Einrichtungstyp ---
    while True:
        user_einrichtungstyp_input = input(f"Einrichtungstyp (k=Kindergarten, s=Schule) [{last_einrichtungstyp}]: ").lower() or last_einrichtungstyp
        if user_einrichtungstyp_input in ["k", "s", "kindergarten", "schule"]:
            user_einrichtungstyp_kuerzel = user_einrichtungstyp_input[0] # Nimm immer den ersten Buchstaben
            break
        print("Ungültige Eingabe. Bitte 'k' oder 's' wählen.")
    # --- ENDE ANPASSUNG ---

    user_einrichtung_event = input(f"Name der Einrichtung/Veranstaltung [{last_einrichtung_event}]: ") or last_einrichtung_event
    user_datum_info = input(f"Allgemeines Datum (für Info-Block) [{last_datum_info}]: ") or last_datum_info
    
    # Die Modusabfrage (intern/extern/familie) bleibt gleich
    while True:
        user_ausgabemodus_input = input(f"Ausgabemodus (i=intern, e=extern, f=familie) [{last_ausgabemodus}]: ").lower() or last_ausgabemodus
        if user_ausgabemodus_input in ["i", "e", "f", "intern", "extern", "familie"]: 
            if user_ausgabemodus_input.startswith("i"): user_ausgabemodus_kuerzel = "i"
            elif user_ausgabemodus_input.startswith("e"): user_ausgabemodus_kuerzel = "e"
            elif user_ausgabemodus_input.startswith("f"): user_ausgabemodus_kuerzel = "f"
            else: user_ausgabemodus_kuerzel = user_ausgabemodus_input
            break
        print("Ungültige Eingabe. Bitte 'i', 'e' oder 'f' wählen.")

    # Speichere alle aktuellen Eingaben
    save_last_inputs(user_einrichtungstyp_kuerzel, user_einrichtung_event, user_datum_info, user_ausgabemodus_kuerzel) 

    GOOGLE_DOC_TITLE_DYNAMIC = f"Listen_{user_einrichtung_event.replace(' ', '_').replace(',', '')}_{user_ausgabemodus_kuerzel}_{datetime.now().strftime('%Y-%m-%d_%H-%M')}"

    print(f"\nInfo: Erstelle Dokument '{GOOGLE_DOC_TITLE_DYNAMIC}' im Modus '{user_ausgabemodus_kuerzel}' für Typ '{user_einrichtungstyp_kuerzel}'")
    
    # ... (Rest des main-Blocks bleibt gleich: Services holen, leeres Dokument erstellen, einmalige Info einfügen) ...
    docs_api_service, drive_api_service = get_services_with_service_account()
    if not docs_api_service: print("Konnte Google Docs API Service nicht initialisieren. Skript wird beendet.")
    else:
        document_id = None
        if drive_api_service and TARGET_FOLDER_ID:
            file_metadata = {'name': GOOGLE_DOC_TITLE_DYNAMIC, 'mimeType': 'application/vnd.google-apps.document', 'parents': [TARGET_FOLDER_ID]}
            try:
                created_file = drive_api_service.files().create(body=file_metadata, fields='id').execute()
                document_id = created_file.get('id'); print(f"Doc in Ordner '{TARGET_FOLDER_ID}' erstellt, ID: {document_id}")
            except Exception as e: print(f"Fehler Drive Erstellung: {e}\n Fallback...")
        if not document_id:
            try:
                doc = docs_api_service.documents().create(body={'title': GOOGLE_DOC_TITLE_DYNAMIC}).execute()
                document_id = doc.get('documentId'); print(f"Doc in Root SA erstellt, ID: {document_id}")
                if TARGET_FOLDER_ID: print(f" Bitte manuell in Ordner '{TARGET_FOLDER_ID}' verschieben.")
            except Exception as e: print(f"Fehler Docs Erstellung: {e}")
        if document_id:
            initial_info_lines = [ "Info zum Kopieren für Ihre manuelle Kopf-/Fußzeile:", user_einrichtung_event, user_datum_info, "\n" + "="*70 + "\n" ]
            initial_text = "\n".join(initial_info_lines) + "\n" 
            initial_requests = [{'insertText': {'location': {'index': 1}, 'text': initial_text}}]
            try: docs_api_service.documents().batchUpdate(documentId=document_id, body={'requests': initial_requests}).execute(); print("Einmalige Info eingefügt.")
            except HttpError as err: print(f"Fehler bei einmaliger Info: {err}")
            
            success_filling = False
            # --- ANPASSUNG: Logik-Weiche ruft jetzt kita/schule Funktion auf ---
            if user_ausgabemodus_kuerzel in ["i", "e"]:
                print(f"Starte Kita/Schule-Modus ('{user_ausgabemodus_kuerzel}')...")
                success_filling = generate_kita_schule_content(docs_api_service, document_id, user_ausgabemodus_kuerzel, user_einrichtungstyp_kuerzel)
            elif user_ausgabemodus_kuerzel == "f":
                print("Starte Familien-Modus...")
                success_filling = generate_familien_content(docs_api_service, document_id)
            else: print(f"FEHLER: Unbekannter Ausgabemodus '{user_ausgabemodus_kuerzel}'")
            # --- ENDE ANPASSUNG ---

            if success_filling:
                 print(f"\n--- SKRIPT BEENDET ---"); print(f"Dokument-ID: {document_id}"); print(f"Link: https://docs.google.com/document/d/{document_id}/edit")
                 print("HINWEIS: Text markieren und 'Einfügen > Tabelle > Tabelle aus Text erstellen...' verwenden.")
                 print("         Kopfzeilen manuell fett formatieren.")
            else: print("\n--- FEHLER BEIM BEFÜLLEN MIT DATEN ---")
        else: print("\n--- FEHLGESCHLAGEN: Kein Dokument erstellt ---")