2 files changed, 161 insertions(+), 151 deletions(-)

M simplytranslate_engines/googletranslate.py
M simplytranslate_engines/libretranslate.py

@@ 1,131 1,135 @@
 import lxml.html as lxml
 import requests
 
-supported_languages = {
-    "Afrikaans": "af",
-    "Albanian": "sq",
-    "Amharic": "am",
-    "Arabic": "ar",
-    "Armenian": "hy",
-    "Azerbaijani": "az",
-    "Basque": "eu",
-    "Belarusian": "be",
-    "Bengali": "bn",
-    "Bosnian": "bs",
-    "Bulgarian": "bg",
-    "Catalan": "ca",
-    "Cebuano": "ceb",
-    "Chichewa": "ny",
-    "Chinese": "zh-CN",
-    "Corsican": "co",
-    "Croatian": "hr",
-    "Czech": "cs",
-    "Danish": "da",
-    "Dutch": "nl",
-    "English": "en",
-    "Esperanto": "eo",
-    "Estonian": "et",
-    "Filipino": "tl",
-    "Finnish": "fi",
-    "French": "fr",
-    "Frisian": "fy",
-    "Galician": "gl",
-    "Georgian": "ka",
-    "German": "de",
-    "Greek": "el",
-    "Gujarati": "gu",
-    "Haitian Creole": "ht",
-    "Hausa": "ha",
-    "Hawaiian": "haw",
-    "Hebrew": "iw",
-    "Hindi": "hi",
-    "Hmong": "hmn",
-    "Hungarian": "hu",
-    "Icelandic": "is",
-    "Igbo": "ig",
-    "Indonesian": "id",
-    "Irish": "ga",
-    "Italian": "it",
-    "Japanese": "ja",
-    "Javanese": "jw",
-    "Kannada": "kn",
-    "Kazakh": "kk",
-    "Khmer": "km",
-    "Kinyarwanda": "rw",
-    "Korean": "ko",
-    "Kurdish (Kurmanji)": "ku",
-    "Kyrgyz": "ky",
-    "Lao": "lo",
-    "Latin": "la",
-    "Latvian": "lv",
-    "Lithuanian": "lt",
-    "Luxembourgish": "lb",
-    "Macedonian": "mk",
-    "Malagasy": "mg",
-    "Malay": "ms",
-    "Malayalam": "ml",
-    "Maltese": "mt",
-    "Maori": "mi",
-    "Marathi": "mr",
-    "Mongolian": "mn",
-    "Myanmar (Burmese)": "my",
-    "Nepali": "ne",
-    "Norwegian": "no",
-    "Odia (Oriya)": "or",
-    "Pashto": "ps",
-    "Persian": "fa",
-    "Polish": "pl",
-    "Portuguese": "pt",
-    "Punjabi": "pa",
-    "Romanian": "ro",
-    "Russian": "ru",
-    "Samoan": "sm",
-    "Scots Gaelic": "gd",
-    "Serbian": "sr",
-    "Sesotho": "st",
-    "Shona": "sn",
-    "Sindhi": "sd",
-    "Sinhala": "si",
-    "Slovak": "sk",
-    "Slovenian": "sl",
-    "Somali": "so",
-    "Spanish": "es",
-    "Sundanese": "su",
-    "Swahili": "sw",
-    "Swedish": "sv",
-    "Tajik": "tg",
-    "Tamil": "ta",
-    "Tatar": "tt",
-    "Telugu": "te",
-    "Thai": "th",
-    "Turkish": "tr",
-    "Turkmen": "tk",
-    "Ukrainian": "uk",
-    "Urdu": "ur",
-    "Uyghur": "ug",
-    "Uzbek": "uz",
-    "Vietnamese": "vi",
-    "Welsh": "cy",
-    "Xhosa": "xh",
-    "Yiddish": "yi",
-    "Yoruba": "yo",
-    "Zulu": "zu",
-}
+class GoogleTranslateEngine:
+    name = "google"
 
-def translate(text, to_language, from_language="auto"):
-    r = requests.get(
-        "https://translate.google.com/m",
-        params={
-            "q": text,
-            "sl": from_language,
-            "tl": to_language
-        })
+    def get_supported_languages(self):
+        return {
+            "Afrikaans": "af",
+            "Albanian": "sq",
+            "Amharic": "am",
+            "Arabic": "ar",
+            "Armenian": "hy",
+            "Azerbaijani": "az",
+            "Basque": "eu",
+            "Belarusian": "be",
+            "Bengali": "bn",
+            "Bosnian": "bs",
+            "Bulgarian": "bg",
+            "Catalan": "ca",
+            "Cebuano": "ceb",
+            "Chichewa": "ny",
+            "Chinese": "zh-CN",
+            "Corsican": "co",
+            "Croatian": "hr",
+            "Czech": "cs",
+            "Danish": "da",
+            "Dutch": "nl",
+            "English": "en",
+            "Esperanto": "eo",
+            "Estonian": "et",
+            "Filipino": "tl",
+            "Finnish": "fi",
+            "French": "fr",
+            "Frisian": "fy",
+            "Galician": "gl",
+            "Georgian": "ka",
+            "German": "de",
+            "Greek": "el",
+            "Gujarati": "gu",
+            "Haitian Creole": "ht",
+            "Hausa": "ha",
+            "Hawaiian": "haw",
+            "Hebrew": "iw",
+            "Hindi": "hi",
+            "Hmong": "hmn",
+            "Hungarian": "hu",
+            "Icelandic": "is",
+            "Igbo": "ig",
+            "Indonesian": "id",
+            "Irish": "ga",
+            "Italian": "it",
+            "Japanese": "ja",
+            "Javanese": "jw",
+            "Kannada": "kn",
+            "Kazakh": "kk",
+            "Khmer": "km",
+            "Kinyarwanda": "rw",
+            "Korean": "ko",
+            "Kurdish (Kurmanji)": "ku",
+            "Kyrgyz": "ky",
+            "Lao": "lo",
+            "Latin": "la",
+            "Latvian": "lv",
+            "Lithuanian": "lt",
+            "Luxembourgish": "lb",
+            "Macedonian": "mk",
+            "Malagasy": "mg",
+            "Malay": "ms",
+            "Malayalam": "ml",
+            "Maltese": "mt",
+            "Maori": "mi",
+            "Marathi": "mr",
+            "Mongolian": "mn",
+            "Myanmar (Burmese)": "my",
+            "Nepali": "ne",
+            "Norwegian": "no",
+            "Odia (Oriya)": "or",
+            "Pashto": "ps",
+            "Persian": "fa",
+            "Polish": "pl",
+            "Portuguese": "pt",
+            "Punjabi": "pa",
+            "Romanian": "ro",
+            "Russian": "ru",
+            "Samoan": "sm",
+            "Scots Gaelic": "gd",
+            "Serbian": "sr",
+            "Sesotho": "st",
+            "Shona": "sn",
+            "Sindhi": "sd",
+            "Sinhala": "si",
+            "Slovak": "sk",
+            "Slovenian": "sl",
+            "Somali": "so",
+            "Spanish": "es",
+            "Sundanese": "su",
+            "Swahili": "sw",
+            "Swedish": "sv",
+            "Tajik": "tg",
+            "Tamil": "ta",
+            "Tatar": "tt",
+            "Telugu": "te",
+            "Thai": "th",
+            "Turkish": "tr",
+            "Turkmen": "tk",
+            "Ukrainian": "uk",
+            "Urdu": "ur",
+            "Uyghur": "ug",
+            "Uzbek": "uz",
+            "Vietnamese": "vi",
+            "Welsh": "cy",
+            "Xhosa": "xh",
+            "Yiddish": "yi",
+            "Yoruba": "yo",
+            "Zulu": "zu",
+        }
 
-    doc = lxml.fromstring(r.text)
-    for container in doc.find_class("result-container"):
-        return container.text_content()
+    def translate(self, text, to_language, from_language="auto"):
+        r = requests.get(
+            "https://translate.google.com/m",
+            params={
+                "q": text,
+                "sl": from_language,
+                "tl": to_language
+            })
 
-    return ""
+        doc = lxml.fromstring(r.text)
+        for container in doc.find_class("result-container"):
+            return container.text_content()
+
+        return ""
 
 if __name__ == "__main__":
-    print(translate("hello", "fr", "en"))
+    print(GoogleTranslateEngine().translate("hello", "fr", "en"))

@@ 1,36 1,42 @@
 import json
 import requests
 
-supported_languages = {
-    "English": "en",
-    "Arabic": "ar",
-    "Chinese": "zh",
-    "French": "fr",
-    "German": "de",
-    "Italian": "it",
-    "Portuguese": "pt",
-    "Russian": "ru",
-    "Spanish": "es",
-}
+class LibreTranslateEngine:
+    def __init__(self, url="https://libretranslate.com"):
+        self.url = url
 
-def translate(text, to_language, from_language="auto", libretranslate_url="https://libretranslate.com/translate"):
-    r = requests.post(
-        libretranslate_url,
-        json={
-            "q": text,
-            "source": from_language,
-            "target": to_language
-        })
+    name = "libre"
 
-    #response = dict(r.text)
-    response = json.loads(r.text)
-    if "translatedText" in response:
-        return response["translatedText"]
-    elif "error" in response:
-        return response["error"]
-    else:
-        return "odd, something went wrong"
+    def get_supported_languages(self):
+        return {
+            "English": "en",
+            "Arabic": "ar",
+            "Chinese": "zh",
+            "French": "fr",
+            "German": "de",
+            "Italian": "it",
+            "Portuguese": "pt",
+            "Russian": "ru",
+            "Spanish": "es",
+        }
 
+    def translate(self, text, to_language, from_language="auto"):
+        r = requests.post(
+            f"{self.url}/translate",
+            json={
+                "q": text,
+                "source": from_language,
+                "target": to_language
+            })
+
+        #response = dict(r.text)
+        response = json.loads(r.text)
+        if "translatedText" in response:
+            return response["translatedText"]
+        elif "error" in response:
+            return response["error"]
+        else:
+            return "odd, something went wrong"
 
 if __name__ == "__main__":
-    print(translate("lutsch meinen schwanz", "en"))
+    print(LibreTranslateEngine().translate("lutsch meinen schwanz", "en"))