fix edge case for color name

EboMike · Sep 16, 2024 · eff7be4 · eff7be4
1 parent 6394972
commit eff7be4
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/helpers/tdfhelper.py b/helpers/tdfhelper.py
@@ -31,7 +31,15 @@ def element_to_color(element: str) -> str:
         "Fire": "Red",
         "Ice": "Blue",
         "Earth": "Green",
-        "None": "None"
+        "None": "None",
+
+        # extras for edge cases
+        "Green": "Green",
+        "Red": "Red",
+        "Blue": "Blue",
+        "Solid Red": "Red",
+        "Solid Blue": "Blue",
+        "Solid Green": "Green",
     }
 
     return conversion[element]