Log daily Stripe revenue in a sheet with a running forecast

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from datetime import datetime, timedelta, timezone
from collections import defaultdict
import stripe
from clickruntime import on_trigger, credentials, ai, notify

SHEET_ID = "your-sheet-id"

@on_trigger
def run(event):
    stripe.api_key = credentials.secret("STRIPE_API_KEY")
    end = datetime.now(timezone.utc).replace(hour=0, minute=0, second=0, microsecond=0)
    day = (end - timedelta(days=1)).date().isoformat()
    t = defaultdict(lambda: {"gross": 0, "fees": 0, "refunds": 0, "net": 0})
    for b in stripe.BalanceTransaction.list(created={"gte": int((end - timedelta(days=1)).timestamp()), "lt": int(end.timestamp())}, limit=100).auto_paging_iter():
        row = t[b.currency.upper()]
        row["refunds" if b.type == "refund" else "gross"] += b.amount
        row["fees"] += b.fee
        row["net"] += b.net
    rows = [[day, cur] + [v[k] / 100 for k in ("gross", "fees", "refunds", "net")] for cur, v in t.items()]
    sheets = build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).spreadsheets()
    if rows:
        sheets.values().append(spreadsheetId=SHEET_ID, range="Revenue!A:F", valueInputOption="USER_ENTERED", body={"values": rows}).execute()
    history = sheets.values().get(spreadsheetId=SHEET_ID, range="Revenue!A:F").execute().get("values", [])[-120:]
    forecast = ai.generate("From these daily rows (date, currency, gross, fees, refunds, net), forecast net revenue for the next 30 days per currency in 3 sentences:\n"
                           + "\n".join(",".join(map(str, r)) for r in history), max_output_tokens=200).text
    notify.email_me(f"Revenue for {day} and 30-day forecast", "\n".join(f"{r[1]}: net {r[5]:,.2f}" for r in rows) + "\n\n" + forecast)
    return {"rows": len(rows)}