Clean up product images from a sheet queue

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

SHEET_ID, SITE = "your-spreadsheet-id", "https://yourstore.com"

STORE = "https://yourstore.com/wp-json/wc/v3"

def wc(method, path, **kw):
    resp = http.request(method, STORE + path, auth=(credentials.secret("WC_CONSUMER_KEY"), credentials.secret("WC_CONSUMER_SECRET")), **kw)
    resp.raise_for_status()
    return resp.json()

def sheets():
    return build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).spreadsheets().values()

@on_trigger
def run(event):
    rows = sheets().get(spreadsheetId=SHEET_ID, range="Images!A2:B").execute().get("values", [])
    todo = [(n, r[0]) for n, r in enumerate(rows, start=2) if r and r[0] and (len(r) < 2 or r[1] != "Done")][:5]
    report_fetched(len(todo))
    for n, pid in todo:
        p = wc("GET", f"/products/{pid}")
        clean = http.post("https://api.remove.bg/v1.0/removebg", headers={"X-Api-Key": credentials.secret("REMOVE_BG_API_KEY")},
                          data={"image_url": p["images"][0]["src"], "bg_color": "ffffff", "format": "jpg"})
        clean.raise_for_status()
        media = http.post(f"{SITE}/wp-json/wp/v2/media", auth=("shop-admin", credentials.secret("WP_APP_PASSWORD")), data=clean.content,
                          headers={"Content-Disposition": f'attachment; filename="{p["slug"]}-clean.jpg"', "Content-Type": "image/jpeg"}).json()
        wc("PUT", f"/products/{pid}", json={"images": [{"id": media["id"]}] + [{"id": i["id"]} for i in p["images"]]})
        sheets().update(spreadsheetId=SHEET_ID, range=f"Images!B{n}", valueInputOption="RAW", body={"values": [["Done"]]}).execute()
    return {"done": len(todo)}