fix: Handle None element

seriaati · Jun 5, 2024 · ef54ff4 · ef54ff4
1 parent 930b45d
commit ef54ff4
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/hoyo_buddy/bot/translator.py b/hoyo_buddy/bot/translator.py
@@ -265,11 +265,15 @@ def get_traveler_name(
         *,
         gender_symbol: bool = True,
     ) -> str:
-        element_str = self.translate(
-            LocaleStr(
-                GenshinElement(character.element.name.lower()).value.title(), warn_no_key=False
-            ),
-            locale,
+        element_str = (
+            self.translate(
+                LocaleStr(
+                    GenshinElement(character.element.name.lower()).value.title(), warn_no_key=False
+                ),
+                locale,
+            )
+            if character.element is not None
+            else ""
         )
         gender_str = ("♂" if "5" in character.id else "♀") if gender_symbol else ""
         return (