refactor: review ResultManager serialization

anta/models.py

@@ -13,7 +13,7 @@
 from string import Formatter
 from typing import TYPE_CHECKING, Any, Callable, ClassVar, Literal, TypeVar
 
-from pydantic import BaseModel, ConfigDict, ValidationError, create_model
+from pydantic import BaseModel, ConfigDict, Field, ValidationError, create_model
 
 from anta import GITHUB_SUGGESTION
 from anta.constants import KNOWN_EOS_ERRORS
@@ -385,7 +385,7 @@ class Input(BaseModel):
         """
 
         model_config = ConfigDict(extra="forbid")
-        result_overwrite: ResultOverwrite | None = None
+        result_overwrite: ResultOverwrite | None = Field(None, exclude=True)
         filters: Filters | None = None
 
         def __hash__(self) -> int:

anta/result_manager/__init__.py

@@ -5,13 +5,14 @@
 
 from __future__ import annotations
 
-import json
 import logging
 from collections import defaultdict
 from functools import cached_property
 from itertools import chain
 from typing import Any
 
+from pydantic import TypeAdapter
+
 from anta.result_manager.models import AntaTestStatus, TestResult
 
 from .models import CategoryStats, DeviceStats, TestStats
@@ -107,8 +108,9 @@ def __init__(self) -> None:
 
     def reset(self) -> None:
         """Create or reset the attributes of the ResultManager instance."""
-        self._result_entries: list[TestResult] = []
-        self.status: AntaTestStatus = AntaTestStatus.UNSET
+        self._result_entries = []
+        self._result_entries_ta = TypeAdapter(list[TestResult])
+        self.status = AntaTestStatus.UNSET
         self.error_status = False
 
         # Initialize the statistics attributes
@@ -135,12 +137,12 @@ def results(self, value: list[TestResult]) -> None:
     @property
     def dump(self) -> list[dict[str, Any]]:
         """Get a list of dictionary of the results."""
-        return [result.model_dump() for result in self._result_entries]
+        return self._result_entries_ta.dump_python(self._result_entries)
 
     @property
     def json(self) -> str:
         """Get a JSON representation of the results."""
-        return json.dumps(self.dump, indent=4)
+        return self._result_entries_ta.dump_json(self._result_entries, exclude_none=True, indent=4).decode()
 
     @property
     def device_stats(self) -> defaultdict[str, DeviceStats]:

anta/result_manager/models.py

@@ -10,7 +10,7 @@
 from enum import Enum
 from typing import Any
 
-from pydantic import BaseModel, SkipValidation, field_serializer
+from pydantic import BaseModel, SerializeAsAny
 
 
 class AntaTestStatus(str, Enum):
@@ -44,12 +44,7 @@ class BaseTestResult(BaseModel, ABC):
     description: str
     result: AntaTestStatus = AntaTestStatus.UNSET
     messages: list[str] = []
-    inputs: SkipValidation[BaseModel | None] = None
-
-    @field_serializer("inputs")
-    def serialize_inputs(self, inputs: BaseModel | None) -> dict[str, Any] | None:
-        """Serialize the inputs field to a dictionary."""
-        return inputs.model_dump(mode="json", serialize_as_any=True, exclude_none=True, exclude={"result_overwrite"}) if inputs else None
+    inputs: SerializeAsAny[BaseModel | None] = None
 
     def is_success(self, message: str | None = None) -> None:
         """Set status to success.