better-notion 2.1.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

better_notion/utils/agents/schemas.py

@@ -458,8 +458,10 @@ class TaskSchema:
                     SelectOption.option("Low", "blue"),
                 ],
             ),
-            # Note: Self-referential relations (Dependencies, Dependent Tasks) removed because
-            # they can't be created during initial database creation (need the database's own ID)
+            # Dependencies: Self-referential relation (will be updated after database creation)
+            "Dependencies": PropertyBuilder.relation("Dependencies", dual_property=False),
+            # 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"),
@@ -557,6 +559,10 @@ 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),
         }
 
 
@@ -599,6 +605,8 @@ 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

@@ -141,6 +141,9 @@ class WorkspaceInitializer:
                 logger.error(error_msg)
                 raise Exception(error_msg) from e
 
+        # Update cross-database relations that require both databases to exist
+        await self._update_cross_database_relations()
+
         # Save workspace metadata
         self.save_database_ids()
 
@@ -340,13 +343,95 @@ class WorkspaceInitializer:
         Note: The database was created with placeholder relations.
         This method updates them to point to themselves.
         """
-        # Get the database
+        # Get the current database schema
         db = await self._client.databases.get(database_id)
 
-        # Update Dependencies and Dependent Tasks to point to self
-        # Note: This would require updating the database schema
-        # via the Notion API, which may not support all updates
-        pass
+        # Update the Dependencies relation to point to itself
+        schema = db.schema
+        if "Dependencies" in schema:
+            schema["Dependencies"]["relation"]["database_id"] = database_id
+
+        # Update the database schema via API
+        await self._client._api._request(
+            "PATCH",
+            f"/databases/{database_id}",
+            json={"properties": schema}
+        )
+
+        logger.info(f"Updated self-referential relations for 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
+        - Incidents: Root Cause Work Issue -> Work Issues
+        """
+        if "tasks" in self._database_ids:
+            await self._update_relation(
+                self._database_ids["tasks"],
+                "Related Work Issue",
+                self._database_ids.get("work_issues")
+            )
+
+        if "work_issues" in self._database_ids:
+            await self._update_relation(
+                self._database_ids["work_issues"],
+                "Blocking Tasks",
+                self._database_ids.get("tasks")
+            )
+            await self._update_relation(
+                self._database_ids["work_issues"],
+                "Caused Incidents",
+                self._database_ids.get("incidents")
+            )
+
+        if "incidents" in self._database_ids:
+            await self._update_relation(
+                self._database_ids["incidents"],
+                "Root Cause Work Issue",
+                self._database_ids.get("work_issues")
+            )
+
+        logger.info("Updated all cross-database relations")
+
+    async def _update_relation(
+        self,
+        database_id: str,
+        property_name: str,
+        target_database_id: Optional[str],
+    ) -> None:
+        """Update a relation property to point to the target database.
+
+        Args:
+            database_id: Database to update
+            property_name: Name of the relation property
+            target_database_id: Target database ID (if None, skip update)
+        """
+        if not target_database_id:
+            logger.warning(f"Skipping {property_name} update: target database not found")
+            return
+
+        # Get the current database schema
+        db = await self._client.databases.get(database_id)
+        schema = db.schema
+
+        # Update the relation property
+        if property_name in schema:
+            schema[property_name]["relation"]["database_id"] = target_database_id
+
+            # 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}")
 
     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.0.dist-info → better_notion-2.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better-notion
-Version: 2.1.0
+Version: 2.1.1
 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.0.dist-info → better_notion-2.1.1.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=eHfGhY90FAPXA3E8qE6gP75dgNzn-9z5Ju1FMwBKnQQ,22120
+better_notion/utils/agents/schemas.py,sha256=EpAEevoth-DGWepGMubRFjdvHfe7zrQxi9i4CHDvem8,22946
 better_notion/utils/agents/state_machine.py,sha256=xUBEeDTbU1Xq-rsRo2sbr6AUYpYrV9DTHOtZT2cWES8,6699
-better_notion/utils/agents/workspace.py,sha256=Uy8bqLsT_VFGYAPoiQJNuCvGdjMceaSiVGbR9saMpoU,16558
-better_notion-2.1.0.dist-info/METADATA,sha256=CGwOskvV6KSl4YON4CUNgzo3W0TWLWfJ83Vd3aRa0Wk,11096
-better_notion-2.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-better_notion-2.1.0.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
-better_notion-2.1.0.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
-better_notion-2.1.0.dist-info/RECORD,,
+better_notion/utils/agents/workspace.py,sha256=-EzPLFPjaD9PCeLkSdu1ddRa9-RlHgFvLJ2qxFou7X8,19713
+better_notion-2.1.1.dist-info/METADATA,sha256=5hImQwg_VfumZ1dMkCtz1QEWUtcYr8ywTvlPlREvnKE,11096
+better_notion-2.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+better_notion-2.1.1.dist-info/entry_points.txt,sha256=D0bUcP7Z00Zyjxw7r2p29T95UrwioDO0aGDoHe9I6fo,55
+better_notion-2.1.1.dist-info/licenses/LICENSE,sha256=BAdN3JpgMY_y_fWqZSCFSvSbC2mTHP-BKDAzF5FXQAI,1069
+better_notion-2.1.1.dist-info/RECORD,,