Tweaks to get_backlinks to handle children referencing parents

Author: jamalex

notion/block.py

@@ -550,9 +550,16 @@ def get_backlinks(self):
             "getBacklinksForBlock",
             {"blockId": self.id},
         ).json()
-        if "backlinks" not in data:
-            return None
-        return [self._client.get_block(block.get("mentioned_from").get("block_id")) for block in data.get("backlinks")]
+        backlinks = []
+        for block in data.get("backlinks") or []:
+            mention = block.get("mentioned_from")
+            if not mention:
+                continue
+            block_id = mention.get("block_id") or mention.get("parent_block_id")
+            if block_id:
+                print("Getting block", block_id)
+                backlinks.append(self._client.get_block(block_id))
+        return backlinks
 
 class BulletedListBlock(BasicBlock):
 

notion/smoke_test.py

@@ -37,6 +37,9 @@ def run_live_smoke_test(token_v2, parent_page_url_or_id):
     assert video in page.children.filter(VideoBlock)
     assert col_list not in page.children.filter(VideoBlock)
 
+    # check that the parent does not yet consider this page to be backlinking
+    assert page not in parent_page.get_backlinks()
+
     page.children.add_new(SubheaderBlock, title="A link back to where I came from:")
     alias = page.children.add_alias(parent_page)
     assert alias.is_alias
@@ -48,6 +51,9 @@ def run_live_smoke_test(token_v2, parent_page_url_or_id):
         ),
     )
 
+    # check that the parent now knows about the backlink
+    assert page in parent_page.get_backlinks()
+
     # ensure __repr__ methods are not breaking
     repr(page)
     repr(page.children)