Files
Brancheneinstufung2/list_generator.py
2025-05-27 05:55:59 +00:00

198 lines
12 KiB
Python

import csv
from datetime import datetime
import collections
import os.path
from google.oauth2 import service_account
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
# Importiere die Bibliothek - korrigierter Import basierend auf Dateisystem
from gdoctableapppy.gdoctableapp import gdoctableapp
# --- Konfiguration ---
CSV_FILENAME = 'Namensliste.csv'
GOOGLE_DOC_TITLE = f"Gruppenlisten_{datetime.now().strftime('%Y-%m-%d_%H-%M')}"
TARGET_FOLDER_ID = "18DNQaH9zbcBzwhckJI-4Uah-WXTXg6bg"
EINRICHTUNG_INFO = "Kinderhaus St. Martin Neuching"
FOTODATUM_INFO = "02. - 05.06.2025"
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_services_with_service_account():
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
def generate_tables_with_gdoctableapp(docs_api_service, document_id_to_fill):
# ... (CSV-Lesen, sorted_gruppen_namen, stand_zeit wie in der letzten funktionierenden Version) ...
kinder_nach_gruppen = collections.defaultdict(list)
try:
with open(CSV_FILENAME, mode='r', encoding='utf-8-sig', newline='') as csvfile:
reader = csv.DictReader(csvfile, delimiter=';')
for row in reader:
vorname = row.get('Vorname','').strip(); nachname = row.get('Nachname','').strip(); gruppe_original = row.get('Gruppe','').strip()
if not vorname or not nachname or not gruppe_original: continue
kinder_nach_gruppen[gruppe_original].append({'Nachname': nachname, 'Vorname': vorname})
except Exception as e: print(f"FEHLER CSV-Lesen: {e}"); return False
if not kinder_nach_gruppen: print("Keine CSV-Daten."); 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")
try:
resource_for_lib = {"docsService": docs_api_service, "documentId": document_id_to_fill}
gtable_manager = gdoctableapp(resource_for_lib)
except Exception as e_dta_init:
print(f"Fehler bei der Initialisierung von gdoctableapp: {e_dta_init}"); return False
non_table_requests = []
for i, gruppe_original in enumerate(sorted_gruppen_namen):
kinder_liste = kinder_nach_gruppen[gruppe_original]
anzahl_kinder = len(kinder_liste)
gruppe_display_name = gruppe_original + GRUPPENNAME_SUFFIX
# table_values ist hier unsere Liste von Listen für die Tabellendaten
table_values_for_lib = []
table_values_for_lib.append(["Nachname", "Vorname", "Gruppe"])
for kind in kinder_liste:
table_values_for_lib.append([kind['Nachname'], kind['Vorname'], gruppe_display_name])
try:
print(f"Erstelle Tabelle für Gruppe '{gruppe_original}' mit gdoctableapp...")
# KORREKTER PARAMETERNAME: tableData
# Die Methode `createTable` benötigt entweder `rows` und `columns` (für eine leere Tabelle)
# ODER `tableData` (für eine Tabelle mit Inhalt).
# Wenn tableData gegeben ist, werden rows/columns daraus abgeleitet.
res_table = gtable_manager.createTable(tableData=table_values_for_lib, index=None)
# ^^^^^^^^^
print(f" Tabelle für '{gruppe_original}' erstellt. Antwort: {res_table}")
# Fettformatierung der Kopfzeile (bleibt als Versuch, Methodennamen müssen stimmen)
# Die Fettformatierung ist nach wie vor schwierig, da wir die genaue
# Methode und die Struktur des table_range Objekts für set_text_style_to_range
# aus der Bibliothek nicht 100% kennen.
# Für den Moment lassen wir den Versuch drin, aber mit geringer Erfolgserwartung.
if res_table and "table_object" in res_table and "created_table" in res_table["table_object"]:
created_table_info = res_table["table_object"]["created_table"]
table_start_idx_for_style = created_table_info.get("startIndex")
if table_start_idx_for_style is not None:
# Versuchen wir, ob set_text_style_to_range auf gtable_manager existiert.
# Wenn nicht, wird dies einen AttributeError geben.
if hasattr(gtable_manager, 'set_text_style_to_range'):
header_range_for_style = { "tableStartIndex": table_start_idx_for_style, "rowIndex": 0, "columnIndex": 0, "rowSpan": 1, "colSpan": 3}
text_style_bold = {"bold": True}; fields_bold = "bold"
gtable_manager.set_text_style_to_range(text_style_bold, fields_bold, table_range=header_range_for_style)
print(f" Kopfzeile für Tabelle '{gruppe_original}' versucht fett zu formatieren.")
else:
print(f" Methode 'set_text_style_to_range' nicht auf gtable_manager gefunden. Kopfzeile nicht fett formatiert.")
else: print(f" Konnte startIndex für Formatierung von Tabelle '{gruppe_original}' nicht erhalten.")
else: print(f" Unerwartete Antwort von createTable für Formatierung: {res_table}")
footer_lines_for_group = ["", f"{anzahl_kinder} angemeldete Kinder", "",
"Dies ist die Liste der bereits angemeldeten Kinder. Bitte die Eltern der noch fehlenden",
"Kinder an die Anmeldung erinnern.", "",
f"Stand {stand_zeit}", ""]
full_footer_text_for_group = "\n".join(footer_lines_for_group) + "\n"
# Wenn gdoctableapp eine Methode zum Einfügen von Text hat (z.B. insert_text), verwenden wir die.
# Andernfalls sammeln wir es für einen manuellen Batch.
if hasattr(gtable_manager, 'insert_text'):
if full_footer_text_for_group.strip():
gtable_manager.insert_text(text=full_footer_text_for_group, index=None)
else:
non_table_requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': full_footer_text_for_group}})
except AttributeError as e_attr: # Fängt Fehler ab, wenn Methoden nicht existieren
print(f"AttributeError bei Tabelle/Footer für Gruppe '{gruppe_original}': {e_attr}")
non_table_requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': f"\nATTRIBUTE_ERROR BEI TABELLE {gruppe_original}: {e_attr}\n"}})
except Exception as e_table:
print(f"Allgemeiner Fehler bei Tabelle für Gruppe '{gruppe_original}': {e_table}")
non_table_requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': f"\nFEHLER BEI TABELLE {gruppe_original}\n"}})
if i < len(sorted_gruppen_namen) - 1:
if hasattr(gtable_manager, 'insert_page_break'):
try:
gtable_manager.insert_page_break(index=None)
except Exception as e_pb:
print(f"Fehler bei gtable_manager.insert_page_break nach {gruppe_original}: {e_pb}")
non_table_requests.append({'insertPageBreak': {'endOfSegmentLocation': {}}})
else:
non_table_requests.append({'insertPageBreak': {'endOfSegmentLocation': {}}})
if non_table_requests and docs_api_service:
try:
print(f"Sende gesammelte Text- und PageBreak-Requests für Doc ID '{document_id_to_fill}'...")
docs_api_service.documents().batchUpdate(documentId=document_id_to_fill, body={'requests': non_table_requests}).execute()
print("Text- und PageBreak-Requests erfolgreich ausgeführt.")
except HttpError as err:
print(f"Fehler bei gesammelten Text-/PageBreak-Requests: {err}")
return False
return True
# --- Main execution block ---
# (Bleibt exakt so wie in der letzten Version, die die einmalige Info eingefügt hat
# und dann generate_tables_with_gdoctableapp aufruft)
if __name__ == "__main__":
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
if drive_api_service and TARGET_FOLDER_ID:
file_metadata = {'name': GOOGLE_DOC_TITLE, '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: {e}\n Versuche Fallback...")
if not document_id:
try:
doc = docs_api_service.documents().create(body={'title': GOOGLE_DOC_TITLE}).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 am Anfang
initial_info_lines = ["Info zum Kopieren für Ihre manuelle Kopfzeile:", EINRICHTUNG_INFO, FOTODATUM_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}")
# 3. Dokument mit Tabellen befüllen
success_filling = generate_tables_with_gdoctableapp(docs_api_service, document_id) # Name der Funktion hier wichtig
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: Es wurde versucht, echte Tabellen via gdoctableapppy zu erstellen.")
print(" Bitte Kopfzeilen ggf. manuell fett formatieren.")
else: print("\n--- FEHLER BEIM BEFÜLLEN MIT TABELLEN ---")
else: print("\n--- FEHLGESCHLAGEN: Kein Dokument erstellt ---")