Files
Brancheneinstufung2/list_generator.py
2025-05-26 19:26:38 +00:00

240 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
# --- Konfiguration ---
CSV_FILENAME = 'Namensliste.csv'
GOOGLE_DOC_TITLE = f"Gruppenlisten Kinderhaus St. Martin Neuching (Service Acc) - {datetime.now().strftime('%Y-%m-%d_%H-%M')}"
TARGET_FOLDER_ID = "18DNQaH9zbcBzwhckJI-4Uah-WXTXg6bg"
EINRICHTUNG = "Kinderhaus St. Martin Neuching"
FOTODATUM = "02. - 05.06.2025"
GRUPPENNAME_SUFFIX = "gruppe"
FOTOGRAF_NAME = "Kinderfotos Erding"
FOTOGRAF_ADRESSE = "Gartenstr. 10 85445 Oberding"
FOTOGRAF_WEB = "www.kinderfotos-erding.de"
FOTOGRAF_TEL = "08122-8470867"
SCOPES = [
'https://www.googleapis.com/auth/documents',
'https://www.googleapis.com/auth/drive.file' # Notwendig für Erstellung im Ordner
# Alternativ: 'https://www.googleapis.com/auth/drive' für vollen Drive-Zugriff
]
SERVICE_ACCOUNT_FILE = 'service_account.json'
# --- Ende Konfiguration ---
def get_services_with_service_account(): # Umbenannt für Klarheit
"""Erstellt Docs und Drive API-Dienste mit einem 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 beim Laden der Service Account Credentials aus '{SERVICE_ACCOUNT_FILE}': {e}")
return None, None
try:
docs_service = build('docs', 'v1', credentials=creds)
print("Google Docs API Service erfolgreich erstellt.")
except Exception as e_docs:
print(f"Konnte Google Docs API Service nicht erstellen: {e_docs}")
try:
# Prüfen, ob Drive Scope vorhanden ist, bevor Drive Service gebaut wird
has_drive_scope = any(s in SCOPES for s in ['https://www.googleapis.com/auth/drive.file', 'https://www.googleapis.com/auth/drive'])
if has_drive_scope:
drive_service = build('drive', 'v3', credentials=creds)
print("Google Drive API Service erfolgreich erstellt.")
else:
print("WARNUNG: Kein Drive Scope in SCOPES definiert. Drive Service wird nicht erstellt.")
print(" Dokumente können nicht in spezifischen Ordnern erstellt werden.")
except Exception as e_drive:
print(f"Konnte Google Drive API Service nicht erstellen: {e_drive}")
return docs_service, drive_service # Gibt beide Services zurück
def create_and_fill_doc(docs_service, drive_service, folder_id, doc_title):
document_id = None
# 1. Dokument erstellen (Logik bleibt wie in der Version, die Drive API bevorzugt)
if drive_service and folder_id:
file_metadata = { 'name': doc_title, 'mimeType': 'application/vnd.google-apps.document', 'parents': [folder_id] }
try:
created_file = drive_service.files().create(body=file_metadata, fields='id').execute()
document_id = created_file.get('id')
print(f"Google Doc via Drive API in Ordner '{folder_id}' erstellt, ID: {document_id}")
except HttpError as err_drive_create:
print(f"Fehler beim Erstellen des Dokuments mit Drive API im Ordner: {err_drive_create}")
if err_drive_create.resp.status == 403: print(" -> Fehlende Scopes oder Drive API nicht aktiv?")
print(" Versuche Fallback...")
except Exception as e_drive_create_general:
print(f"Allg. Fehler bei Drive API Erstellung: {e_drive_create_general}\n Versuche Fallback...")
if not document_id:
if not docs_service: print("FEHLER: Docs API Service nicht verfügbar."); return None
try:
doc = docs_service.documents().create(body={'title': doc_title}).execute()
document_id = doc.get('documentId')
print(f"Google Doc via Docs API (Root des SA) erstellt, ID: {document_id}")
if folder_id: print(f" BITTE manuell in Ordner '{folder_id}' verschieben.")
except Exception as e_docs_create: print(f"Konnte Doc auch nicht mit Docs API erstellen: {e_docs_create}"); return None
if not document_id: print("Konnte keine Doc-ID erhalten."); return None
# 2. Daten aus CSV lesen (bleibt gleich)
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:
# ... (CSV Leselogik wie gehabt) ...
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 beim CSV-Lesen: {e}"); return document_id
if not kinder_nach_gruppen: print("Keine CSV-Daten."); return document_id
for gk in kinder_nach_gruppen: kinder_nach_gruppen[gk].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")
# 3. Dokument befüllen - MIT ECHTEN TABELLEN (Fokus auf Seite 1)
requests = []
current_doc_index = 1 # Start des Dokuments
for i, gruppe_original in enumerate(sorted_gruppen_namen):
# ---- NUR FÜR TESTZWECKE: Bearbeite nur die erste Gruppe ----
if i > 0:
print(f"Überspringe Gruppe '{gruppe_original}' für diesen Testlauf (nur erste Gruppe wird mit Tabelle erstellt).")
# Damit der Footer der ersten Seite korrekt kommt und ein evtl. PageBreak,
# wenn es mehr als eine Gruppe gibt, müssen wir den Rest der Logik hier anpassen.
# Für den reinen Tabellentest ist dieser `break` ausreichend.
break
# ---- ENDE TEST ----
kinder_liste = kinder_nach_gruppen[gruppe_original]
anzahl_kinder = len(kinder_liste)
gruppe_display_name = gruppe_original + GRUPPENNAME_SUFFIX
# --- A. Seiten-Header ---
header_text = (f"{EINRICHTUNG}\t\t\t{FOTOGRAF_NAME}\n{FOTODATUM}\n\n")
requests.append({'insertText': {'location': {'index': current_doc_index}, 'text': header_text}})
current_doc_index += len(header_text)
# --- B. Tabelle erstellen (leer) ---
num_rows = len(kinder_liste) + 1 # +1 für Kopfzeile
num_cols = 3
# Die Tabelle wird an current_doc_index eingefügt.
table_start_index = current_doc_index
requests.append({
'insertTable': {
'location': {'index': table_start_index},
'rows': num_rows,
'columns': num_cols
}
})
# --- C. Tabelleninhalt einfügen ---
# Der Inhalt der ersten Zelle beginnt direkt nach dem Einfügen der Tabelle,
# plus einem Offset für die Tabellenstruktur-Tags.
# Ein häufig funktionierender Offset ist +2 (für TableStart, RowStart).
# Wir nehmen an, dass der `current_doc_index` jetzt auf den Beginn der Tabelle zeigt.
# Der Inhalt der ersten Zelle beginnt bei `table_start_index + 2` (relativ zum Dokumentenanfang)
# oder, anders gesagt, Index 1 *innerhalb* der Tabellenstruktur, die bei table_start_index beginnt.
#
# Die Google Docs API ist so gedacht, dass wenn man Text mit Tabs/Newlines an den
# Startindex einer Zelle sendet, sie den Text in die Tabelle verteilt.
table_content_start_index = table_start_index + 2 # Annahme: Index der ersten Zelle (Beginn des Inhaltsbereichs)
# Erstelle den gesamten Text für die Tabelle
table_lines = []
table_lines.append("Nachname\tVorname\tGruppe") # Kopfzeile
for kind in kinder_liste:
table_lines.append(f"{kind['Nachname']}\t{kind['Vorname']}\t{gruppe_display_name}")
full_table_text_for_cells = "\n".join(table_lines) + "\n" # Wichtig: endet mit \n
requests.append({
'insertText': {
'location': {'index': table_content_start_index},
'text': full_table_text_for_cells
}
})
# --- D. Index aktualisieren und Footer ---
# Den Index nach einer Tabelle korrekt weiterzuführen ist der schwierigste Teil.
# Die Tabelle selbst (Struktur) plus ihr Inhalt haben eine bestimmte Länge.
# Für diesen Testlauf mit nur einer Tabelle ist es weniger kritisch, was danach kommt,
# aber für Folgeseiten wäre es das.
# Wir verwenden hier endOfSegmentLocation für den Footer, um diese Komplexität zu umgehen.
footer_text = (
f"\n{anzahl_kinder} angemeldete Kinder\n\n"
# ... (Rest des Footers wie gehabt) ...
f"Dies ist die Liste der bereits angemeldeten Kinder. Bitte die Eltern der noch fehlenden\nKinder an die Anmeldung erinnern.\n\nStand {stand_zeit}\n\n{FOTOGRAF_NAME}\n{FOTOGRAF_ADRESSE}\n{FOTOGRAF_WEB}\n{FOTOGRAF_TEL}\n"
)
requests.append({'insertText': {'endOfSegmentLocation': {}, 'text': footer_text}})
# --- E. Seitenumbruch (für diesen Test nicht relevant, da wir nach einer Gruppe abbrechen) ---
# if i < len(sorted_gruppen_namen) - 1:
# requests.append({'insertPageBreak': {'endOfSegmentLocation': {}}})
# current_doc_index = 1 # Für die nächste Seite (vereinfachte Annahme, wenn EOS verwendet wird)
# oder muss komplexer berechnet werden.
# Batch-Update ausführen (bleibt gleich)
if requests:
if not docs_service: print("FEHLER: Docs Service nicht da."); return document_id
try:
print(f"Sende Batch Update (Test mit 1. Tabelle) für Doc ID '{document_id}'...")
docs_service.documents().batchUpdate(documentId=document_id, body={'requests': requests}).execute()
print("Dokument (Test mit 1. Tabelle) erfolgreich befüllt.")
except HttpError as err:
print(f"Fehler beim Befüllen (Test mit 1. Tabelle) Doc ID '{document_id}': {err}")
# ... (Fehlerdetails wie gehabt) ...
error_details = "Keine Fehlerdetails im Content."
if err.content:
try: error_details = err.content.decode('utf-8')
except: pass
print(f"Details zum Fehler ({err.resp.status} {err._get_reason()}): {error_details}")
# Wichtig: Gib die Requests aus, um zu sehen, was gesendet wurde
print("\n--- Gesendete Requests für diesen Fehler: ---")
for req_idx, req_content in enumerate(requests):
print(f"Request [{req_idx}]: {req_content}")
print("--- Ende Requests ---")
return document_id
# --- Main execution block ---
if __name__ == "__main__":
print(f"Info: Verwendetes Fotodatum: {FOTODATUM}")
print(f"Info: Gruppennamen werden mit Suffix '{GRUPPENNAME_SUFFIX}' versehen.")
print(f"Info: Zieldokumente sollen in Ordner-ID '{TARGET_FOLDER_ID}' landen.")
docs_api_service, drive_api_service = get_services_with_service_account()
if docs_api_service:
final_doc_id = create_and_fill_doc(
docs_api_service,
drive_api_service,
TARGET_FOLDER_ID,
GOOGLE_DOC_TITLE
)
if final_doc_id:
print(f"\n--- SKRIPT BEENDET (Testlauf erste Tabelle) ---")
print(f"Dokument-ID: {final_doc_id}")
print(f"Link: https://docs.google.com/document/d/{final_doc_id}/edit")
if not drive_api_service or not TARGET_FOLDER_ID:
print("Das Dokument wurde im Root-Verzeichnis des Servicekontos erstellt.")
else:
print("\n--- FEHLGESCHLAGEN (Testlauf erste Tabelle) ---")
else:
print("Konnte Google Docs API Service nicht initialisieren. Skript wird beendet.")