Merge pull request #1 from DominiqueGarmier/feature/new-parser

new parser

Author: dominiquegarmier

notion_graph/client.py

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+from contextlib import asynccontextmanager
+from typing import Any
+from typing import cast
+
+from aiohttp import ClientSession
+
+from notion_graph.config import config
+
+
+@asynccontextmanager
+async def HeaderSession(auth: str | None = None):
+    auth = auth or config.notion_key
+    headers = {
+        'Authorization': f'Bearer {auth}',
+        'Notion-Version': '2022-02-22',
+    }
+    async with ClientSession(headers=headers) as session:
+        yield session
+
+
+async def _api_get(endpoint: str) -> dict[str, Any]:
+    async with HeaderSession() as session:
+        async with session.get(endpoint) as resp:
+            return cast(dict[str, Any], await resp.json())
+
+
+async def get_page(page_id: str) -> dict[str, Any]:
+    return await _api_get(f'https://api.notion.com/v1/pages/{page_id}')
+
+
+async def get_children(block_id: str) -> dict[str, Any]:
+    return await _api_get(f'https://api.notion.com/v1/blocks/{block_id}/children')
+
+
+async def get_block(block_id: str) -> dict[str, Any]:
+    return await _api_get(f'https://api.notion.com/v1/blocks/{block_id}')

notion_graph/config.py

@@ -1,12 +1,16 @@
 from __future__ import annotations
 
+import logging
 import os
 from functools import cache
 from pathlib import Path
 from typing import NamedTuple
 
 from dotenv import load_dotenv
 
+logger = logging.getLogger(__name__)
+
+
 __all__ = [
     'Config',
     'config',
@@ -24,8 +28,12 @@ class Config(NamedTuple):
 @cache
 def get_config() -> Config:
     load_dotenv()
-    notion_key = os.environ['NOTION_KEY']
-    root_id = os.environ['ROOT_ID']
+    try:
+        notion_key = os.environ['NOTION_KEY']
+        root_id = os.environ['ROOT_ID']
+    except KeyError:
+        logger.warning('Please set NOTION_KEY and ROOT_ID in .env')
+        raise SystemExit(1)
     return Config(notion_key=notion_key, root_id=root_id)
 
 

notion_graph/main.py

@@ -1,13 +1,17 @@
 from __future__ import annotations
 
 import argparse
+import logging
 from typing import cast
 
 from notion_graph.parser import parser_main
 from notion_graph.server import server_main
 
 
 def main() -> int:
+
+    logging.basicConfig(level=logging.INFO, format='%(message)s')
+
     parser = argparse.ArgumentParser()
     subparsers = parser.add_subparsers(required=True)