forbid extra inputs in device inventory

aristanetworks · Nov 24, 2023 · ebb420d · ebb420d
1 parent f07f2e7
commit ebb420d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/anta/inventory/models.py b/anta/inventory/models.py
@@ -9,7 +9,7 @@
 from typing import List, Optional, Union
 
 # Need to keep List for pydantic in python 3.8
-from pydantic import BaseModel, IPvAnyAddress, IPvAnyNetwork, conint, constr
+from pydantic import BaseModel, ConfigDict, IPvAnyAddress, IPvAnyNetwork, conint, constr
 
 logger = logging.getLogger(__name__)
 
@@ -30,6 +30,8 @@ class AntaInventoryHost(BaseModel):
         disable_cache (bool): Disable cache per host. Defaults to False.
     """
 
+    model_config = ConfigDict(extra="forbid")
+
     name: Optional[str] = None
     host: Union[constr(pattern=RFC_1123_REGEX), IPvAnyAddress]  # type: ignore
     port: Optional[conint(gt=1, lt=65535)] = None  # type: ignore
@@ -47,6 +49,8 @@ class AntaInventoryNetwork(BaseModel):
         disable_cache (bool): Disable cache per network. Defaults to False.
     """
 
+    model_config = ConfigDict(extra="forbid")
+
     network: IPvAnyNetwork
     tags: Optional[List[str]] = None
     disable_cache: bool = False
@@ -63,6 +67,8 @@ class AntaInventoryRange(BaseModel):
         disable_cache (bool): Disable cache per range of hosts. Defaults to False.
     """
 
+    model_config = ConfigDict(extra="forbid")
+
     start: IPvAnyAddress
     end: IPvAnyAddress
     tags: Optional[List[str]] = None
@@ -79,6 +85,8 @@ class AntaInventoryInput(BaseModel):
         range (list[AntaInventoryRange],Optional): List of AntaInventoryRange objects for ranges.
     """
 
+    model_config = ConfigDict(extra="forbid")
+
     networks: Optional[List[AntaInventoryNetwork]] = None
     hosts: Optional[List[AntaInventoryHost]] = None
     ranges: Optional[List[AntaInventoryRange]] = None