Save client email attachments into Drive, one folder per company

import base64
from googleapiclient.http import MediaInMemoryUpload
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, credentials, report_fetched

PARENT_ID, DONE_LABEL = "your-clients-folder-id", "Label_saved"

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    gmail, drive = build("gmail", "v1", credentials=creds, cache_discovery=False), build("drive", "v3", credentials=creds, cache_discovery=False)
    msgs = gmail.users().messages().list(userId="me", q="label:Clients has:attachment -label:saved").execute().get("messages", [])
    report_fetched(len(msgs))
    if not msgs:
        return {"saved": 0}
    for m in msgs:
        msg = gmail.users().messages().get(userId="me", id=m["id"]).execute()
        sender = next(h["value"] for h in msg["payload"]["headers"] if h["name"] == "From")
        company = sender.split("@")[-1].split(".")[0].rstrip(">").title()
        found = drive.files().list(q=f"name='{company}' and '{PARENT_ID}' in parents and trashed=false", fields="files(id)").execute()["files"]
        folder = found[0]["id"] if found else drive.files().create(body={"name": company, "parents": [PARENT_ID],
                 "mimeType": "application/vnd.google-apps.folder"}, fields="id").execute()["id"]
        for part in msg["payload"].get("parts", []):
            if part.get("filename") and part["body"].get("attachmentId"):
                data = gmail.users().messages().attachments().get(userId="me", messageId=m["id"], id=part["body"]["attachmentId"]).execute()
                drive.files().create(body={"name": part["filename"], "parents": [folder]},
                                     media_body=MediaInMemoryUpload(base64.urlsafe_b64decode(data["data"]), mimetype=part["mimeType"])).execute()
        gmail.users().messages().modify(userId="me", id=m["id"], body={"addLabelIds": [DONE_LABEL]}).execute()
    return {"saved": len(msgs)}