add column block

Author: hoangv97

README.md

@@ -6,6 +6,10 @@ Python 3 tools for interacting with Notion API:
 
 - Relational database wrapper
 
+## Changelog
+
+- 11/11: Add column list, column block
+
 ## Installation
 
 `pip install notiondb`

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="notiondb",
-    version="1.0.0",
+    version="1.0.1",
     author="Viet Hoang",
     author_email="[email protected]",
     description="Python 3 tools for interacting with Notion API",

src/notiondb/block.py

@@ -111,6 +111,28 @@ def language(self):
         return self._language
 
 
+class ColumnBlock(BaseBlock):
+  
+    def __init__(self, blocks: List[BaseBlock]):
+        super().__init__(value=None)
+        self._children = blocks
+
+    block_type = 'column'
+    
+    properties = ['children']
+
+    def children(self):
+        return [block.value for block in self._children]
+
+
+class ColumnListBlock(ColumnBlock):
+  
+    def __init__(self, columns: List[ColumnBlock]):
+        super().__init__(blocks=columns)
+
+    block_type = 'column_list'
+
+
 class ChildPageBlock(BaseBlock):
 
     block_type = 'child_page'

tests/test_database.py

@@ -164,6 +164,10 @@ def test_model_c_append_children(self):
             TableOfContentsBlock(),
             BreadcrumbBlock(),
             DividerBlock(),
+            ColumnListBlock(columns=[
+                ColumnBlock([ParagraphBlock('paragraph 1'), ParagraphBlock('paragraph 2')]),
+                ColumnBlock([ParagraphBlock('paragraph 3'), ParagraphBlock('paragraph 4')]),
+            ]),
             ParagraphBlock('paragraph 1'),
             ParagraphBlock('paragraph 2', link='https://developers.notion.com/reference/block'),
             HeadingOneBlock('heading 1'),