Fixed: relationship configuration

Author: mdelmans

python_notion_api/models/__init__.py

@@ -62,7 +62,6 @@
     NumberPropertyConfiguration,
     PeoplePropertyConfiguration,
     PhoneNumberPropertyConfiguration,
-    RelationConfigurationObject,
     RelationPropertyConfiguration,
     RollupConfigurationObject,
     RollupPropertyConfiguration,
@@ -220,7 +219,6 @@
     "PhoneNumberPropertyConfiguration",
     "FormulaConfigurationObject",
     "FormulaPropertyConfiguration",
-    "RelationConfigurationObject",
     "RelationPropertyConfiguration",
     "RollupConfigurationObject",
     "RollupPropertyConfiguration",

python_notion_api/models/configurations.py

@@ -131,16 +131,45 @@ class FormulaPropertyConfiguration(NotionPropertyConfiguration):
     formula: FormulaConfigurationObject
 
 
-class RelationConfigurationObject(BaseModel):
+class RelationPropertyConfiguration(NotionPropertyConfiguration):
+    @property
+    def _class_key_field(self):
+        return self.relation["type"]
+
+    _class_map = {
+        "single_property": "SinglePropertyConfiguration",
+        "dual_property": "DualPropertyConfiguration",
+    }
+
+    relation: Dict
+
+
+class SinglePropertyConfigurationObject(BaseModel):
     database_id: str
-    synced_property_name: Optional[str]
-    synced_property_id: Optional[str]
+    relation_type: str = typeField
+    single_property: Dict
 
 
-class RelationPropertyConfiguration(NotionPropertyConfiguration):
+class SinglePropertyConfiguration(RelationPropertyConfiguration):
+    _class_key_field = None
+
+    relation: SinglePropertyConfigurationObject
+
+
+class SyncedPropertyConfigurationObject(BaseModel):
+    synced_property_id: str
+    synced_property_name: str
+
+
+class DualPropertyConfigurationObject(BaseModel):
+    database_id: str
+    dual_property: SyncedPropertyConfigurationObject
+
+
+class DualPropertyConfiguration(RelationPropertyConfiguration):
     _class_key_field = None
 
-    relation: RelationConfigurationObject
+    relation: DualPropertyConfigurationObject
 
 
 class RollupConfigurationObject(BaseModel):