Implement helpers.is_full_page_or_database

Author: ramnes

notion_client/helpers.py

@@ -89,6 +89,13 @@ def is_full_database(response: Dict[Any, Any]) -> bool:
     return "title" in response
 
 
+def is_full_page_or_database(response: Dict[Any, Any]) -> bool:
+    """Return `true` if `response` is a full database or a full page."""
+    if response.get("object") == "database":
+        return is_full_database(response)
+    return is_full_page(response)
+
+
 def is_full_user(response: Dict[Any, Any]) -> bool:
     """Return `true` if response is a full user."""
     return "type" in response

tests/test_helpers.py

@@ -12,6 +12,7 @@
     is_full_comment,
     is_full_database,
     is_full_page,
+    is_full_page_or_database,
     is_full_user,
     iterate_paginated_api,
     pick,
@@ -132,6 +133,15 @@ def test_is_full_database(client, database_id):
     assert is_full_database(response)
 
 
+@pytest.mark.vcr()
+def test_is_full_page_or_database(client, database_id, page_id):
+    response = client.pages.retrieve(page_id=page_id)
+    assert is_full_page_or_database(response)
+
+    response = client.databases.retrieve(database_id=database_id)
+    assert is_full_page_or_database(response)
+
+
 @pytest.mark.vcr()
 def test_is_full_user(client):
     response = client.users.me()