database: unstable create & insert api

.gitignore

@@ -2,3 +2,4 @@ build
 dist
 pywolai.egg-info
 pywolai/__pycache__/
+.idea

pywolai/api.py

@@ -1,9 +1,10 @@
 from typing import List
 import requests
 import json
-from pywolai.enums import ErrorCode, BlockTypes
+from pywolai.enums import ErrorCode, BlockTypes, RequestMethod, RequestAddr
 from pywolai.block import Block
 from pywolai.block_format import BlockFormat
+from pywolai.database_format import DatabaseFormat
 
 def safe_request(url, method, params=None, data=None, headers=None):
     """安全请求
@@ -176,6 +177,25 @@ def create_blocks(self, parent_id:str, blocks:List[Block] ):
 
         return data
 
-
-    # Database
-    # TODO: Database API
+    def get_database(self, database_id:str) -> DatabaseFormat:
+        """
+        :param database_id: 数据库ID
+        :return 序列化后的数据库实例
+        """
+        url = RequestAddr.DATABASE_GET % (self._base_url, database_id)
+
+        data = safe_request(url, RequestMethod.GET, headers={"Authorization": self.get_token()})
+        return DatabaseFormat(**data)
+
+    def insert_database(self, database_id:str, value:[{str:any}]):
+        """
+        :param database_id: 数据库ID
+        :return:
+        """
+        url = RequestAddr.DATABASE_INSERT % (self._base_url, database_id)
+        data = {
+            "rows": value
+        }
+
+        data = safe_request(url, "POST", data = data, headers={"Authorization": self.get_token()})
+        return data

pywolai/database_format.py

@@ -0,0 +1,62 @@
+from pywolai.enums import BlockTypes
+
+
+class DatabaseRowDataItemFormat:
+    """
+    Database format column
+    """
+    def __init__(
+            self,
+            type: BlockTypes,
+            value: str,
+    ):
+        self.id = id
+        self.type = type
+        self.value = value
+
+
+class DatabaseRowDataFormat:
+    """
+    Database format column
+    """
+    def __init__(
+            self,
+            标题: DatabaseRowDataItemFormat,
+            标签: DatabaseRowDataItemFormat,
+            字段名: DatabaseRowDataItemFormat,
+    ):
+        self.id = id
+        self.标题 = 标题
+        self.标签 = 标签
+        self.字段名 = 字段名
+
+
+
+class DatabaseRowFormat:
+    """
+    Database format row
+    """
+    def __init__(
+            self,
+            page_id: str,
+            data: DatabaseRowDataFormat,
+    ):
+        self.id = id
+        self.page_id = page_id
+        self.data = data
+
+
+class DatabaseFormat:
+    """
+    Database format
+    """
+    def __init__(
+            self,
+            column_order: [str],
+            rows: [DatabaseRowFormat],
+            **kwargs
+    ):
+        self.column_order = column_order
+        self.rows = rows
+        self.other_fields = kwargs
+

pywolai/enums/__init__.py

@@ -7,3 +7,5 @@
 from pywolai.enums.rich_text import RichText
 
 from pywolai.enums.error_code import ErrorCode
+
+from pywolai.enums.request_enum import RequestMethod, RequestAddr

pywolai/enums/request_enum.py

@@ -0,0 +1,10 @@
+class RequestAddr:
+    DATABASE_GET = '%s/databases/%s'
+    DATABASE_INSERT = '%s/databases/%s/rows'
+
+
+class RequestMethod:
+    GET = 'GET'
+    POST = 'POST'
+    PUT = 'PUT'
+    DELETE = 'DELETE'