better-notion 1.8.4__py3-none-any.whl → 1.8.5__py3-none-any.whl

better_notion/plugins/official/agents_sdk/models.py

@@ -12,15 +12,82 @@ with caching support through the plugin system.
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, AsyncIterator
 
 from better_notion._sdk.base.entity import BaseEntity
 
 if TYPE_CHECKING:
     from better_notion._sdk.client import NotionClient
+    from better_notion._sdk.models.block import Block
+    from better_notion._sdk.models.database import Database
+    from better_notion._sdk.models.page import Page
 
 
-class Organization(BaseEntity):
+class DatabasePageEntityMixin:
+    """Mixin providing BaseEntity abstract method implementations for database pages.
+
+    This mixin provides parent() and children() implementations for entities
+    that are pages in databases (like Organization, Project, etc.).
+    """
+
+    async def parent(self) -> "Database | Page | None":
+        """Get parent object (database for entity pages).
+
+        Returns:
+            Parent Database or Page or None
+        """
+        from better_notion._sdk.models.database import Database
+
+        # Get parent from data
+        parent_data = self._data.get("parent")
+        if not parent_data:
+            return None
+
+        # Check cache
+        cached = self._cache_get("parent")
+        if cached:
+            return cached
+
+        # Database parent
+        if parent_data.get("type") == "database_id":
+            db_id = parent_data.get("database_id")
+            parent = await Database.get(db_id, client=self._client)
+            self._cache_set("parent", parent)
+            return parent
+
+        # Page parent
+        if parent_data.get("type") == "page_id":
+            from better_notion._sdk.models.page import Page
+            page_id = parent_data.get("page_id")
+            parent = await Page.get(page_id, client=self._client)
+            self._cache_set("parent", parent)
+            return parent
+
+        return None
+
+    async def children(self) -> AsyncIterator["Block"]:
+        """Iterate over child blocks.
+
+        Yields:
+            Block objects that are direct children
+        """
+        from better_notion._api.utils.pagination import AsyncPaginatedIterator
+        from better_notion._sdk.models.block import Block
+
+        async def _get_blocks(offset: str | None = None) -> dict[str, Any]:
+            return await self._client._request(
+                "GET",
+                f"/blocks/{self.id}/children",
+                params={"start_cursor": offset} if offset else None
+            )
+
+        iterator = AsyncPaginatedIterator(_get_blocks, lambda data: data.get("results", []))
+
+        async for block_data in iterator:
+            yield Block(self._client, block_data)
+
+
+class Organization(DatabasePageEntityMixin, BaseEntity):
     """
     Organization entity representing a company or team.
 
@@ -348,7 +415,7 @@ class Organization(BaseEntity):
         return [Project(self._client, page_data) for page_data in response.get("results", [])]
 
 
-class Project(BaseEntity):
+class Project(DatabasePageEntityMixin, BaseEntity):
     """
     Project entity representing a software project.
 
@@ -634,7 +701,7 @@ class Project(BaseEntity):
         return [Version(self._client, page_data) for page_data in response.get("results", [])]
 
 
-class Version(BaseEntity):
+class Version(DatabasePageEntityMixin, BaseEntity):
     """
     Version entity representing a project version/release.
 
@@ -874,7 +941,7 @@ class Version(BaseEntity):
         return [Task(self._client, page_data) for page_data in response.get("results", [])]
 
 
-class Task(BaseEntity):
+class Task(DatabasePageEntityMixin, BaseEntity):
     """
     Task entity representing a work task.
 
@@ -1190,7 +1257,7 @@ class Task(BaseEntity):
         return True
 
 
-class Idea(BaseEntity):
+class Idea(DatabasePageEntityMixin, BaseEntity):
     """
     Idea entity representing an improvement opportunity discovered during work.
 
@@ -1577,7 +1644,7 @@ class Idea(BaseEntity):
         )
 
 
-class WorkIssue(BaseEntity):
+class WorkIssue(DatabasePageEntityMixin, BaseEntity):
     """
     Work Issue entity representing a development-time problem.
 
@@ -1938,7 +2005,7 @@ class WorkIssue(BaseEntity):
         return idea
 
 
-class Incident(BaseEntity):
+class Incident(DatabasePageEntityMixin, BaseEntity):
     """
     Incident entity representing a production incident.
 

{better_notion-1.8.4.dist-info → better_notion-1.8.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better-notion
-Version: 1.8.4
+Version: 1.8.5
 Summary: A high-level Python SDK for the Notion API with developer experience in mind.
 Project-URL: Homepage, https://github.com/nesalia-inc/better-notion
 Project-URL: Documentation, https://github.com/nesalia-inc/better-notion#readme

{better_notion-1.8.4.dist-info → better_notion-1.8.5.dist-info}/RECORD

@@ -118,7 +118,7 @@ better_notion/plugins/official/agents_schema.py,sha256=NQRAJFoBAXRuxB9_9Eaf-4tVt
 better_notion/plugins/official/productivity.py,sha256=_-whP4pYA4HufE1aUFbIdhrjU-O9njI7xUO_Id2M1J8,8726
 better_notion/plugins/official/agents_sdk/__init__.py,sha256=luQBzZLsJ7fC5U0jFu8dfzMviiXj2SBZXcTohM53wkQ,725
 better_notion/plugins/official/agents_sdk/managers.py,sha256=0zMZPu63zhdyqiudO2gKsmM3YOJh0nFAR9FrMlwkV2A,31186
-better_notion/plugins/official/agents_sdk/models.py,sha256=qrgu6E80wJq_lLEHh3jVJfuJ8uwmBA8aypKeLP4P4E8,80341
+better_notion/plugins/official/agents_sdk/models.py,sha256=vWe1cut11VmQSd_vDG0ijs5EXYvYvQ1xgbkii61rL8Q,82919
 better_notion/plugins/official/agents_sdk/plugin.py,sha256=bs9O8Unv6SARGj4lBU5Gj9HGbLTUNqTacJ3RLUhdbI4,4479
 better_notion/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 better_notion/utils/helpers.py,sha256=HgFuUQlG_HzBOB0z2GA9RxPLoXgwRc0DIxa9Fg6C-Jk,2337
@@ -133,8 +133,8 @@ better_notion/utils/agents/rbac.py,sha256=8ZA8Y7wbOiVZDbpjpH7iC35SZrZ0jl4fcJ3xWC
 better_notion/utils/agents/schemas.py,sha256=eHfGhY90FAPXA3E8qE6gP75dgNzn-9z5Ju1FMwBKnQQ,22120
 better_notion/utils/agents/state_machine.py,sha256=xUBEeDTbU1Xq-rsRo2sbr6AUYpYrV9DTHOtZT2cWES8,6699
 better_notion/utils/agents/workspace.py,sha256=Uy8bqLsT_VFGYAPoiQJNuCvGdjMceaSiVGbR9saMpoU,16558
-better_notion-1.8.4.dist-info/METADATA,sha256=3fwjaBaqq3PPC4tSzn8Q4PJSJcW-yBRkqjw8Ic979iE,11096
-better_notion-1.8.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-better_notion-1.8.4.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
-better_notion-1.8.4.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
-better_notion-1.8.4.dist-info/RECORD,,
+better_notion-1.8.5.dist-info/METADATA,sha256=aP3q-b0_38YX28OaYWBxtCZsgrFiNuSeZHe71ct9j5Q,11096
+better_notion-1.8.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+better_notion-1.8.5.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
+better_notion-1.8.5.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
+better_notion-1.8.5.dist-info/RECORD,,