better-notion 2.1.3__py3-none-any.whl → 2.1.5__py3-none-any.whl

better_notion/utils/agents/schemas.py

@@ -423,8 +423,10 @@ class TaskSchema:
     def get_schema() -> Dict[str, Dict[str, Any]]:
         """Return Notion database schema for Tasks.
 
-        Note: "Dependencies" self-referential relation is NOT included here.
-        It will be added after the database is created via database.update().
+        Note: The following properties are NOT included here and will be added
+        after database creation via database.update():
+        - "Dependencies": Self-referential relation (requires database's own ID)
+        - "Related Work Issue": Relation to Work Issues (Work Issues created after Tasks)
         """
         return {
             "Title": PropertyBuilder.title("Title"),
@@ -462,8 +464,6 @@ class TaskSchema:
                     SelectOption.option("Low", "blue"),
                 ],
             ),
-            # Related Work Issue: Relation to Work Issues (will be updated after database creation)
-            "Related Work Issue": PropertyBuilder.relation("Related Work Issue", dual_property=False),
             "Estimated Hours": PropertyBuilder.number("Estimated Hours"),
             "Actual Hours": PropertyBuilder.number("Actual Hours"),
             "Assignee": PropertyBuilder.people("Assignee"),
@@ -522,7 +522,13 @@ class WorkIssueSchema:
 
     @staticmethod
     def get_schema() -> Dict[str, Dict[str, Any]]:
-        """Return Notion database schema for Work Issues."""
+        """Return Notion database schema for Work Issues.
+
+        Note: The following properties are NOT included here and will be added
+        after database creation via database.update():
+        - "Blocking Tasks": Relation to Tasks (added in post-processing)
+        - "Caused Incidents": Relation to Incidents (Incidents created after Work Issues)
+        """
         return {
             "Title": PropertyBuilder.title("Title"),
             "Project": PropertyBuilder.relation("Project"),
@@ -561,10 +567,6 @@ class WorkIssueSchema:
             "Proposed Solution": PropertyBuilder.text("Proposed Solution"),
             "Related Idea": PropertyBuilder.relation("Related Idea", dual_property=False),
             "Fix Tasks": PropertyBuilder.relation("Fix Tasks", dual_property=False),
-            # Blocking Tasks: Tasks blocked by this work issue (will be updated after database creation)
-            "Blocking Tasks": PropertyBuilder.relation("Blocking Tasks", dual_property=False),
-            # Caused Incidents: Incidents caused by this work issue (will be updated after database creation)
-            "Caused Incidents": PropertyBuilder.relation("Caused Incidents", dual_property=False),
         }
 
 
@@ -573,7 +575,11 @@ class IncidentSchema:
 
     @staticmethod
     def get_schema() -> Dict[str, Dict[str, Any]]:
-        """Return Notion database schema for Incidents."""
+        """Return Notion database schema for Incidents.
+
+        Note: "Root Cause Work Issue" is NOT included here and will be added
+        after database creation via database.update() for consistency.
+        """
         return {
             "Title": PropertyBuilder.title("Title"),
             "Project": PropertyBuilder.relation("Project"),
@@ -607,8 +613,6 @@ class IncidentSchema:
                 ],
             ),
             "Fix Task": PropertyBuilder.relation("Fix Task", dual_property=False),
-            # Root Cause Work Issue: Relation to Work Issues (will be updated after database creation)
-            "Root Cause Work Issue": PropertyBuilder.relation("Root Cause Work Issue", dual_property=False),
             "Root Cause": PropertyBuilder.text("Root Cause"),
             "Detected Date": PropertyBuilder.date("Detected Date"),
             "Resolved Date": PropertyBuilder.date("Resolved Date"),

better_notion/utils/agents/workspace.py

@@ -169,6 +169,14 @@ class WorkspaceInitializer:
         # Update cross-database relations that require both databases to exist
         await self._update_cross_database_relations()
 
+        # Add self-referential Dependencies relation to Tasks database
+        if "tasks" in self._database_ids:
+            await self._update_relation(
+                self._database_ids["tasks"],
+                "Dependencies",
+                self._database_ids["tasks"]  # Self-reference
+            )
+
         # Save workspace metadata
         self.save_database_ids()
 
@@ -249,7 +257,7 @@ class WorkspaceInitializer:
         )
 
     async def _create_tasks_db(self, parent: Page) -> None:
-        """Create Tasks database with relations to Versions (self-referencing for dependencies)."""
+        """Create Tasks database with relations to Versions."""
         schema = TaskSchema.get_schema()
 
         # Update relations with Versions database ID
@@ -259,8 +267,6 @@ class WorkspaceInitializer:
                 "versions"
             ]
 
-        # Tasks reference themselves for dependencies
-        # Will be updated after database creation
         db = await self._client.databases.create(
             parent=parent,
             title="Tasks",
@@ -269,9 +275,6 @@ class WorkspaceInitializer:
 
         self._database_ids["tasks"] = db.id
 
-        # Update self-referential relations
-        await self._update_self_relations(db.id)
-
         # Update Versions database with reverse relation
         await self._add_reverse_relation(
             self._database_ids["versions"],
@@ -397,50 +400,16 @@ class WorkspaceInitializer:
             logger.warning(f"Failed to delete database {database_id}: {str(e)}")
             # Continue with other deletions even if this one fails
 
-    async def _update_self_relations(self, database_id: str) -> None:
-        """Add self-referential Dependencies relation to Tasks database.
-
-        The Dependencies property cannot be included in the initial schema
-        because it requires the database's own ID. This method adds it
-        after the database is created.
-
-        Args:
-            database_id: ID of the Tasks database
-        """
-        # Get the current database schema
-        db = await self._client.databases.get(database_id)
-        schema = db.schema
-
-        # Add the Dependencies self-referential relation
-        # This creates a relation from Tasks to Tasks itself
-        schema["Dependencies"] = {
-            "relation": {
-                "database_id": database_id,  # Self-reference
-                "type": "dual_property",
-                "dual_property": {
-                    "synced_property_name": "Dependent Tasks",
-                    "synced_property_type": "relation"
-                }
-            }
-        }
-
-        # Update the database schema via API
-        await self._client._api._request(
-            "PATCH",
-            f"/databases/{database_id}",
-            json={"properties": schema}
-        )
-
-        logger.info(f"Added Dependencies self-referential relation to Tasks database {database_id}")
-
     async def _update_cross_database_relations(self) -> None:
         """Update cross-database relations after all databases are created.
 
-        This updates:
-        - Tasks: Related Work Issue -> Work Issues
-        - Work Issues: Blocking Tasks -> Tasks
-        - Work Issues: Caused Incidents -> Incidents
+        This adds:
+        - Tasks: Dependencies (self-referential), Related Work Issue -> Work Issues
+        - Work Issues: Blocking Tasks -> Tasks, Caused Incidents -> Incidents
         - Incidents: Root Cause Work Issue -> Work Issues
+
+        All these properties are added AFTER database creation because they
+        reference databases that may not exist yet during initial creation.
         """
         if "tasks" in self._database_ids:
             await self._update_relation(
@@ -476,7 +445,7 @@ class WorkspaceInitializer:
         property_name: str,
         target_database_id: Optional[str],
     ) -> None:
-        """Update a relation property to point to the target database.
+        """Add or update a relation property to point to the target database.
 
         Args:
             database_id: Database to update
@@ -491,20 +460,92 @@ class WorkspaceInitializer:
         db = await self._client.databases.get(database_id)
         schema = db.schema
 
-        # Update the relation property
+        # Add or update the relation property
+        # Some properties (like Dependencies, Related Work Issue) are added AFTER
+        # database creation, so they may not exist in the initial schema
         if property_name in schema:
+            # Property exists, just update the database_id
             schema[property_name]["relation"]["database_id"] = target_database_id
+        else:
+            # Property doesn't exist, ADD it
+            # This handles cases where the property couldn't be in the initial schema
+            if property_name == "Dependencies":
+                # Self-referential relation
+                schema[property_name] = {
+                    "relation": {
+                        "database_id": database_id,  # Self-reference
+                        "type": "dual_property",
+                        "dual_property": {
+                            "synced_property_name": "Dependent Tasks",
+                            "synced_property_type": "relation"
+                        }
+                    }
+                }
+            elif property_name == "Related Work Issue":
+                # Relation to Work Issues database
+                schema[property_name] = {
+                    "relation": {
+                        "database_id": target_database_id,
+                        "type": "dual_property",
+                        "dual_property": {
+                            "synced_property_name": "Related Tasks",
+                            "synced_property_type": "relation"
+                        }
+                    }
+                }
+            elif property_name == "Blocking Tasks":
+                # Relation from Work Issues to Tasks
+                schema[property_name] = {
+                    "relation": {
+                        "database_id": target_database_id,
+                        "type": "dual_property",
+                        "dual_property": {
+                            "synced_property_name": "Blocked By Work Issue",
+                            "synced_property_type": "relation"
+                        }
+                    }
+                }
+            elif property_name == "Caused Incidents":
+                # Relation from Work Issues to Incidents
+                schema[property_name] = {
+                    "relation": {
+                        "database_id": target_database_id,
+                        "type": "dual_property",
+                        "dual_property": {
+                            "synced_property_name": "Root Cause Work Issue",
+                            "synced_property_type": "relation"
+                        }
+                    }
+                }
+            elif property_name == "Root Cause Work Issue":
+                # Relation from Incidents to Work Issues
+                schema[property_name] = {
+                    "relation": {
+                        "database_id": target_database_id,
+                        "type": "dual_property",
+                        "dual_property": {
+                            "synced_property_name": "Caused Incidents",
+                            "synced_property_type": "relation"
+                        }
+                    }
+                }
+            else:
+                # Generic relation
+                schema[property_name] = {
+                    "relation": {
+                        "database_id": target_database_id,
+                        "dual_property": {}
+                    }
+                }
 
-            # Update the database schema via API
-            await self._client._api._request(
-                "PATCH",
-                f"/databases/{database_id}",
-                json={"properties": schema}
-            )
+        # Update the database schema via API
+        await self._client._api._request(
+            "PATCH",
+            f"/databases/{database_id}",
+            json={"properties": schema}
+        )
 
-            logger.info(f"Updated {property_name} relation to {target_database_id}")
-        else:
-            logger.warning(f"Property {property_name} not found in database {database_id}")
+        logger.info(f"Added/Updated {property_name} relation to {target_database_id}")
 
     def save_database_ids(self, path: Optional[Path] = None) -> None:
         """Save workspace metadata (database IDs and workspace info) to config file.

{better_notion-2.1.3.dist-info → better_notion-2.1.5.dist-info}/METADATA

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

@@ -130,11 +130,11 @@ better_notion/utils/agents/dependency_resolver.py,sha256=PfHHDIQztGih4LwylMb0_My
 better_notion/utils/agents/metadata.py,sha256=thQSXfYx9mWgmBud8HtlNTLr5SwH6E_O1AjSNSnMFoo,6614
 better_notion/utils/agents/project_context.py,sha256=aJlzy5H2rL4sAfW2jHL_3K2VkBJ4ihUhCRVolkpuO78,7477
 better_notion/utils/agents/rbac.py,sha256=8ZA8Y7wbOiVZDbpjpH7iC35SZrZ0jl4fcJ3xWCm3SsE,11820
-better_notion/utils/agents/schemas.py,sha256=pO4MiSaWzSOCZu6J3uZPH7mtMVpE2H3iDHzd6WH0wXc,22926
+better_notion/utils/agents/schemas.py,sha256=2QF71tL-lt_n7u-oizVDMj4nd29fl8uUbhMKTWxNzr4,22721
 better_notion/utils/agents/state_machine.py,sha256=xUBEeDTbU1Xq-rsRo2sbr6AUYpYrV9DTHOtZT2cWES8,6699
-better_notion/utils/agents/workspace.py,sha256=W9MKf4b2fvoGCKHp8EuZROnr1tfmR3lS5GrmCuKrZQQ,22841
-better_notion-2.1.3.dist-info/METADATA,sha256=jQ5fHLFQ9DN3gfYKaiJuUMPIWjDhrC8VeIAQl_KxDMQ,11096
-better_notion-2.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-better_notion-2.1.3.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
-better_notion-2.1.3.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
-better_notion-2.1.3.dist-info/RECORD,,
+better_notion/utils/agents/workspace.py,sha256=zmJ2BRklEBC9fzB1AudA33h-yvFYvm1iSR2huc60oE4,25007
+better_notion-2.1.5.dist-info/METADATA,sha256=bF0A80PsoJc2QHKhiYMB14I89zMtUbrCz_LkzadHVGs,11096
+better_notion-2.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+better_notion-2.1.5.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
+better_notion-2.1.5.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
+better_notion-2.1.5.dist-info/RECORD,,