Add page, database, divider, heading, and paragraph functionality

.gitignore

@@ -0,0 +1,2 @@
+.idea
+blueprints/__pycache__

blueprints/architect.py

@@ -0,0 +1,24 @@
+from .blocks import create_page, create_database, create_divider, create_heading, create_paragraph
+
+
+def process_blueprint(parent_id, block_json):
+    block_type = block_json["type"]
+
+    if block_type == "page":
+        title = block_json.get("title", "Untitled Page")
+        page_id = create_page(parent_id, title)["id"]
+        for child in block_json.get("children", []):
+            process_blueprint(page_id, child)
+    elif block_type == "database":
+        title = block_json.get("title", "Untitled Database")
+        schema = block_json.get("schema", {})
+        create_database(parent_id, title, schema)
+    elif block_type == "divider":
+        create_divider(parent_id)
+    elif block_type.startswith("heading_"):
+        text = block_json.get("text", "")
+        level = int(block_type.split("_")[1])
+        create_heading(parent_id, text, level)
+    elif block_type == "paragraph":
+        text = block_json.get("text", "")
+        create_paragraph(parent_id, text)

blueprints/blocks.py

@@ -0,0 +1,118 @@
+import os
+
+from notion_client import Client
+
+notion = Client(auth=os.environ["NOTION_KEY"])
+
+
+def create_page(parent_id, title):
+    new_page = notion.pages.create(**{
+        "parent": {
+            "page_id": parent_id
+        },
+        "properties": {
+            "title": {
+                "title": [
+                    {
+                        "type": "text",
+                        "text": {
+                            "content": title,
+                        },
+                    }
+                ]
+            },
+        },
+    })
+    return new_page
+
+
+def create_database(parent_id, title, schema):
+    approved_types = (
+        "title", "rich_text", "number", "checkbox", "date", "people",
+        "files", "url", "email", "phone_number", "select", "multi_select"
+    )
+
+    properties = {}
+    for name, info in schema.items():
+        prop_type = info["type"]
+        if prop_type not in approved_types:
+            raise ValueError(f"Property type '{prop_type}' is currently unsupported.")
+
+        if prop_type in ["select", "multi_select"]:
+            properties[name] = {prop_type: {"options": info["options"]}}
+        else:
+            properties[name] = {prop_type: {}}
+
+    new_database = notion.databases.create(**{
+        "parent": {"page_id": parent_id},
+        "title": [
+            {
+                "type": "text",
+                "text": {
+                    "content": title
+                }
+            }
+        ],
+        "properties": properties
+    })
+    return new_database
+
+
+def create_divider(parent_id):
+    divider_block = notion.blocks.children.append(**{
+        "block_id": parent_id,
+        "children": [
+            {
+                "object": "block",
+                "type": "divider",
+                "divider": {},
+            }
+        ]
+    })
+    return divider_block
+
+
+def create_heading(parent_id, text, level):
+    heading_block = notion.blocks.children.append(**{
+        "block_id": parent_id,
+        "children": [
+            {
+                "object": "block",
+                "type": f"heading_{level}",
+                f"heading_{level}": {
+                    "rich_text": [
+                        {
+                            "type": "text",
+                            "text": {
+                                "content": text
+                            }
+                        }
+                    ],
+                },
+            }
+        ]
+    })
+    return heading_block
+
+
+def create_paragraph(parent_id, text):
+    paragraph_block = notion.blocks.children.append(**{
+        "block_id": parent_id,
+        "children": [
+            {
+                "object": "block",
+                "type": "paragraph",
+                "paragraph": {
+                    "rich_text": [
+                        {
+                            "type": "text",
+                            "text": {
+                                "content": text
+                            }
+                        }
+                    ],
+                },
+            }
+        ]
+    })
+    return paragraph_block

run.py

@@ -0,0 +1,26 @@
+import argparse
+import json
+import os
+
+from blueprints.architect import process_blueprint
+
+
+def load_blueprint(filename):
+    with open(filename, "r") as file:
+        return json.load(file)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Process a JSON blueprint to create content in Notion.")
+    parser.add_argument("json_file", help="Location of the JSON blueprint file")
+    args = parser.parse_args()
+
+    notion_page_id = os.environ["NOTION_PAGE_ID"]
+
+    blueprint = load_blueprint(args.json_file)
+    process_blueprint(notion_page_id, blueprint)
+
+
+if __name__ == "__main__":
+    main()
+

sample_data.json

@@ -0,0 +1,101 @@
+{
+  "type": "page",
+  "title": "Weekly Grocery Planning",
+  "children": [
+    {
+      "type": "heading_1",
+      "text": "Grocery List Overview"
+    },
+    {
+      "type": "paragraph",
+      "text": "Organize your weekly grocery shopping with a detailed list and budget tracking."
+    },
+    {
+      "type": "database",
+      "title": "Grocery List",
+      "schema": {
+        "Name": {
+          "type": "title"
+        },
+        "Description": {
+          "type": "rich_text"
+        },
+        "In stock": {
+          "type": "checkbox"
+        },
+        "Food group": {
+          "type": "select",
+          "options": [
+            {
+              "name": "🥦Vegetable",
+              "color": "green"
+            },
+            {
+              "name": "🍎Fruit",
+              "color": "red"
+            },
+            {
+              "name": "💪Protein",
+              "color": "yellow"
+            }
+          ]
+        },
+        "Price": {
+          "type": "number",
+          "format": "dollar"
+        },
+        "Last ordered": {
+          "type": "date"
+        },
+        "Store availability": {
+          "type": "multi_select",
+          "options": [
+            {
+              "name": "Duc Loi Market",
+              "color": "blue"
+            },
+            {
+              "name": "Rainbow Grocery",
+              "color": "gray"
+            },
+            {
+              "name": "Nijiya Market",
+              "color": "purple"
+            },
+            {
+              "name": "Gus's Community Market",
+              "color": "yellow"
+            }
+          ]
+        },
+        "+1": {
+          "type": "people"
+        },
+        "Photo": {
+          "type": "files"
+        }
+      }
+    },
+    {
+      "type": "divider"
+    },
+    {
+      "type": "heading_2",
+      "text": "Shopping Tips"
+    },
+    {
+      "type": "paragraph",
+      "text": "Consider seasonal produce for fresher options and better prices. Don't forget to check pantry staples before heading out."
+    },
+    {
+      "type": "page",
+      "title": "Favorite Recipes",
+      "children": [
+        {
+          "type": "paragraph",
+          "text": "Compile your go-to recipes here for quick access while planning meals."
+        }
+      ]
+    }
+  ]
+}