v1 Basic Elementary function

Author: QuentinPhrl

__init__.py

@@ -0,0 +1 @@
+NOTION_VERSION = "2022-02-22"

database.py

@@ -0,0 +1,29 @@
+from request_notion import Request
+
+class Database:
+ def __init__(self, integrations_token):
+ """
+ init
+ :param integrations_token: Notion Internal Integration Token
+ """
+ self.properties_list = []
+ self.url = 'https://api.notion.com/v1/databases'
+ self.result = {}
+ self.request = Request(self.url, integrations_token=integrations_token)
+ 
+
+ def retrieve_database(self, database_id, fields_key):
+ """
+ Retrieve a database
+ :param database_id: Identifier for a Notion database
+ :param get_properties: Get properties_list trigger
+ :return:
+ """
+
+ self.result = self.request.call_api_get(self.url + "/" + database_id)
+
+ for field_key in fields_key:
+ self.properties_list.append(self.result["properties"][field_key]["type"])
+
+
+ 

main.py

@@ -1,19 +1,93 @@
+from dataclasses import field
+import imp
+import requests
+import json
 import os
 
-token = os.environ["NOTION_TOKEN_PUT"]
-BEARER_TOKEN = os.environ["NOTION_BEARER_TOKEN"]
+from properties import Properties
+from page import Page
+from database import Database
 
-main_page_id = 'fa2142e7178a4af980ef0bc2bcde14ec'
+
+token = os.environ["NOTION_TOKEN_PUT_EXPERIMENT"]
+
+database_id = 'd08f239718194d1f990c7ed7c67a6653'
 
 headers = {
  "Accept": "application/json",
- "Notion-Version": "2021-08-16",
+ "Notion-Version": "2022-02-22",
  "Content-Type": "application/json",
  "Authorization": "Bearer " + token
 }
 
+PROPERTY = Properties()
+
 def main():
- print("main")
+
+ fields_key, data_extract = json_reader()
+ 
+ column_type = read_database(fields_key)
+
+ write_database(column_type,fields_key,data_extract)
+ 
+def json_reader():
+ 
+ data_extract =[]
+
+ json_file = open('data.json')
+ 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"]))
+
+ data_extract.append(fields_data)
+ data_extract.append(json_dict["entries"][0]["body"])
+ 
+ json_file.close()
+
+ return fields_key,data_extract
+
+def read_database(fields_key):
+
+ D = Database(integrations_token=token)
+ D.retrieve_database(database_id=database_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)
+
+def set_property(type,key,data):
+
+ match type:
+ case 'title':
+ PROPERTY.set_title(key, data)
+ case 'rich_text':
+ PROPERTY.set_rich_text(key, data)
+ case 'number':
+ PROPERTY.set_number(key, data)
+ case 'select':
+ PROPERTY.set_select(key, data)
+ case 'select':
+ PROPERTY.set_multi_select(key, data)
+ case 'checkbox':
+ PROPERTY.set_checkbox(key, data)
+ case 'url':
+ PROPERTY.set_checkbox(key, data)
+ case 'email':
+ PROPERTY.set_email(key, data)
+ case 'phone_number':
+ PROPERTY.set_phone_number(key, data)
+ print(PROPERTY.result)
 
 if __name__ == '__main__':
  main()

page.py

@@ -0,0 +1,34 @@
+
+from properties import Properties
+from request_notion import Request
+
+
+class Page:
+ def __init__(self, integrations_token):
+ """
+ init
+ :param integrations_token: Notion Internal Integration Token
+ """
+ self.url = 'https://api.notion.com/v1/pages'
+ self.result = {}
+ self.request = Request(self.url, integrations_token=integrations_token)
+
+ 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:
+ properties = Properties()
+ properties = properties
+
+ body = {
+ "parent": {
+ "database_id": database_id
+ },
+ "properties": properties.result,
+ }
+ self.result = self.request.call_api_post(self.url, body)

properties.py

@@ -0,0 +1,125 @@
+class Properties:
+ def __init__(self):
+ """
+ init
+ """
+ self.result = {}
+
+ def set_title(self, col, text=None):
+ """
+ title configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if text:
+ text = [{"text": {"content": text}}]
+ else:
+ text = {}
+ self.result.update({col: {"title": text}})
+
+ def set_rich_text(self, col, text=None):
+ """
+ rich_text configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if text:
+ text = [{"text": {"content": text}}]
+ else:
+ text = {}
+ self.result.update({col: {"rich_text": text}})
+
+ def set_number(self, col, text=None):
+ """
+ number configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if text:
+ text = int(text)
+ else:
+ text = {}
+ self.result.update({col: {"number": text}})
+
+ def set_select(self, col, text=None):
+ """
+ select configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if text:
+ text = {"name": text}
+ else:
+ text = {}
+ self.result.update({col: {"select": text}})
+
+ def set_multi_select(self, col, text_list=None):
+ """
+ multi select configuration
+ :param col: column name
+ :param text_list: page text list. If no text is given, for database only.
+ :return:
+ """
+ if text_list:
+ data = []
+ for i in text_list:
+ data.append({"name": i})
+ else:
+ data = {}
+
+ self.result.update({col: {"multi_select": data}})
+
+ def set_checkbox(self, col, text=None):
+ """
+ checkbox configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if not text:
+ text = {}
+ self.result.update({col: {"checkbox": text}})
+
+ def set_url(self, col, text=None):
+ """
+ url configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if not text:
+ text = {}
+ self.result.update({col: {"url": text}})
+
+ def set_email(self, col, text=None):
+ """
+ email configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if not text:
+ text = {}
+ self.result.update({col: {"email": text}})
+
+ def set_phone_number(self, col, text=None):
+ """
+ phone_number configuration
+ :param col: column name
+ :param text: page text. If no text is given, for database only.
+ :return:
+ """
+ if not text:
+ text = {}
+ self.result.update({col: {"phone_number": text}})
+
+ def clear(self):
+ """
+ Clear result
+ :return:
+ """
+ self.result.clear()

request_notion.py

@@ -0,0 +1,47 @@
+import json
+
+import requests
+
+from __init__ import NOTION_VERSION
+
+class Request:
+ def __init__(self, url, integrations_token):
+ """
+ init
+ :param url: Notion API URL
+ :param integrations_token: Notion Internal Integration Token
+ """
+ self.NOTION_KEY = integrations_token
+ self.HEADER = {"Authorization": f"Bearer {self.NOTION_KEY}",
+ "Content-Type": "application/json",
+ "Notion-Version": NOTION_VERSION}
+ self.url = url
+
+ def call_api_post(self, url, body):
+ """
+ request post
+ :param url:
+ :param body:
+ :return:
+ """
+ r = requests.post(url, data=json.dumps(body), headers=self.HEADER).json()
+ return r
+
+ def call_api_get(self, url):
+ """
+ request get
+ :param url:
+ :return:
+ """
+ r = requests.get(url, headers=self.HEADER).json()
+ return r
+
+ def call_api_patch(self, url, body):
+ """
+ request patch
+ :param url:
+ :param body:
+ :return:
+ """
+ r = requests.patch(url, data=json.dumps(body), headers=self.HEADER).json()
+ return r