Turn one idea into a week of posts in Notion

from datetime import date, timedelta
from clickruntime import on_trigger, credentials, http, ai

NOTION = "https://api.notion.com/v1"

def headers():
    return {"Authorization": f"Bearer {credentials.secret('NOTION_TOKEN')}", "Notion-Version": "2022-06-28"}

def query(db, body=None):
    resp = http.post(f"{NOTION}/databases/{db}/query", headers=headers(), json=body or {})
    resp.raise_for_status()
    return resp.json()["results"]

def title(page, prop="Name"):
    return "".join(t["plain_text"] for t in page["properties"][prop]["title"])

IDEAS, CALENDAR = "ideas-database-id", "content-calendar-database-id"
SCHEMA = {"type": "object", "properties": {"posts": {"type": "array", "items": {"type": "string"}}}, "required": ["posts"]}

@on_trigger
def run(event):
    ideas = query(IDEAS, {"filter": {"property": "Status", "select": {"equals": "Ready"}},
                          "sorts": [{"timestamp": "created_time", "direction": "ascending"}], "page_size": 1})
    if not ideas:
        return {"drafted": 0}
    idea = title(ideas[0])
    posts = ai.generate(f"Write 7 different LinkedIn posts (under 150 words each) about: {idea}",
                        json_schema=SCHEMA, max_output_tokens=2000).json["posts"][:7]
    monday = date.today()
    for i, post in enumerate(posts):
        http.post(f"{NOTION}/pages", headers=headers(), json={"parent": {"database_id": CALENDAR}, "properties": {
            "Name": {"title": [{"text": {"content": f"{idea} ({i + 1}/7)"}}]},
            "Date": {"date": {"start": (monday + timedelta(days=i)).isoformat()}},
            "Draft": {"rich_text": [{"text": {"content": post[:2000]}}]},
        }}).raise_for_status()
    http.patch(f"{NOTION}/pages/{ideas[0]['id']}", headers=headers(),
               json={"properties": {"Status": {"select": {"name": "Used"}}}}).raise_for_status()
    return {"idea": idea, "drafted": len(posts)}