allow setting of page ID

thrau · Mar 28, 2024 · a4f35f2 · a4f35f2
1 parent 9c3e77f
commit a4f35f2
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/notion_objects/__init__.py b/notion_objects/__init__.py
@@ -3,4 +3,4 @@
 from .properties import *
 from .values import *
 
-__version__ = "0.5.0"
+__version__ = "0.5.1"
diff --git a/notion_objects/objects.py b/notion_objects/objects.py
@@ -3,7 +3,7 @@
 from typing import Iterable, List
 
 from .encode import _ConverterMixin
-from .properties import ChangeTracker, Properties, Property, RootProperty
+from .properties import ChangeTracker, Id, Properties, Property, RootProperty
 
 
 class NotionClass(type):
@@ -90,6 +90,6 @@ def __str__(self):
 
 
 class Page(NotionObject):
-    id: str = RootProperty("id")
+    id: str = Id()
     created_time: datetime = RootProperty()
     last_edited_time: datetime = RootProperty()
diff --git a/notion_objects/properties.py b/notion_objects/properties.py
@@ -134,6 +134,16 @@ def get(self, field: str, obj: dict):
         return value
 
 
+class Id(RootProperty[str]):
+    def __init__(self):
+        super().__init__("id")
+
+    def __set__(self, instance, value):
+        if self.object_locator:
+            instance = getattr(instance, self.object_locator)
+        instance[self.field] = value
+
+
 class TitlePlainText(Property[str]):
     def get(self, field: str, obj: dict) -> str:
         items = obj["properties"][field]["title"]