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

better_notion/_api/properties/__init__.py

@@ -10,6 +10,7 @@ from better_notion._api.properties.number import Number
 from better_notion._api.properties.url import URL
 from better_notion._api.properties.email import Email
 from better_notion._api.properties.phone import Phone
+from better_notion._api.properties.relation import Relation
 
 __all__ = [
     "Property",
@@ -27,4 +28,5 @@ __all__ = [
     "URL",
     "Email",
     "Phone",
+    "Relation",
 ]

better_notion/_api/properties/relation.py

@@ -0,0 +1,27 @@
+"""Relation property builder."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from better_notion._api.properties.base import Property
+
+
+class Relation(Property):
+    """Builder for relation properties."""
+
+    def __init__(self, page_ids: list[str]) -> None:
+        """Initialize a relation property.
+
+        Args:
+            page_ids: List of related page IDs.
+        """
+        super().__init__("Relation")
+        self._page_ids = page_ids
+
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to Notion API format."""
+        return {
+            "type": "relation",
+            "relation": [{"id": page_id} for page_id in self._page_ids]
+        }

better_notion/_api/properties/rich_text.py

@@ -61,15 +61,20 @@ class RichText(Property):
         if self._code:
             annotations["code"] = True
 
-        rich_text_obj: dict[str, Any] = {
+        # Build the inner text object
+        inner_text_obj: dict[str, Any] = {
             "type": "text",
             "text": text_obj,
         }
 
         if annotations:
-            rich_text_obj["annotations"] = annotations
+            inner_text_obj["annotations"] = annotations
 
-        return rich_text_obj
+        # Return proper rich_text property format
+        return {
+            "type": "rich_text",
+            "rich_text": [inner_text_obj]
+        }
 
 
 class Text(Property):

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.3.dist-info → better_notion-1.8.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better-notion
-Version: 1.8.3
+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.3.dist-info → better_notion-1.8.5.dist-info}/RECORD

@@ -15,14 +15,15 @@ better_notion/_api/entities/comment.py,sha256=li4bZEA0Q4VwpeSSvg1QJRfIKZL_h79LNT
 better_notion/_api/entities/database.py,sha256=MsR7AjBihzgo6oGLubECJ0rmeFOxMxa3S-0oESAKAQg,1753
 better_notion/_api/entities/page.py,sha256=7Xi1IES6dE6gASB1RLHOGLvaUk5yYy98gkRgLHavgns,4595
 better_notion/_api/entities/user.py,sha256=EVjwvj3ksSoOVIbMlR6Ez3j0PtzNRchvTb8rr5aNJ-s,1145
-better_notion/_api/properties/__init__.py,sha256=fdnuzzh3BerMh0GZni-HPjqN_opQ0VXimCgT2-UemOk,926
+better_notion/_api/properties/__init__.py,sha256=WH4hLn4aQuPFFMjGE7N0iawidEcOHaIRrIAnWu2che4,1002
 better_notion/_api/properties/base.py,sha256=3lSkj50Zm9smlbSANx1ngrjP82ygXnfux3tG02aeFms,895
 better_notion/_api/properties/checkbox.py,sha256=DTJjq3KxOukoPlpQ7UB173Sf9WHDlND5-7QSwydL7Pc,691
 better_notion/_api/properties/date.py,sha256=E2rIMcgTMuPf8rS5AdT8M0UyYEDobJ8zJRYoBaFSR9s,2372
 better_notion/_api/properties/email.py,sha256=JaWHUNJoO837PqGGNPQ7fixG7al3U7i7Ipap7wTLOok,659
 better_notion/_api/properties/number.py,sha256=6XpDOpnU5D4_Fr7DOX7IoGOk4o9ggyMbcZua-UDL0Kk,671
 better_notion/_api/properties/phone.py,sha256=EDKKUku51RK86rPt92sbuobJgbhvPhuYe9IlfHzSnI4,678
-better_notion/_api/properties/rich_text.py,sha256=4pyHK56nGFRCoZ8Qla5wM8PsjYkU9-3GXL6VcTDr7LQ,3466
+better_notion/_api/properties/relation.py,sha256=e0VKMNhCyoxaqPkTGRZP4D22eNatDvej5Rz726FskpI,708
+better_notion/_api/properties/rich_text.py,sha256=yqdr-snhTQYAcaNs881-OptNkbz2-1N-eTg6fauRO80,3629
 better_notion/_api/properties/select.py,sha256=5NMzj6dykXiH6xCWx8xWP3eeyUvQICScdTPjJJOQZ_c,1347
 better_notion/_api/properties/title.py,sha256=SmcSXmiPtnM05hBH7EnJeJW7SYfGrOoPO9cIw1Zs-Jo,815
 better_notion/_api/properties/url.py,sha256=1_vHKHENevN1b8F3b6pYsZboQXGAPOfsEMgjCe6HXKM,632
@@ -117,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
@@ -132,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.3.dist-info/METADATA,sha256=KU9Yunh85byLx9y_lw6uNtP4UPmHlC6tzEOphJNQDe0,11096
-better_notion-1.8.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-better_notion-1.8.3.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
-better_notion-1.8.3.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
-better_notion-1.8.3.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,,