Snapshot your Droplets every night

from datetime import date, datetime, timedelta, timezone
from clickruntime import on_trigger, credentials, http, notify

DO = "https://api.digitalocean.com/v2"

def do(method, path, **kwargs):
    resp = http.request(method, f"{DO}{path}", headers={"Authorization": f"Bearer {credentials.secret('DIGITALOCEAN_TOKEN')}"}, **kwargs)
    resp.raise_for_status()
    return resp.json() if resp.content else {}

@on_trigger
def run(event):
    cutoff = datetime.now(timezone.utc) - timedelta(days=7)
    droplets = do("GET", "/droplets", params={"tag_name": "prod", "per_page": 200})["droplets"]
    ids = {str(d["id"]) for d in droplets}
    created, removed = [], []
    for d in droplets:
        name = f"{d['name']}-{date.today()}"
        do("POST", f"/droplets/{d['id']}/actions", json={"type": "snapshot", "name": name})
        created.append(name)
    for s in do("GET", "/snapshots", params={"resource_type": "droplet", "per_page": 200})["snapshots"]:
        if s["resource_id"] in ids and datetime.fromisoformat(s["created_at"].replace("Z", "+00:00")) < cutoff:
            do("DELETE", f"/snapshots/{s['id']}")
            removed.append(s["name"])
    notify.email_me(f"Snapshots: {len(created)} created, {len(removed)} removed",
                    "Created:\n" + "\n".join(created) + "\n\nRemoved:\n" + ("\n".join(removed) or "none"))
    return {"created": len(created), "removed": len(removed)}