Convert Version to a dataclass

AA-Turner · AA-Turner · commit 23b228e722e3 · 2025-04-12T04:52:00.000+01:00
diff --git a/build_docs.py b/build_docs.py
@@ -58,7 +58,6 @@
 import sys
 from bisect import bisect_left as bisect
 from contextlib import contextmanager, suppress
-from functools import total_ordering
 from pathlib import Path
 from string import Template
 from time import perf_counter, sleep
@@ -103,11 +102,23 @@ def __reversed__(self) -> Iterator[Version]:
 
     @classmethod
     def from_json(cls, data: dict) -> Versions:
-        versions = sorted(
-            [Version.from_json(name, release) for name, release in data.items()],
-            key=Version.as_tuple,
-        )
-        return cls(versions)
+        """Load versions from the devguide's JSON representation."""
+        permitted = ", ".join(sorted(Version.STATUSES | Version.SYNONYMS.keys()))
+
+        versions = []
+        for name, release in data.items():
+            branch = release["branch"]
+            status = release["status"]
+            status = Version.SYNONYMS.get(status, status)
+            if status not in Version.STATUSES:
+                msg = (
+                    f"Saw invalid version status {status!r}, "
+                    f"expected to be one of {permitted}."
+                )
+                raise ValueError(msg)
+            versions.append(Version(name=name, branch_or_tag=branch, status=status))
+
+        return cls(sorted(versions, key=Version.as_tuple))
 
     def filter(self, branches: Sequence[str] = ()) -> Sequence[Version]:
         """Filter the given versions.
@@ -143,10 +154,14 @@ def setup_indexsidebar(self, current: Version, dest_path: Path) -> None:
         dest_path.write_text(rendered_template, encoding="UTF-8")
 
 
-@total_ordering
+@dataclasses.dataclass(frozen=True, kw_only=True, slots=True)
 class Version:
     """Represents a CPython version and its documentation build dependencies."""
 
+    name: str
+    branch_or_tag: str | None
+    status: str
+
     STATUSES = {"EOL", "security-fixes", "stable", "pre-release", "in development"}
 
     # Those synonyms map branch status vocabulary found in the devguide
@@ -159,19 +174,6 @@ class Version:
         "prerelease": "pre-release",
     }
 
-    def __init__(
-        self, name: str, *, status: str, branch_or_tag: str | None = None
-    ) -> None:
-        status = self.SYNONYMS.get(status, status)
-        if status not in self.STATUSES:
-            raise ValueError(
-                "Version status expected to be one of: "
-                f"{', '.join(self.STATUSES | set(self.SYNONYMS.keys()))}, got {status!r}."
-            )
-        self.name = name
-        self.branch_or_tag = branch_or_tag
-        self.status = status
-
     def __repr__(self) -> str:
         return f"Version({self.name})"
 
@@ -181,11 +183,6 @@ def __eq__(self, other: Version) -> bool:
     def __gt__(self, other: Version) -> bool:
         return self.as_tuple() > other.as_tuple()
 
-    @classmethod
-    def from_json(cls, name: str, values: dict) -> Version:
-        """Loads a version from devguide's json representation."""
-        return cls(name, status=values["status"], branch_or_tag=values["branch"])
-
     @property
     def requirements(self) -> list[str]:
         """Generate the right requirements for this version.
