Python×Notion｜下書きDBへAPIで流し込む

• 1
• 2
• 3

python3 -m venv .venv
source .venv/bin/activate   # Windowsは .venv\Scripts\activate
pip install requests

• 1
• 2

export NOTION_TOKEN="（発行したトークン）"
export NOTION_DATABASE_ID="（データベースのID）"

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16
• 17
• 18
• 19
• 20
• 21
• 22
• 23
• 24
• 25
• 26
• 27
• 28
• 29
• 30
• 31
• 32
• 33
• 34
• 35
• 36
• 37
• 38
• 39
• 40
• 41
• 42
• 43
• 44
• 45
• 46
• 47
• 48
• 49
• 50
• 51
• 52
• 53
• 54
• 55
• 56
• 57
• 58
• 59
• 60
• 61
• 62
• 63
• 64
• 65
• 66

import os
import time
from pathlib import Path

import requests

NOTION_TOKEN = os.environ["NOTION_TOKEN"]
DATABASE_ID = os.environ["NOTION_DATABASE_ID"]
HEADERS = {
    "Authorization": f"Bearer {NOTION_TOKEN}",
    "Notion-Version": "2022-06-28",
    "Content-Type": "application/json",
}

BASE_DIR = Path(__file__).resolve().parent
DRAFT_DIR = BASE_DIR / "drafts"
DONE_DIR = BASE_DIR / "done"


def paragraph(text: str) -> dict:
    # 1つのテキストは2000文字までなので、念のため切り詰める
    return {
        "object": "block",
        "type": "paragraph",
        "paragraph": {"rich_text": [{"type": "text", "text": {"content": text[:2000]}}]},
    }


def create_draft(title: str, slug: str, paragraphs: list[str]) -> str:
    payload = {
        "parent": {"database_id": DATABASE_ID},
        "properties": {
            "Page": {"title": [{"text": {"content": title}}]},
            "Slug": {"rich_text": [{"text": {"content": slug}}]},
            "Tags": {"multi_select": [{"name": "下書き"}]},
            "Published": {"checkbox": False},
        },
        # 1回のリクエストで追加できるブロックは100個まで
        "children": [paragraph(p) for p in paragraphs][:100],
    }
    res = requests.post("https://api.notion.com/v1/pages", headers=HEADERS, json=payload, timeout=30)
    res.raise_for_status()
    return res.json()["url"]


def main() -> None:
    DONE_DIR.mkdir(exist_ok=True)
    for path in sorted(DRAFT_DIR.glob("*.txt")):
        lines = path.read_text(encoding="utf-8").splitlines()
        if not lines:
            continue
        title = lines[0].strip()
        # 空行で区切られたかたまりを、1つの段落にする
        body = "\n".join(lines[1:]).strip()
        paragraphs = [p.strip() for p in body.split("\n\n") if p.strip()]

        url = create_draft(title, path.stem, paragraphs)
        print(f"登録しました：{title} -> {url}")
        path.rename(DONE_DIR / path.name)

        # APIの利用制限にかからないよう、少し間を空ける
        time.sleep(0.5)


if __name__ == "__main__":
    main()