From 26f6e4b1e0833a4e1e82b54df0c10c4f7a40158b Mon Sep 17 00:00:00 2001
From: Peter Kruse <kruse.peter.1990@gmail.com>
Date: Mon, 26 Aug 2024 07:18:03 +0200
Subject: [PATCH] add earthen classes

---
 simc_support/game_data/WowClass.py | 46 ++++++++++++++++++++++--------
 1 file changed, 34 insertions(+), 12 deletions(-)

diff --git a/simc_support/game_data/WowClass.py b/simc_support/game_data/WowClass.py
index 46dec32..3b6b000 100644
--- 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 = (