From c9ae3d733ab66ee31f275969ee2aaf52b7f19f8c Mon Sep 17 00:00:00 2001 From: Floke Date: Wed, 28 May 2025 07:58:14 +0000 Subject: [PATCH] feat: Add user input for date/facility, save inputs, adjust layout - Prompts user for event date and facility name at script start. - Saves last entered values to a JSON file for future default. - Uses user-provided values for the initial info block on page 1. - Adds two line breaks before each group's data block for spacing. - Document title now includes the facility name for better identification. --- list_generator.py | 102 +++++++++++++++++++++++++++++----------------- 1 file changed, 65 insertions(+), 37 deletions(-) diff --git a/list_generator.py b/list_generator.py index 7ae88a80..4f89abdf 100644 --- a/list_generator.py +++ b/list_generator.py @@ -2,6 +2,7 @@ import csv from datetime import datetime import collections import os.path +import json # Für das Speichern/Laden der letzten Eingaben from google.oauth2 import service_account from googleapiclient.discovery import build @@ -9,12 +10,13 @@ from googleapiclient.errors import HttpError # --- Konfiguration --- CSV_FILENAME = 'Namensliste.csv' -GOOGLE_DOC_TITLE = f"Gruppenlisten_{datetime.now().strftime('%Y-%m-%d_%H-%M')}" # Einfacherer Titel +# GOOGLE_DOC_TITLE wird jetzt dynamischer mit dem Einrichtungsnamen TARGET_FOLDER_ID = "18DNQaH9zbcBzwhckJI-4Uah-WXTXg6bg" +LAST_INPUT_FILE = "last_input.json" # Datei zum Speichern der letzten Eingaben -# Diese werden nur für den einmaligen Infoblock verwendet -EINRICHTUNG_INFO = "Kinderhaus St. Martin Neuching" -FOTODATUM_INFO = "02. - 05.06.2025" +# Standardwerte, falls keine gespeicherten Werte vorhanden sind +DEFAULT_EINRICHTUNG = "Kinderhaus St. Martin Neuching" +DEFAULT_FOTODATUM = "02. - 05.06.2025" GRUPPENNAME_SUFFIX = "gruppe" @@ -25,7 +27,25 @@ SCOPES = [ SERVICE_ACCOUNT_FILE = 'service_account.json' # --- Ende Konfiguration --- +def get_last_inputs(): + """Lädt die zuletzt verwendeten Eingaben aus einer JSON-Datei.""" + try: + with open(LAST_INPUT_FILE, 'r') as f: + data = json.load(f) + return data.get("einrichtung", DEFAULT_EINRICHTUNG), data.get("fotodatum", DEFAULT_FOTODATUM) + except (FileNotFoundError, json.JSONDecodeError): + return DEFAULT_EINRICHTUNG, DEFAULT_FOTODATUM + +def save_last_inputs(einrichtung, fotodatum): + """Speichert die aktuellen Eingaben in einer JSON-Datei.""" + try: + with open(LAST_INPUT_FILE, 'w') as f: + json.dump({"einrichtung": einrichtung, "fotodatum": fotodatum}, f) + except IOError: + print("Fehler beim Speichern der letzten Eingaben.") + def get_services_with_service_account(): + # ... (Diese Funktion bleibt 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 @@ -39,11 +59,7 @@ def get_services_with_service_account(): return docs_service, drive_service def generate_group_data_for_doc(docs_service, document_id_to_fill): - """ - Befüllt ein existierendes Dokument mit: - - Pro Gruppe: Tabellenüberschrift, Kinderliste, Anzahl Kinder, Erinnerungstext, Stand. - """ - + # ... (CSV-Lesen und initiale Datenaufbereitung bleiben unverändert) ... kinder_nach_gruppen = collections.defaultdict(list) try: with open(CSV_FILENAME, mode='r', encoding='utf-8-sig', newline='') as csvfile: @@ -58,8 +74,6 @@ def generate_group_data_for_doc(docs_service, document_id_to_fill): 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()) - - # Das Stand-Datum wird einmalig ermittelt und für alle Gruppen verwendet stand_zeit = datetime.now().strftime("%d.%m.%Y %H:%M Uhr") requests = [] @@ -68,44 +82,41 @@ def generate_group_data_for_doc(docs_service, document_id_to_fill): for i, gruppe_original in enumerate(sorted_gruppen_namen): kinder_liste = kinder_nach_gruppen[gruppe_original] - anzahl_kinder = len(kinder_liste) # Anzahl Kinder für die aktuelle Gruppe + anzahl_kinder = len(kinder_liste) gruppe_display_name = gruppe_original + GRUPPENNAME_SUFFIX group_block_lines = [] - # "Tabellen"-Daten für die aktuelle Gruppe + # --- ANPASSUNG: Zwei Zeilenumbrüche vor der "Tabelle" --- + group_block_lines.append("") # Erster zusätzlicher Zeilenumbruch + group_block_lines.append("") # Zweiter zusätzlicher Zeilenumbruch + kopf_nachname = "Nachname".ljust(col_width_nachname) kopf_vorname = "Vorname".ljust(col_width_vorname) - group_block_lines.append(f"{kopf_nachname}\t{kopf_vorname}\tGruppe") # Überschriften + group_block_lines.append(f"{kopf_nachname}\t{kopf_vorname}\tGruppe") for kind in kinder_liste: nachname_gepadded = kind['Nachname'].ljust(col_width_nachname) vorname_gepadded = kind['Vorname'].ljust(col_width_vorname) group_block_lines.append(f"{nachname_gepadded}\t{vorname_gepadded}\t{gruppe_display_name}") - group_block_lines.append("") # Leerzeile nach der Kinderliste - - # --- HIER DIE GEWÜNSCHTEN INFOS WIEDER HINZUFÜGEN --- + group_block_lines.append("") group_block_lines.append(f"{anzahl_kinder} angemeldete Kinder") - group_block_lines.append("") # Leerzeile + group_block_lines.append("") group_block_lines.append("Dies ist die Liste der bereits angemeldeten Kinder. Bitte die Eltern der noch fehlenden") group_block_lines.append("Kinder an die Anmeldung erinnern.") - group_block_lines.append("") # Leerzeile - group_block_lines.append(f"Stand {stand_zeit}") # Verwendet das einmalig ermittelte Stand-Datum - group_block_lines.append("") # Weitere Leerzeile für Abstand - # --- ENDE HINZUGEFÜGTE INFOS --- + group_block_lines.append("") + group_block_lines.append(f"Stand {stand_zeit}") + group_block_lines.append("") full_group_block_text = "\n".join(group_block_lines) + "\n" - if i == 0: # Erster Block nach der einmaligen Info (die in main eingefügt wurde) - requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': full_group_block_text}}) - else: - requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': full_group_block_text}}) + requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': full_group_block_text}}) if i < len(sorted_gruppen_namen) - 1: requests.append({'insertPageBreak': {'endOfSegmentLocation': {}}}) - # ... (Rest der Funktion: Batch-Update ausführen, bleibt gleich) ... if requests: + # ... (Rest der Funktion: Batch-Update ausführen, bleibt unverändert) ... if not docs_service: print("FEHLER: Docs Service nicht da für Befüllung."); return False try: print(f"Sende Batch Update (Gruppenlisten, Anzahl, Stand) für Doc ID '{document_id_to_fill}'...") @@ -119,38 +130,54 @@ def generate_group_data_for_doc(docs_service, document_id_to_fill): return True # --- Main execution block --- -# (Der Main Block bleibt exakt so wie in der letzten Version, die die einmalige Info eingefügt hat) if __name__ == "__main__": + # 1. Zuletzt verwendete Eingaben laden oder Standardwerte verwenden + last_einrichtung, last_fotodatum = get_last_inputs() + + # 2. Benutzer nach Einrichtung und Datum fragen, mit Vorschlägen + print("\nBitte geben Sie die Details für die Gruppenlisten ein:") + user_einrichtung = input(f"Name der Einrichtung [{last_einrichtung}]: ") or last_einrichtung + user_fotodatum = input(f"Datum der Veranstaltung (z.B. TT.MM.JJJJ - TT.MM.JJJJ) [{last_fotodatum}]: ") or last_fotodatum + + # 3. Aktuelle Eingaben für den nächsten Lauf speichern + save_last_inputs(user_einrichtung, user_fotodatum) + + # Dynamischen Dokumenttitel erstellen + GOOGLE_DOC_TITLE_DYNAMIC = f"Gruppenlisten {user_einrichtung.replace(' ', '_')}_{datetime.now().strftime('%Y-%m-%d_%H-%M')}" + + + print(f"\nInfo: Erstelle Dokument '{GOOGLE_DOC_TITLE_DYNAMIC}'") print(f"Info: Ordner-ID für Dokument: '{TARGET_FOLDER_ID}'") + 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 - # 1. Dokument erstellen (leer oder nur mit Titel) + # Dokument erstellen if drive_api_service and TARGET_FOLDER_ID: - file_metadata = {'name': GOOGLE_DOC_TITLE, 'mimeType': 'application/vnd.google-apps.document', 'parents': [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"Neues leeres Doc via Drive API in Ordner '{TARGET_FOLDER_ID}' erstellt, ID: {document_id}") except Exception as e: print(f"Fehler Drive API Erstellung für leeres Doc: {e}\n Versuche Fallback...") - if not document_id: # Fallback + if not document_id: try: - doc = docs_api_service.documents().create(body={'title': GOOGLE_DOC_TITLE}).execute() + doc = docs_api_service.documents().create(body={'title': GOOGLE_DOC_TITLE_DYNAMIC}).execute() document_id = doc.get('documentId') print(f"Neues leeres Doc via Docs API (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"Konnte leeres Doc auch nicht mit Docs API erstellen: {e}") if document_id: - # 2. Einmalige Info (Kita, Datum) GANZ AM ANFANG des Dokuments einfügen + # Einmalige Info (Kita, Datum aus Benutzereingabe) GANZ AM ANFANG einfügen initial_info_lines = [ "Info zum Kopieren für Ihre manuelle Kopfzeile:", - EINRICHTUNG_INFO, # Verwendet die Konstanten - FOTODATUM_INFO, # Verwendet die Konstanten + user_einrichtung, # Verwendet die Benutzereingabe + user_fotodatum, # Verwendet die Benutzereingabe "\n" + "="*70 + "\n" ] initial_text = "\n".join(initial_info_lines) + "\n" @@ -162,13 +189,14 @@ if __name__ == "__main__": except HttpError as err: print(f"Fehler beim Einfügen der einmaligen Info: {err}") - # 3. Dokument mit gruppenspezifischen Daten befüllen - success_filling = generate_group_data_for_doc(docs_api_service, document_id) # Funktionsname angepasst + # Dokument mit gruppenspezifischen Daten befüllen + success_filling = generate_group_data_for_doc(docs_api_service, document_id) 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") + # ... (Restliche Hinweise bleiben gleich) ... print("HINWEIS: Die Kinderlisten wurden als tabulatorgetrennter Text eingefügt.") print(" Markieren Sie den Text in Google Docs und verwenden Sie 'Einfügen > Tabelle > Tabelle aus Text erstellen...'") print(" Formatieren Sie die Überschriften 'Nachname, Vorname, Gruppe' danach manuell fett.")