Save a monthly Stripe revenue report to Drive

import csv, io, time, stripe
from datetime import date
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from googleapiclient.http import MediaIoBaseUpload
from clickruntime import on_trigger, credentials, notify, ai

FOLDER_ID = "your-accounting-folder-id"

@on_trigger
def run(event):
    stripe.api_key = credentials.secret("STRIPE_API_KEY")
    end = date.today().replace(day=1)
    start = end.replace(year=end.year - 1, month=12) if end.month == 1 else end.replace(month=end.month - 1)
    period = {"gte": int(time.mktime(start.timetuple())), "lt": int(time.mktime(end.timetuple()))}
    out, totals = io.StringIO(), {}
    w = csv.writer(out)
    w.writerow(["Date", "Type", "Description", "Gross", "Fee", "Net", "Currency"])
    for tx in stripe.BalanceTransaction.list(created=period, limit=100).auto_paging_iter():
        w.writerow([date.fromtimestamp(tx.created), tx.type, tx.description, tx.amount / 100, tx.fee / 100, tx.net / 100, tx.currency.upper()])
        totals[tx.type] = round(totals.get(tx.type, 0) + tx.amount / 100, 2)
    drive = build("drive", "v3", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    f = drive.files().create(body={"name": f"Stripe {start:%Y-%m}.csv", "parents": [FOLDER_ID]}, fields="webViewLink",
                             media_body=MediaIoBaseUpload(io.BytesIO(out.getvalue().encode()), mimetype="text/csv")).execute()
    summary = ai.generate(f"Stripe totals by type for {start:%B %Y}: {totals}", system="Write a 3-sentence plain summary for a business owner.", max_output_tokens=200).text
    notify.email_me(f"Stripe report {start:%B %Y}", f"{summary}\n\nFile: {f['webViewLink']}")
    return {"types": totals}