smarter language evaluation in translate function. some credit to thecommcraft

Author: FAReTek1

scratchattach/other/other_apis.py

@@ -149,14 +149,24 @@ def scratch_team_members() -> dict:
 
 
 def translate(language: str | Languages, text: str = "hello"):
-    if language.lower() not in Languages.all_of("code", str.lower):
-        if language.lower() in Languages.all_of("name", str.lower):
-            language = Languages.find(language.lower(), apply_func=str.lower).code
+    lang = language
+    if isinstance(language, str):
+        if language.lower() in Languages.all_of("code", str.lower):
+            lang = Languages.find(language.lower(), "code", apply_func=str.lower)
+
+        elif language.lower() in Languages.all_of("name", str.lower):
+            lang = Languages.find(language.lower(), apply_func=str.lower)
+
+    elif isinstance(language, Languages):
+        lang = language.value
+    else:
+        # The code will work so long as the language has a 'code' attribute, however, this is bad practice.
+        warnings.warn(f"{language} is not {str} or {Languages}, but {type(language)}.")
 
-    lang = Languages.find(language, "code", str.lower)
     if lang is None:
         raise ValueError(f"{language} is not a supported translate language")
 
+    print(lang.__dict__)
     response_json = requests.get(
         f"https://translate-service.scratch.mit.edu/translate?language={lang.code}&text={text}").json()
 

scratchattach/utils/supportedlangs.py

@@ -116,6 +116,7 @@ def apply_func(x):
 
         for lang_enum in Languages:
             lang = lang_enum.value
+
             try:
                 if apply_func(getattr(lang, by)) == value:
                     return lang