feat: add notion token parameter to Exporter (#60)

notion2md/console/commands/export_block.py

@@ -1,4 +1,3 @@
-import os
 import sys
 import time
 
@@ -11,14 +10,14 @@
 from notion2md.console.ui.indicator import progress
 from notion2md.exporter.block import CLIExporter
 
-
 ARGS_NEW_KEY_MAP = {
     "id": "block_id",
     "url": "block_url",
     "name": "output_filename",
     "path": "output_path",
     "download": "download",
     "unzipped": "unzipped",
+    "token": "token",
 }
 
 
@@ -27,6 +26,7 @@ class ExportBlockCommand(Command):
     description = "Export a Notion block object to markdown."
 
     options = [
+        option("token", "t", "The token of your Notion account.", flag=False),
         option("url", "u", "The url of Notion block object.", flag=False),
         option("id", "i", "The id of Notion block object.", flag=False),
         option("name", "n", "The name of Notion block object", flag=False),

notion2md/exporter/block.py

@@ -16,6 +16,7 @@ def __init__(
         output_path: str = None,
         download: bool = False,
         unzipped: bool = False,
+        token: str = None,
     ):
         self._config = Config(
             block_id=block_id,
@@ -25,7 +26,7 @@ def __init__(
             download=download,
             unzipped=unzipped,
         )
-        self._client = NotionClient()
+        self._client = NotionClient(token)
         self._io = None
         self._block_convertor = None
 

notion2md/notion_api.py

@@ -8,18 +8,19 @@
 def singleton(cls):
     instance = {}
 
-    def get_instance():
+    def get_instance(token=""):
         if cls not in instance:
-            instance[cls] = cls()
+            instance[cls] = cls(token)
         return instance[cls]
 
     return get_instance
 
 
 @singleton
 class NotionClient:
-    def __init__(self):
-        token = self._get_env_variable()
+    def __init__(self, token=""):
+        if not token:
+            token = self._get_env_variable()
         self._client = Client(auth=token)
 
     def _get_env_variable(self):