From da4d1d7cc2a4fc590c6acb0a7c0f43be9fc54568 Mon Sep 17 00:00:00 2001
From: Goldy <153996346+g0ldyy@users.noreply.github.com>
Date: Sat, 31 Aug 2024 21:32:53 +0200
Subject: [PATCH] feat: add multi to languages

---
 comet/api/core.py          | 6 +++---
 comet/templates/index.html | 1 +
 comet/utils/general.py     | 8 ++++----
 3 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/comet/api/core.py b/comet/api/core.py
index 0738071..1c9f7c8 100644
--- a/comet/api/core.py
+++ b/comet/api/core.py
@@ -23,11 +23,11 @@ async def health():
 
 
 indexers = settings.INDEXER_MANAGER_INDEXERS
+languages = [language for language in PTT.parse.LANGUAGES_TRANSLATION_TABLE.values()]
+languages.insert(0, "Multi")
 web_config = {
     "indexers": [indexer.replace(" ", "_").lower() for indexer in indexers],
-    "languages": [
-        language for language in PTT.parse.LANGUAGES_TRANSLATION_TABLE.values()
-    ],
+    "languages": languages,
     "resolutions": [resolution.value for resolution in RTN.models.Resolution],
     "resultFormat": ["Title", "Metadata", "Size", "Tracker", "Languages"],
 }
diff --git a/comet/templates/index.html b/comet/templates/index.html
index 7fc7779..792fb95 100644
--- a/comet/templates/index.html
+++ b/comet/templates/index.html
@@ -594,6 +594,7 @@
 
                 <script type="module">
                     const languagesEmojis = {
+                        "Multi": "🌎",
                         "English": "🇬🇧",
                         "Japanese": "🇯🇵",
                         "Chinese": "🇨🇳",
diff --git a/comet/utils/general.py b/comet/utils/general.py
index 38bf85f..7bea0ec 100644
--- a/comet/utils/general.py
+++ b/comet/utils/general.py
@@ -14,7 +14,7 @@ from comet.utils.logger import logger
 from comet.utils.models import settings, ConfigModel
 
 languages_emojis = {
-    "dubbed": "🌎",
+    "multi": "🌎", # Dubbed
     "en": "🇬🇧",  # English
     "ja": "🇯🇵",  # Japanese
     "zh": "🇨🇳",  # Chinese
@@ -523,7 +523,7 @@ def get_balanced_hashes(hashes: dict, config: dict):
 
         if not include_all_languages and not any(
             lang in hash_info["languages"] for lang in config_languages
-        ):
+        ) and (not "multi" in languages if hash_info["dubbed"] else True):
             continue
 
         resolution = hash_info["resolution"]
@@ -602,11 +602,11 @@ def format_title(data: dict, config: dict):
 
     if "All" in config["resultFormat"] or "Languages" in config["resultFormat"]:
         languages = data["languages"]
+        if data["dubbed"]:
+            languages.insert(0, "multi")
         formatted_languages = (
             "/".join(get_language_emoji(language) for language in languages)
             if languages
-            else get_language_emoji("dubbed")
-            if data["dubbed"]
             else None
         )
         languages_str = "\n" + formatted_languages if formatted_languages else ""