Merge pull request #252 from adamtheturtle/database-support

Add support for publishing to a database

Author: adamtheturtle

README.rst

@@ -141,9 +141,10 @@ Usage
 
 Arguments:
 
-- ``-f, --file``: Path to the JSON file generated by the Notion builder
-- ``-p, --parent-page-id``: The ID of the parent page in Notion (must be shared with your integration)
-- ``-t, --title``: Title for the new page in Notion
+- ``--file``: Path to the JSON file generated by the Notion builder
+- ``--parent-id``: The ID of the parent page or database in Notion (must be shared with your integration)
+- ``--parent-type``: "page" or "database"
+- ``--title``: Title for the new page in Notion
 
 The command will create a new page if one with the given title doesn't exist, or update the existing page if one with the given title already exists.
 

src/_notion_scripts/upload.py

@@ -5,8 +5,9 @@
 
 import json
 import sys
+from enum import Enum
 from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
 from urllib.parse import urlparse
 from urllib.request import url2pathname
 
@@ -21,6 +22,10 @@
 from ultimate_notion.file import UploadedFile
 from ultimate_notion.obj_api.blocks import Block as UnoObjAPIBlock
 
+if TYPE_CHECKING:
+    from ultimate_notion.database import Database
+    from ultimate_notion.page import Page
+
 
 @beartype
 def _upload_local_file(
@@ -66,6 +71,16 @@ def _block_from_details(
     return block
 
 
+@beartype
+class _ParentType(Enum):
+    """
+    Type of parent that new page will live under.
+    """
+
+    PAGE = "page"
+    DATABASE = "database"
+
+
 @click.command()
 @click.option(
     "--file",
@@ -79,10 +94,16 @@ def _block_from_details(
     ),
 )
 @click.option(
-    "--parent-page-id",
-    help="Parent page ID (integration connected)",
+    "--parent-id",
+    help="Parent page or database ID (integration connected)",
     required=True,
 )
+@click.option(
+    "--parent-type",
+    help="Parent type",
+    required=True,
+    type=click.Choice(choices=_ParentType, case_sensitive=False),
+)
 @click.option(
     "--title",
     help="Title of the page to update (or create if it does not exist)",
@@ -97,7 +118,8 @@ def _block_from_details(
 def main(
     *,
     file: Path,
-    parent_page_id: str,
+    parent_id: str,
+    parent_type: _ParentType,
     title: str,
     icon: str | None = None,
 ) -> None:
@@ -108,11 +130,17 @@ def main(
 
     blocks = json.loads(s=file.read_text(encoding="utf-8"))
 
-    parent_page = session.get_page(page_ref=parent_page_id)
+    parent: Page | Database
+    match parent_type:
+        case _ParentType.PAGE:
+            parent = session.get_page(page_ref=parent_id)
+            subpages = parent.subpages
+        case _ParentType.DATABASE:
+            parent = session.get_db(db_ref=parent_id)
+            subpages = parent.get_all_pages().to_pages()
+
     pages_matching_title = [
-        child_page
-        for child_page in parent_page.subpages
-        if child_page.title == title
+        child_page for child_page in subpages if child_page.title == title
     ]
 
     if pages_matching_title:
@@ -123,7 +151,7 @@ def main(
         assert len(pages_matching_title) == 1, msg
         (page,) = pages_matching_title
     else:
-        page = session.create_page(parent=parent_page, title=title)
+        page = session.create_page(parent=parent, title=title)
         sys.stdout.write(f"Created new page: {title} (ID: {page.id})\n")
 
     if icon: