Merge pull request #28 from kevinjalbert/add-ability-to-update-a-block

Author: kevinjalbert

server/README.md

@@ -14,6 +14,7 @@ As of right now the following endpoints exist:
 
 ```python
 @app.route('/blocks/<block_id>/append', methods=['POST'])
+@app.route('/blocks/<block_id>/update', methods=['PUT'])
 @app.route('/blocks/<block_id>/view', methods=['GET'])
 @app.route('/collections/<collection_id>/<view_id>/append', methods=['POST'])
 @app.route('/collections/<collection_id>/<view_id>/view', methods=['GET'])

server/src/api.py

@@ -46,6 +46,19 @@ def block_append(notion_token, block_id):
         return jsonify(error=str(error)), 500
 
 
+@app.route('/blocks/<block_id>/update', methods=['PUT'])
+@token_required
+def block_update(notion_token, block_id):
+    try:
+        notion_api = NotionApi(notion_token)
+
+        block = notion_api.block_update(block_id, request.json['text'])
+
+        return jsonify(block_id=block.id), 200
+    except Exception as error:
+        return jsonify(error=str(error)), 500
+
+
 @app.route('/blocks/<block_id>/view', methods=['GET'])
 @token_required
 def block_view(notion_token, block_id):

shared/notionscripts/notion_api.py

@@ -32,6 +32,13 @@ def block_append(self, block_id, text):
 
         return block.children.add_new(TextBlock, title=text)
 
+    def block_update(self, block_id, text):
+        block = self.client().get_block(block_id)
+
+        block.title = text
+
+        return block
+
     def collection_view_content(self, collection_id, view_id):
         collection_view = self.__collection_view(collection_id, view_id)
         results = collection_view.default_query().execute()

shared/notionscripts/tests/test_notion_api.py

@@ -4,7 +4,7 @@
 
 from notion.block import TextBlock
 
-import pytest
+import pytest  # noqa, F401
 
 
 def test_block_content(notion_token):
@@ -17,7 +17,6 @@ def test_block_content(notion_token):
     assert content == "test get block content"
 
 
-@pytest.mark.focus
 def test_block_append(notion_token):
     notion_api = NotionApi(token=notion_token)
     test_page = get_test_page()
@@ -29,6 +28,17 @@ def test_block_append(notion_token):
     assert new_block.parent.id == block.id
 
 
+def test_block_update(notion_token):
+    notion_api = NotionApi(token=notion_token)
+    test_page = get_test_page()
+
+    block = test_page.children.add_new(TextBlock, title="test block update")
+    updated_block = notion_api.block_update(block.id, "test block has been updated")
+
+    assert updated_block.title == "test block has been updated"
+    assert updated_block.id == block.id
+
+
 def test_collection_view_content(notion_token):
     notion_api = NotionApi(token=notion_token)