Make language case-insensitive

Kometa-Team · May 10, 2024 · c65cd6d · c65cd6d
1 parent d92f5bb
commit c65cd6d
Showing 1 changed file with 79 additions and 80 deletions.
diff --git a/modules/anidb.py b/modules/anidb.py
@@ -17,87 +17,86 @@
     "login": f"{base_url}/perl-bin/animedb.pl"
 }
 weights = {"anidb": 1000, "anidb_3_0": 600, "anidb_2_5": 500, "anidb_2_0": 400, "anidb_1_5": 300, "anidb_1_0": 200, "anidb_0_5": 100}
-language_matcher = {
-    "af" : "AFRIKAANS",
-    "sq" : "ALBANIAN",
-    "al" : "ALBANIAN",
-    "ar" : "ARABIC",
-    "bn" : "BENGALI",
-    "bd" : "BENGALI",
-    "bs" : "BOSNIAN",
-    "bg" : "BULGARIAN",
-    "my" : "BURMESE",
-    "zh" : "CHINESE",
-    "zh-nan" : "CHINESE_TAIWAN",
-    "zh-Hant" : "CHINESE_TRADITIONAL",
-    "x-zht" : "CHINESE_TRANSLITERATED",
-    "zh-Hans" : "CHINESE_SIMPLIFIED",
-    "zh-cmn" : "CHINESE_SIMPLIFIED",
-    "hr" : "CROATIAN",
-    "cs" : "CZECH",
-    "da" : "DANISH",
-    "nl" : "DUTCH",
-    "en" : "ENGLISH",
-    "eo" : "ESPERANTO",
-    "et" : "ESTONIAN",
-    "fil" : "FILIPINO",
-    "fi" : "FINNISH",
-    "fr" : "FRENCH",
-    "ka" : "GEORGIAN",
-    "de" : "GERMAN",
-    "el" : "GREEK",
-    "grc" : "GREEK_ANCIENT",
-    "ht" : "HAITIAN",
-    "he" : "HEBREW",
-    "hi" : "HINDI",
-    "hu" : "HUNGARIAN",
-    "is" : "ICELANDIC",
-    "id" : "INDONESIAN",
-    "it" : "ITALIAN",
-    "ja" : "JAPANESE",
-    "x-jat" : "JAPANESE_TRANSLITERATED",
-    "jv" : "JAVANESE",
-    "ko" : "KOREAN",
-    "x-kot" : "KOREAN_TRANSLITERATED",
-    "la" : "LATIN",
-    "lv" : "LATVIAN",
-    "lt" : "LITHUANIAN",
-    "ms" : "MALAY",
-    "mn" : "MONGOLIAN",
-    "ne" : "NEPALI",
-    "no" : "NORWEGIAN",
-    "fa" : "PERSIAN",
-    "pl" : "POLISH",
-    "pt" : "PORTUGUESE",
-    "pt-BR" : "PORTUGUESE_BRAZIL",
-    "ro" : "ROMANIAN",
-    "ru" : "RUSSIAN",
-    "sr" : "SERBIAN",
-    "si" : "SINHALA",
-    "sk" : "SLOVAK",
-    "sl" : "SLOVENIAN",
-    "es" : "SPANISH",
-    "es-PV" : "SPANISH_BASQUE",
-    "es-CT" : "SPANISH_CATALAN",
-    "es-CA" : "SPANISH_CATALAN",
-    "es-GA" : "SPANISH_GALICIA",
-    "es-419" : "SPANISH_LATIN",
-    "sv" : "SWEDISH",
-    "tl" : "TAGALOG",
-    "ta" : "TAMIL",
-    "tt" : "TATAR",
-    "te" : "TELUGU",
-    "th" : "THAI",
-    "x-tht" : "THAI_TRANSLITERATED",
-    "tr" : "TURKISH",
-    "uk" : "UKRAINIAN",
-    "ur" : "URDU",
-    "vi" : "VIETNAMESE",
-    "x-unk" : "UNKNOWN",
-    "x-other" : "OTHER"
+LANGUAGE_MATCHER = {
+    "AF" : "AFRIKAANS",
+    "SQ" : "ALBANIAN",
+    "AL" : "ALBANIAN",
+    "AR" : "ARABIC",
+    "BN" : "BENGALI",
+    "BD" : "BENGALI",
+    "BS" : "BOSNIAN",
+    "BG" : "BULGARIAN",
+    "MY" : "BURMESE",
+    "ZH" : "CHINESE",
+    "ZH-NAN" : "CHINESE_TAIWAN",
+    "ZH-HANT" : "CHINESE_TRADITIONAL",
+    "X-ZHT" : "CHINESE_TRANSLITERATED",
+    "ZH-HANS" : "CHINESE_SIMPLIFIED",
+    "ZH-CMN" : "CHINESE_SIMPLIFIED",
+    "HR" : "CROATIAN",
+    "CS" : "CZECH",
+    "DA" : "DANISH",
+    "NL" : "DUTCH",
+    "EN" : "ENGLISH",
+    "EO" : "ESPERANTO",
+    "ET" : "ESTONIAN",
+    "FIL" : "FILIPINO",
+    "FI" : "FINNISH",
+    "FR" : "FRENCH",
+    "KA" : "GEORGIAN",
+    "DE" : "GERMAN",
+    "EL" : "GREEK",
+    "GRC" : "GREEK_ANCIENT",
+    "HT" : "HAITIAN",
+    "HE" : "HEBREW",
+    "HI" : "HINDI",
+    "HU" : "HUNGARIAN",
+    "IS" : "ICELANDIC",
+    "ID" : "INDONESIAN",
+    "IT" : "ITALIAN",
+    "JA" : "JAPANESE",
+    "X-JAT" : "JAPANESE_TRANSLITERATED",
+    "JV" : "JAVANESE",
+    "KO" : "KOREAN",
+    "X-KOT" : "KOREAN_TRANSLITERATED",
+    "LA" : "LATIN",
+    "LV" : "LATVIAN",
+    "LT" : "LITHUANIAN",
+    "MS" : "MALAY",
+    "MN" : "MONGOLIAN",
+    "NE" : "NEPALI",
+    "NO" : "NORWEGIAN",
+    "FA" : "PERSIAN",
+    "PL" : "POLISH",
+    "PT" : "PORTUGUESE",
+    "PT-BR" : "PORTUGUESE_BRAZIL",
+    "RO" : "ROMANIAN",
+    "RU" : "RUSSIAN",
+    "SR" : "SERBIAN",
+    "SI" : "SINHALA",
+    "SK" : "SLOVAK",
+    "SL" : "SLOVENIAN",
+    "ES" : "SPANISH",
+    "ES-PV" : "SPANISH_BASQUE",
+    "ES-CT" : "SPANISH_CATALAN",
+    "ES-CA" : "SPANISH_CATALAN",
+    "ES-GA" : "SPANISH_GALICIA",
+    "ES-419" : "SPANISH_LATIN",
+    "SV" : "SWEDISH",
+    "TL" : "TAGALOG",
+    "TA" : "TAMIL",
+    "TT" : "TATAR",
+    "TE" : "TELUGU",
+    "TH" : "THAI",
+    "X-THT" : "THAI_TRANSLITERATED",
+    "TR" : "TURKISH",
+    "UK" : "UKRAINIAN",
+    "UR" : "URDU",
+    "VI" : "VIETNAMESE",
+    "X-UNK" : "UNKNOWN",
+    "X-OTHER" : "OTHER"
 }
 
-
 class AniDBObj:
     def __init__(self, anidb, anidb_id, data, from_json=False):
         self._anidb = anidb
@@ -124,7 +123,7 @@ def _parseJSON(self):
                     self.main_title = title["title"]
                 if title["type"] == "OFFICIAL":
                     self.titles.append(title)
-                if title["type"] == "OFFICIAL" and (title["language"] == self._anidb.language.upper() or title["language"] == language_matcher[self._anidb.language]):
+                if title["type"] == "OFFICIAL" and self._anidb.language and (title["language"] == self._anidb.language.upper() or title["language"] == language_matcher[self._anidb.language.upper()]):
                     self.official_title = title["title"]
             if not self.official_title:
                 self.official_title = self.main_title