Implement reset_rank class method to set rank to 0

Author: tomschr

src/docbuild/models/manifest.py

@@ -118,16 +118,10 @@ def _increment_rank() -> int:
     rank: int = Field(default_factory=_increment_rank)
     translations: list[CategoryTranslation] = Field(default_factory=list)
 
-    # @model_validator(mode="before")
-    # @classmethod
-    # def prevent_manual_rank(cls, data: Any) -> Any:
-    #     """Prevent setting the rank attribute as it is set automatically."""
-    #     # If 'rank' is in the input data, it means the user tried to set it manually
-    #     if isinstance(data, dict) and "rank" in data:
-    #         raise ValueError(
-    #             "The 'rank' field is automatic and cannot be set manually."
-    #         )
-    #     return data
+    @classmethod
+    def reset_rank(cls: type[Self]) -> None:
+        """Reset the rank counter."""
+        cls._current_rank = 0
 
     @classmethod
     def from_xml_node(

tests/models/test_manifest.py

@@ -160,7 +160,7 @@ def test_category_from_xml_node() -> None:
     """
     node = etree.fromstring(doc, parser=None)
     # Reset class variable for predictable rank
-    Category._current_rank = 0
+    Category.reset_rank()
     models = list(Category.from_xml_node(node))
 
     assert len(models) == 4
@@ -181,7 +181,7 @@ def test_category_from_xml_node() -> None:
 
 def test_category_rank() -> None:
     # Just to be sure, we reset the current rank:
-    Category._current_rank = 0
+    Category.reset_rank()
     for idx, i in enumerate(["A", "B", "C"], 1):
         cat = Category(id=i, translations=[])
         serizalized = cat.model_dump()