Move to _missing_

Author: Sharpienero

tests/test_enums.py

@@ -57,8 +57,8 @@ def test_at_random(choice: mock.MagicMock) -> None:
 
 def test_base_enum_missing() -> None:
     """A test for the BaseEnum.__missing__ method."""
-    assert wom.Metric["new_fake_metric"] == wom.Metric.Unknown
-    assert wom.Metric["another_fake_metric"] == wom.Metric.Unknown
+    assert wom.Metric("new_fake_metric") == wom.Metric.Unknown
+    assert wom.Metric("another_fake_metric") == wom.Metric.Unknown
     assert wom.Metric.Unknown.value == "unknown"
     assert wom.Metric.Unknown != wom.Metric.Vardorvis
     assert wom.Metric.Attack.value == "attack"

wom/enums.py

@@ -27,7 +27,6 @@
 import sys
 import typing as t
 from enum import Enum
-from enum import EnumMeta
 
 T = t.TypeVar("T", bound="BaseEnum")
 
@@ -42,20 +41,7 @@
 )
 
 
-class MetaEnum(EnumMeta):
-    def __getitem__(cls: MetaEnum, name: str) -> t.Any:
-        try:
-            return super().__getitem__(name)
-        except KeyError:
-            print(
-                f"{name!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 BaseEnum(Enum, metaclass=MetaEnum):
+class BaseEnum(Enum):
     """The base enum all library enums inherit from."""
 
     def __str__(self) -> str:
@@ -82,6 +68,15 @@ 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."""