Merge pull request #5856 from opsmill/dga-202502060-core-schema

Split core schema into multiple files

Author: dgarros

backend/infrahub/core/schema/attribute_schema.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import enum
+from enum import Enum
 from typing import TYPE_CHECKING, Any, Optional, Union
 
 from pydantic import field_validator, model_validator
@@ -36,6 +37,13 @@ def is_relationship(self) -> bool:
     def is_deprecated(self) -> bool:
         return bool(self.deprecation)
 
+    def to_dict(self) -> dict:
+        data = self.model_dump(exclude_unset=True, exclude_none=True, exclude_defaults=True)
+        for field_name, value in data.items():
+            if isinstance(value, Enum):
+                data[field_name] = value.value
+        return data
+
     @field_validator("kind")
     @classmethod
     def kind_options(cls, v: str) -> str:

backend/infrahub/core/schema/basenode_schema.py

@@ -4,6 +4,7 @@
 import keyword
 import os
 from dataclasses import asdict, dataclass
+from enum import Enum
 from typing import TYPE_CHECKING, Any, Callable, Iterable, Literal, Optional, Union, overload
 
 from infrahub_sdk.utils import compare_lists, intersection
@@ -62,6 +63,17 @@ def __hash__(self) -> int:
         Be careful hash generated from hash() have a salt by default and they will not be the same across run"""
         return hash(self.get_hash())
 
+    def to_dict(self) -> dict:
+        data = self.model_dump(
+            exclude_unset=True, exclude_none=True, exclude_defaults=True, exclude={"attributes", "relationships"}
+        )
+        for field_name, value in data.items():
+            if isinstance(value, Enum):
+                data[field_name] = value.value
+        data["attributes"] = [attr.to_dict() for attr in self.attributes]
+        data["relationships"] = [rel.to_dict() for rel in self.relationships]
+        return data
+
     def get_hash(self, display_values: bool = False) -> str:
         """Extend the Hash Calculation to account for attributes and relationships."""