dataclasses = conciseness & clean

Author: FAReTek1

scratchattach/other/other_apis.py

@@ -166,7 +166,6 @@ def translate(language: str | Languages, text: str = "hello"):
     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

@@ -4,32 +4,18 @@
 """
 
 from enum import Enum
+from dataclasses import dataclass
+
 from typing import Callable
 
 
+@dataclass(init=True, repr=True)
 class _Language:
-    def __init__(self, name: str = None, code: str = None, locales: list[str] = None, tts_locale: str = None,
-                 single_gender: bool = None):
-        self.name = name
-        self.code = code
-        self.locales = locales
-        self.tts_locale = tts_locale
-        self.single_gender = single_gender
-
-    def __repr__(self):
-        ret = "Language("
-        for attr in self.__dict__.keys():
-            if not attr.startswith("_"):
-                val = getattr(self, attr)
-                ret += f"{repr(val)}, "
-        if ret.endswith(", "):
-            ret = ret[:-2]
-
-        ret += ')'
-        return ret
-
-    def __str__(self):
-        return f"Language<{self.name} - {self.code}>"
+    name: str = None
+    code: str = None
+    locales: list[str] = None
+    tts_locale: str = None
+    single_gender: bool = None
 
 
 class Languages(Enum):