Support adding a row from a template

- Change is a fork of jamalex#266
  with some changes

Author: yashgorana

notion/client.py

@@ -9,6 +9,7 @@
 from requests.adapters import HTTPAdapter
 from requests.packages.urllib3.util.retry import Retry
 from getpass import getpass
+from time import sleep
 
 from .block import Block, BLOCK_TYPES
 from .collection import (
@@ -178,6 +179,29 @@ def get_block(self, url_or_id, force_refresh=False):
             block_class = BLOCK_TYPES.get(block.get("type", ""), Block)
         return block_class(self, block_id)
 
+    def duplicate_block(self, source_block_id, target_block_id, wait_interval=1, wait_tries=10):
+        """
+        Duplicate a block
+        """
+
+        data = {
+            "task": {
+                "eventName": "duplicateBlock",
+                "request": {
+                    "sourceBlockId": source_block_id,
+                    "targetBlockId": target_block_id,
+                    "appendContentOnly": True,
+                },
+            }
+        }
+        response = self.post("enqueueTask", data).json()
+        task_id = response.get("taskId")
+
+        if task_id:
+            self.wait_for_task(task_id, interval=wait_interval, tries=wait_tries)
+
+        return True
+
     def get_collection(self, collection_id, force_refresh=False):
         """
         Retrieve an instance of Collection that maps to the collection identified by the ID passed in.
@@ -403,6 +427,42 @@ def create_record(self, table, parent, **kwargs):
 
         return record_id
 
+    def get_task_status(self, task_id):
+        """
+        Get a status of a single task
+        """
+        data = self.post("getTasks", dict(taskIds=[task_id])).json()
+
+        results = data.get("results")
+        if results is None:
+            return None
+
+        if not results:
+            # Notion does not know about such a task
+            print("Invalid task ID.")
+            return None
+
+        try:
+            task = results.pop()
+            return task.get("state", None)
+        except IndexError:
+            logger.error("There is no task {}".format(results))
+            return None
+
+    def wait_for_task(self, task_id, interval=1, tries=10):
+        """
+        Wait for a task by looping 'tries' times ever 'interval' seconds.
+        The 'interval' parameter can be used to specify milliseconds using double (e.g 0.75).
+        """
+        for i in range(tries):
+            state = self.get_task_status(task_id)
+            if state in ["not_started", "in_progress"]:
+                sleep(interval)
+            elif state == "success":
+                return state
+
+        logger.debug("Task takes more time than expected. Specify 'interval' or 'tries' to wait more.")
+
 
 class Transaction(object):
 

notion/collection.py

@@ -197,14 +197,24 @@ def get_schema_property(self, identifier):
                 return prop
         return None
 
-    def add_row(self, update_views=True, **kwargs):
+    def add_row(self, update_views=True, source_block_id=None, source_block=None, **kwargs):
         """
         Create a new empty CollectionRowBlock under this collection, and return the instance.
+        Specify 'source_block' to create a row from a template block.
         """
 
         row_id = self._client.create_record("block", self, type="page")
         row = CollectionRowBlock(self._client, row_id)
 
+        if source_block:
+            # The source block can be either an ID or a URL
+            source_block = self._client.get_block(source_block)
+            source_block_id = source_block.id
+
+        # User wants to create a row from a template
+        if source_block_id:
+            self._client.duplicate_block(source_block_id, row_id)
+
         with self._client.as_atomic_transaction():
             for key, val in kwargs.items():
                 setattr(row, key, val)