Get a weekly list of invoices from your inbox

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
import base64
from datetime import datetime
from zoneinfo import ZoneInfo
from clickruntime import on_trigger, credentials, state, ai, notify

SCHEMA = {"type": "object", "properties": {"vendor": {"type": "string"}, "amount": {"type": "string"},
          "due_date": {"type": "string"}}, "required": ["vendor", "amount", "due_date"]}

@on_trigger
def run(event):
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    found, done = state.get("invoices") or [], set(state.get("done") or [])
    msgs = gmail.users().messages().list(userId="me", q="has:attachment (invoice OR receipt) newer_than:2d").execute().get("messages", [])
    for m in [m for m in msgs if m["id"] not in done]:
        msg = gmail.users().messages().get(userId="me", id=m["id"]).execute()
        names = [p["filename"] for p in msg["payload"].get("parts", []) if p.get("filename")]
        text = next((base64.urlsafe_b64decode(p["body"]["data"]).decode(errors="ignore")
                     for p in msg["payload"].get("parts", []) if p["mimeType"] == "text/plain" and p["body"].get("data")), msg["snippet"])
        found.append(ai.generate(f"Extract vendor, total amount and due date. Attachments: {names}\n{text[:4000]}",
                                 json_schema=SCHEMA, max_output_tokens=100).json)
        done.add(m["id"])
    now = datetime.now(ZoneInfo("Europe/London"))
    if now.weekday() == 4 and now.hour == 16 and found:  # Friday 4pm: send the week's list
        notify.email_me(f"{len(found)} invoices this week",
                        "\n".join(f"{i['vendor']}: {i['amount']}, due {i['due_date']}" for i in found))
        found = []
    state.set("invoices", found)
    state.set("done", sorted(done)[-1000:])
    return {"pending": len(found)}