add earthen classes

Bloodmallet · Aug 26, 2024 · 26f6e4b · 26f6e4b
1 parent d10b96b
commit 26f6e4b
Showing 1 changed file with 34 additions and 12 deletions.
diff --git a/simc_support/game_data/WowClass.py b/simc_support/game_data/WowClass.py
@@ -33,13 +33,24 @@ def __init__(
             self.translations = Language.Translation(translations=translations)
 
 
-_all_non_dracthyr_races = tuple(
-    [r for r in Race.RACES if r not in (Race.DRACTHYR_ALLIANCE, Race.DRACTHYR_HORDE)]
+_all_non_dracthyr_and_earthen_races = tuple(
+    [
+        r
+        for r in Race.RACES
+        if r
+        not in (
+            Race.DRACTHYR_ALLIANCE,
+            Race.DRACTHYR_HORDE,
+            Race.EARTHEN_ALLIANCE,
+            Race.EARTHEN_HORDE,
+        )
+    ]
 )
+_earthen = (Race.EARTHEN_ALLIANCE, Race.EARTHEN_HORDE)
 
 DEATHKNIGHT = WowClass(
     6,
-    _all_non_dracthyr_races,
+    _all_non_dracthyr_and_earthen_races,
     Language.EmptyTranslation(),
     "Death Knight",
     "death_knight",
@@ -80,11 +91,15 @@ def __init__(
     "evoker",
 )
 HUNTER = WowClass(
-    3, _all_non_dracthyr_races, Language.EmptyTranslation(), "Hunter", "hunter"
+    3,
+    _all_non_dracthyr_and_earthen_races + _earthen,
+    Language.EmptyTranslation(),
+    "Hunter",
+    "hunter",
 )
 MAGE = WowClass(
     8,
-    _all_non_dracthyr_races,
+    _all_non_dracthyr_and_earthen_races + _earthen,
     Language.EmptyTranslation(),
     "Mage",
     "mage",
@@ -116,7 +131,8 @@ def __init__(
         Race.MAGHARORC,
         Race.ZANDALARITROLL,
         Race.VULPERA,
-    ),
+    )
+    + _earthen,
     Language.EmptyTranslation(),
     "Monk",
     "monk",
@@ -132,21 +148,22 @@ def __init__(
         Race.BLOODELF,
         Race.TAUREN,
         Race.ZANDALARITROLL,
-    ),
+    )
+    + _earthen,
     Language.EmptyTranslation(),
     "Paladin",
     "paladin",
 )
 PRIEST = WowClass(
     5,
-    _all_non_dracthyr_races,
+    _all_non_dracthyr_and_earthen_races + _earthen,
     Language.EmptyTranslation(),
     "Priest",
     "priest",
 )
 ROGUE = WowClass(
     4,
-    _all_non_dracthyr_races,
+    _all_non_dracthyr_and_earthen_races + _earthen,
     Language.EmptyTranslation(),
     "Rogue",
     "rogue",
@@ -168,20 +185,25 @@ def __init__(
         Race.MAGHARORC,
         Race.ZANDALARITROLL,
         Race.VULPERA,
-    ),
+    )
+    + _earthen,
     Language.EmptyTranslation(),
     "Shaman",
     "shaman",
 )
 WARLOCK = WowClass(
     9,
-    _all_non_dracthyr_races,
+    _all_non_dracthyr_and_earthen_races + _earthen,
     Language.EmptyTranslation(),
     "Warlock",
     "warlock",
 )
 WARRIOR = WowClass(
-    1, _all_non_dracthyr_races, Language.EmptyTranslation(), "Warrior", "warrior"
+    1,
+    _all_non_dracthyr_and_earthen_races + _earthen,
+    Language.EmptyTranslation(),
+    "Warrior",
+    "warrior",
 )
 
 WOWCLASSES = (