Update to notion api (#4)

* UPDATE: use notion-sdk-py instead of notion client

Author: stevedsun

.gitignore

@@ -129,3 +129,5 @@ dmypy.json
 .pyre/
 .idea/
 graph_view.html
+run_local.sh
+.vscode

Pipfile

@@ -1,36 +1,64 @@
+![](images/snap.png)
+
 # Notion Graph View
 
 ![](https://img.shields.io/github/pipenv/locked/python-version/stevedsun/notion-graph-view)
 
-Export [Notion](https://notion.so) pages to a Roam Research like graph view.
+Export [Notion](https://notion.so) pages to a Roam-Research like graph view.
+
+## Usage
 
-![](https://tva1.sinaimg.cn/large/008eGmZEly1gnhdionmecj30yf0u0115.jpg)
+### Environment
 
-## How to use it
+- Python 3.7 or later (3.9 is recommended)
 
-Install dependencies with `pipenv` or `pip`:
+### Install
 
 ```shell
 pip install -r requirements.txt
-# or
-pipenv install
 ```
 
-Login notion.so and get `token_v2` from browser ([How?](https://www.redgregory.com/notion/2020/6/15/9zuzav95gwzwewdu1dspweqbv481s5)). Paste it to `config.py`>`my_token_v2`.
+### Setup Notion API
+
+1. Create a [notion internal integration](https://www.notion.so/my-integrations) and generate an `Internal Integration Token`. [Learn more about authorization](https://developers.notion.com/docs/authorization)
+2. Open one notion page on browser and share it to your integration.
+3. Find your base `Page ID` from browser url, for example:
+   > if this page url is like: https://www.notion.so/yourName/PageTitle-8a4b5ff100d648fb8d39d4bfa756ff3f
+   > the `8a4b5ff100da48fb8d39d4bfa756ff3f` is the `Page ID`
+4. Paste `Internal Integration Token` and `Page ID` to `config.py`:
+
+```python
+NOTION_TOKEN = "secret_TBqfsxyH1slTpaignyZqQnDAAAn0MaeDEc2l96cdubD"
+PAGE_ID = "8a4b5ff100d648fb8d39d4bfa756ff3f"
+```
 
-Paste the page url which your want to analyse into `config.py`>`my_url`.
+> On Linux, you can export these two environment varibles instead.
 
-Then run:
+## Run
 
 ```shell
 python main.py
 ```
 
-Finally `graph_view.html` will be generated at the current path, open it with any browser.
+Finally `graph_view.html` would be generated at the project path, open it with any browser. (make sure `/lib` and `graph_view.html` in the same folder)
 
-## Todo
+## Block link support
 
-- [x] Read Notion pages, export to graph view image
-- [ ] Upgrade python version
-- [ ] Replace notion-client by notion-sdk-py
-- [ ] Use lightweight renderer solution
+|                    | database | page |
+| ------------------ | -------- | ---- |
+| paragraph          | ✔️       | ✔️   |
+| bulleted_list_item | ✔️       | ✔️   |
+| numbered_list_item | ✔️       | ✔️   |
+| to_do              | ✔️       | ✔️   |
+| toggle             | ✔️       | ✔️   |
+| child_page         | ✔️       | ✔️   |
+| child_database     | ✔️       | ✔️   |
+| embed              |          |      |
+| callout            |          |      |
+| quote              |          |      |
+| column             |          |      |
+| column_list        |          |      |
+| synced_block       |          |      |
+| link_to_page       |          |      |
+| table              | ✔️       | ✔️   |
+| table_row          |          |      |

Pipfile.lock

@@ -1,2 +1,15 @@
-my_token_v2 = "{your_token_v2}"
-my_url = "{root_page_url}"
+import os
+from notion_client import Client
+
+
+# Paste your NOTION token (e.g. secret_balabalabalabala) and page id (e.g. 43a57ade6650460b980e96fde16b03f1) in "" below.
+NOTION_TOKEN = ""
+PAGE_ID = ""
+
+if not NOTION_TOKEN:
+    NOTION_TOKEN = os.getenv("NOTION_TOKEN")
+
+if not PAGE_ID:
+    PAGE_ID = os.getenv("PAGE_ID")
+
+notion = Client(auth=NOTION_TOKEN)

README.md

@@ -0,0 +1,38 @@
+class Graph:
+    def __init__(self):
+        self.nodes = []
+        self.edges = []
+        self.parsed_block_ids = set()
+
+    def has_node_parsed(self, block):
+        return block.id in self.parsed_block_ids and block.title != '<block>'
+
+    def add_node(self, block):
+        if self.has_node_parsed(block):
+            return
+
+        page_block_id = ''
+        if block.parent_id:
+            # for text block
+            page_block_id = block.parent_id
+        else:
+            # for page and database block
+            page_block_id = block.id
+            if block.title != '<block>':
+                self.nodes.append(
+                    dict(id=page_block_id, label=block.title, size=2))
+
+        for child_id in block.children_ids:
+            self.edges.append(dict({"from": page_block_id, "to": child_id}))
+
+        for linked_block in block.linked_blocks:
+            linked_id = linked_block['id']
+            self.edges.append(dict({"from": page_block_id, "to": linked_id}))
+
+        self.parsed_block_ids.add(block.id)
+
+    def get_graph(self) -> dict[str, list]:
+        return dict(nodes=self.nodes, edges=self.edges)
+
+
+my_graph = Graph()

config.py

@@ -1,26 +1,14 @@
-import os
-from notion.client import NotionClient
-
-from config import *
-from parser import Parser
+from graph import my_graph
+from parser import BlockParser
 from render import render
+from config import *
 
 
-def read_page(token_v2, page_url):
-    client = NotionClient(token_v2=token_v2)
-    base_block = client.get_block(page_url)
-    print("Root Page:", base_block.title)
-    bfs_block(base_block)
-    print("Graph view generated")
-    os.system("open graph_view.html")
-
-
-def bfs_block(block):
-    print(block.type, block.title)
-    parser = Parser(block)
-    graph = parser.get_graph()
-    render(block.title, graph)
+def read_page():
+    block = notion.blocks.retrieve(PAGE_ID)
+    BlockParser(block)
+    render(my_graph.get_graph())
 
 
 if __name__ == '__main__':
-    read_page(my_token_v2, my_url)
+    read_page()