Put birthdays, anniversaries and probation dates on the calendar

from datetime import date, timedelta
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, state

SHEET_ID = "your-spreadsheet-id"

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    rows = build("sheets", "v4", credentials=creds, cache_discovery=False).spreadsheets().values().get(
        spreadsheetId=SHEET_ID, range="Staff!A2:D").execute().get("values", [])
    cal = build("calendar", "v3", credentials=creds, cache_discovery=False)
    done, today, added = set(state.get("done") or []), date.today(), 0
    for name, birthday, start, probation in [(r + [""] * 4)[:4] for r in rows]:
        dates = []
        for label, raw, yearly in (("Birthday", birthday, True), ("Work anniversary", start, True), ("Probation ends", probation, False)):
            if raw:
                d = date.fromisoformat(raw)
                d = d.replace(year=today.year if d.replace(year=today.year) >= today else today.year + 1) if yearly else d
                dates.append((f"{label}: {name}", d))
        for title, d in dates:
            key = f"{title}|{d}"
            if today <= d <= today + timedelta(days=30) and key not in done:
                cal.events().insert(calendarId="primary", body={"summary": title, "start": {"date": d.isoformat()},
                                                                "end": {"date": (d + timedelta(days=1)).isoformat()}}).execute()
                done.add(key); added += 1
    state.set("done", list(done))
    return {"added": added}