Add unknown to all models that extend BaseEnum, rewrite _missing_

Author: Sharpienero

wom/enums.py

@@ -59,6 +59,15 @@ def __eq__(self, other: object) -> bool:
     def __hash__(self) -> int:
         return hash(self.value)
 
+    @classmethod
+    def _missing_(cls, value: object) -> BaseEnum:
+        print(
+            f"{value!r} is not a valid {cls.__name__} variant. "
+            "Please report this issue on github at https://github.com/Jonxslays/wom.py/issues/new",
+            file=sys.stderr,
+        )
+        return cls.Unknown  # type: ignore
+
     @classmethod
     def at_random(cls: t.Type[T]) -> T:
         """Generates a random variant of this enum.
@@ -68,15 +77,6 @@ def at_random(cls: t.Type[T]) -> T:
         """
         return random.choice(tuple(cls))
 
-    @classmethod
-    def _missing_(cls: t.Type[T], value: object) -> t.Optional[T]:
-        print(
-            f"{value!r} is not a valid {cls.__name__} variant. "
-            "Please report this issue on github at https://github.com/Jonxslays/wom.py/issues/new",
-            file=sys.stderr,
-        )
-        return cls.__getitem__("Unknown")
-
 
 class Period(BaseEnum):
     """A period of time used by the API."""
@@ -215,7 +215,7 @@ class Metric(BaseEnum):
     Unknown = "unknown"
 
 
-ComputedMetrics: t.FrozenSet[Metric] = frozenset({Metric.Ehp, Metric.Ehb})
+ComputedMetrics: t.FrozenSet[Metric] = frozenset({Metric.Ehp, Metric.Ehb, Metric.Unknown})
 """Set containing all the types of computed metrics."""
 
 Skills: t.FrozenSet[Metric] = frozenset(

wom/models/competitions/enums.py

@@ -33,6 +33,7 @@ class CompetitionType(BaseEnum):
 
     Classic = "classic"
     Team = "team"
+    Unknown = "unknown"
 
 
 class CompetitionStatus(BaseEnum):
@@ -41,6 +42,7 @@ class CompetitionStatus(BaseEnum):
     Upcoming = "upcoming"
     Ongoing = "ongoing"
     Finished = "finished"
+    Unknown = "unknown"
 
 
 class CompetitionCSVTableType(BaseEnum):
@@ -49,3 +51,4 @@ class CompetitionCSVTableType(BaseEnum):
     Participants = "participants"
     Team = "team"
     Teams = "teams"
+    Unknown = "unknown"

wom/models/groups/enums.py

@@ -32,6 +32,7 @@ class GroupActivityType(BaseEnum):
     ChangedRole = "changed_role"
     Joined = "joined"
     Left = "left"
+    Unknown = "unknown"
 
 
 class GroupRole(BaseEnum):
@@ -306,3 +307,4 @@ class GroupRole(BaseEnum):
     Zarosian = "zarosian"
     Zealot = "zealot"
     Zenyte = "zenyte"
+    Unknown = "unknown"

wom/models/names/enums.py

@@ -32,6 +32,7 @@ class NameChangeStatus(BaseEnum):
     Pending = "pending"
     Approved = "approved"
     Denied = "denied"
+    Unknown = "unknown"
 
 
 class NameChangeReviewReason(BaseEnum):
@@ -42,3 +43,4 @@ class NameChangeReviewReason(BaseEnum):
     TransitionTooLong = "transition_period_too_long"
     ExcessiveGains = "excessive_gains"
     TotalLevelTooLow = "total_level_too_low"
+    Unknown = "unknown"

wom/models/players/enums.py

@@ -53,6 +53,7 @@ class PlayerBuild(BaseEnum):
     Def1 = "def1"
     Hp10 = "hp10"
     F2pLvl3 = "f2p_lvl3"
+    Unknown = "unknown"
 
 
 class PlayerStatus(BaseEnum):
@@ -63,6 +64,7 @@ class PlayerStatus(BaseEnum):
     Flagged = "flagged"
     Archived = "archived"
     Banned = "banned"
+    Unknown = "unknown"
 
 
 class AchievementMeasure(BaseEnum):
@@ -73,6 +75,7 @@ class AchievementMeasure(BaseEnum):
     Kills = "kills"
     Score = "score"
     Value = "value"
+    Unknown = "unknown"
 
 
 class Country(BaseEnum):
@@ -327,3 +330,4 @@ class Country(BaseEnum):
     Za = "ZA"
     Zm = "ZM"
     Zw = "ZW"
+    Unknown = "unknown"