infrahub-server 1.2.0rc0__py3-none-any.whl → 1.2.1__py3-none-any.whl

infrahub/core/schema/relationship_schema.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Optional, Union
+from enum import Enum
+from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel
 
@@ -34,10 +35,17 @@ class RelationshipSchema(GeneratedRelationshipSchema):
     def is_deprecated(self) -> bool:
         return bool(self.deprecation)
 
+    def to_dict(self) -> dict:
+        data = self.model_dump(exclude_unset=True, exclude_none=True)
+        for field_name, value in data.items():
+            if isinstance(value, Enum):
+                data[field_name] = value.value
+        return data
+
     def get_class(self) -> type[Relationship]:
         return Relationship
 
-    def get_peer_schema(self, db: InfrahubDatabase, branch: Optional[Union[Branch, str]] = None) -> MainSchemaTypes:
+    def get_peer_schema(self, db: InfrahubDatabase, branch: Branch | str | None = None) -> MainSchemaTypes:
         return db.schema.get(name=self.peer, branch=branch, duplicate=False)
 
     @property
@@ -71,11 +79,11 @@ class RelationshipSchema(GeneratedRelationshipSchema):
         self,
         db: InfrahubDatabase,
         filter_name: str,
-        filter_value: Optional[Union[str, int, bool]] = None,
-        name: Optional[str] = None,  # noqa: ARG002
-        branch: Optional[Branch] = None,
+        filter_value: str | int | bool | None = None,
+        name: str | None = None,  # noqa: ARG002
+        branch: Branch | None = None,
         include_match: bool = True,
-        param_prefix: Optional[str] = None,
+        param_prefix: str | None = None,
         partial_match: bool = False,
         support_profiles: bool = False,  # noqa: ARG002
     ) -> tuple[list[QueryElement], dict[str, Any], list[str]]: