fix: re-add pass validator (#213)

api still does not return F ranks
NiceAesth · Jan 30, 2024 · 49ec318 · 49ec318
1 parent bfe1d3a
commit 49ec318
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/aiosu/models/lazer.py b/aiosu/models/lazer.py
@@ -222,3 +222,10 @@ def mode(self) -> Gamemode:
     @cached_property
     def mods_str(self) -> str:
         return "".join(str(mod) for mod in self.mods)
+
+    @model_validator(mode="before")
+    @classmethod
+    def _fail_rank(cls, values: dict[str, object]) -> dict[str, object]:
+        if not values["passed"]:
+            values["rank"] = "F"
+        return values
diff --git a/aiosu/models/score.py b/aiosu/models/score.py
@@ -210,6 +210,13 @@ def completion(self) -> Optional[float]:
 
         return calculate_score_completion(self.mode, self.statistics, self.beatmap)
 
+    @model_validator(mode="before")
+    @classmethod
+    def _fail_rank(cls, values: dict[str, object]) -> dict[str, object]:
+        if not values["passed"]:
+            values["rank"] = "F"
+        return values
+
     async def request_beatmap(self, client: v1.Client) -> None:
         r"""For v1 Scores: requests the beatmap from the API and sets it.