v1.1

Author: QuentinPhrl

CHANGELOG.md

@@ -1,10 +1,32 @@
 # Change log
 
+## v1.1
+
+### New Features
+
+
+- Recovery of the token with error message in case of failure 
+- Reading options and arguments
+
+### Added:
+
+- File ’parser.py’
+- File ’notion_api.py’
+
 ## v1.0 
 
-Initial release.
+### New Features
+
+- Notion database connexion
+- Notion read and write on the database
+- Reading json file
+- Property formating
 
 ### Added:
 
 - File ’__init__.py’
-- File ’main.py’
+- File ’main.py’
+- File ’page.py’
+- File ’database.py’
+- File ’properties.py’
+- File ’request_notion.py’

main.py

@@ -1,70 +1,77 @@
-from dataclasses import field
-import imp
-import requests
+import optparse
 import json
 import os
 
 from properties import Properties
 from page import Page
 from database import Database
+from parse import Parser
+from notion_api import GetToken
 
+#token = os.environ["NOTION_TOKEN_PUT_EXPERIMENT"]
 
-token = os.environ["NOTION_TOKEN_PUT_EXPERIMENT"]
-
-database_id = 'd08f239718194d1f990c7ed7c67a6653'
-
-headers = {
- "Accept": "application/json",
- "Notion-Version": "2022-02-22",
- "Content-Type": "application/json",
- "Authorization": "Bearer " + token
-}
+#database_id = 'd08f239718194d1f990c7ed7c67a6653'
 
 PROPERTY = Properties()
+PARSER = Parser()
+GETTOKEN = GetToken()
 
 def main():
 
- fields_key, data_extract = json_reader()
- 
- column_type = read_database(fields_key)
+ parser = optparse.OptionParser() 
+ PARSER.get_option(parser=parser)
+
+ datas = json_reader()
+
+ for data in datas:
+
+ fields_key = data[0]
+ data_extract = data[1]
+
+ column_type = read_database(fields_key)
+
+ write_database(column_type,fields_key,data_extract)
 
- write_database(column_type,fields_key,data_extract)
 
 def json_reader():
- 
- data_extract =[]
+ datas = []
+ data_extract = []
 
- json_file = open('data.json')
+ json_file = open(PARSER.options.json_input)
  json_dict = json.load(json_file)
-
- data_extract.append(json_dict["entries"][0]["_id"])
 
- fields_key=list(json_dict["entries"][0]["fields"])
- fields_data=list(map(lambda l:json_dict["entries"][0]["fields"][l],json_dict["entries"][0]["fields"]))
+ for i in range(len(json_dict["entries"])):
+ 
+ fields_key=list(json_dict["entries"][i]["fields"])
+ fields_data=list(map(lambda l:json_dict["entries"][i]["fields"][l],json_dict["entries"][0]["fields"]))
+
+ data_extract=[json_dict["entries"][i]["_id"], fields_data, json_dict["entries"][i]["body"]]
+
+ datas.append([fields_key,data_extract])
 
- data_extract.append(fields_data)
- data_extract.append(json_dict["entries"][0]["body"])
- 
  json_file.close()
 
- return fields_key,data_extract
+ return datas
+
 
 def read_database(fields_key):
 
- D = Database(integrations_token=token)
- D.retrieve_database(database_id=database_id,fields_key=fields_key)
+ D = Database(integrations_token=GETTOKEN.token)
+ D.retrieve_database(database_id=PARSER.options.notion_db_id,fields_key=fields_key)
 
  return D.properties_list
 
+
 def write_database(column_type,fields_key,data_extract):
 
  set_property('title','id',data_extract[0])
 
  for type, key, data in zip(column_type,fields_key,data_extract[1]):
  set_property(type,key,data)
 
- P = Page(integrations_token=token)
- P.create_page(database_id=database_id, properties=PROPERTY)
+ P = Page(integrations_token=GETTOKEN.token)
+ P.create_page(database_id=PARSER.options.notion_db_id, properties=PROPERTY)
+ 
 
 def set_property(type,key,data):
 
@@ -89,5 +96,6 @@ def set_property(type,key,data):
  PROPERTY.set_phone_number(key, data)
  print(PROPERTY.result)
 
+
 if __name__ == '__main__':
  main()

notion_api.py

@@ -0,0 +1,25 @@
+import os
+
+class GetToken:
+ def __init__(self):
+ self.token = self._get_env_variable()
+
+ def _get_env_variable(self):
+ try:
+ return os.environ["NOTION_TOKEN_PUT_EXPERIMENT"]
+ except:
+ print("Notion Integration Token is not found")
+ print(
+ """
+ Welcome to notionput!
+ To get started, you need to save your Notion Integration Token.
+ Find your token at
+ https://www.notion.so/my-integrations
+ Then run shell command:
+ $export NOTION_TOKEN="<Your Token>"
+ 
+ If you want to save this environment variable after reboot,
+ put upper command in your shell resource(ex: .bashrc or .zshrc)
+ """
+ )
+ exit(1)

page.py

@@ -18,7 +18,6 @@ def create_page(self, database_id, properties=None):
  Create a page
  :param database_id: Identifier for a Notion database
  :param properties: Property values of this page
- :param children: Page content for the new page
  :return:
  """
  if properties is None:

parse.py

@@ -0,0 +1,25 @@
+class Parser:
+ def __init__(self):
+ self.options = object
+
+ def get_option(self,parser):
+ # create OptionParser object
+ 
+ # add options
+ parser.add_option('-d','--notion-db-id',
+ dest = 'notion_db_id',
+ type = 'string',
+ help = 'specify the id of the database notion')
+ parser.add_option('-f', '--json-file',
+ dest = 'json_input',
+ type = 'string',
+ help = 'specify the input json file')
+ 
+ (options, args) = parser.parse_args()
+ self.options = options
+ if (options.notion_db_id == None):
+ print("exit")
+ exit(0)
+ elif (options.json_input == None):
+ print("exit")
+ exit(0)