Track focus sessions from Telegram in a sheet

from datetime import datetime, timezone
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_event, credentials, http, state

SHEET_ID = "your-spreadsheet-id"

@on_event
def handle(event):
    msg = (event.json or {}).get("message") or {}
    text, chat = (msg.get("text") or "").strip(), msg.get("chat", {}).get("id")
    api = f"https://api.telegram.org/bot{credentials.secret('TELEGRAM_BOT_TOKEN')}"
    now = datetime.now(timezone.utc)
    if text.startswith("/start"):
        state.set("open", {"task": text[6:].strip() or "focus", "at": now.isoformat()})
        reply = "Started."
    elif text.startswith("/stop") and state.get("open"):
        s = state.get("open")
        mins = round((now - datetime.fromisoformat(s["at"])).total_seconds() / 60)
        sheets = build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).spreadsheets()
        sheets.values().append(spreadsheetId=SHEET_ID, range="Sessions!A:C", valueInputOption="USER_ENTERED",
                               body={"values": [[now.date().isoformat(), s["task"], mins]]}).execute()
        rows = sheets.values().get(spreadsheetId=SHEET_ID, range="Sessions!A:C").execute().get("values", [])
        total = sum(int(r[2]) for r in rows if len(r) > 2 and r[0] == now.date().isoformat() and r[2].isdigit())
        state.delete("open")
        reply = f"{s['task']}: {mins} min. Today: {total} min."
    else:
        reply = "Send /start <task> or /stop."
    http.post(f"{api}/sendMessage", json={"chat_id": chat, "text": reply})
    return {"reply": reply}