Update utils.py

utils.py

@@ -14,3 +14,81 @@ def convert_timestamp(timestamp):
     time = time[0].split(':')
     timestamp = int(date[0] + date[1] + date[2] + time[0] + time[1] + time[2])
     return timestamp
+
+def set_pt_lang(yt_lang, conf_lang):
+    YOUTUBE_LANGUAGE = {
+        "arabic": 'ar',
+        "english": 'en',
+        "french": 'fr',
+        "german": 'de',
+        "hindi": 'hi',
+        "italian": 'it',
+        "japanese": 'ja',
+        "korean": 'ko',
+        "mandarin": 'zh-CN',
+        "portuguese": 'pt-PT',
+        "punjabi": 'pa',
+        "russian": 'ru',
+        "spanish": 'es'
+    }
+    PEERTUBE_LANGUAGE = {
+        "arabic": "ar",
+        "english": "en",
+        "french": "fr",
+        "german": "de",
+        "hindi": "hi",
+        "italian": "it",
+        "japanese": "ja",
+        "korean": "ko",
+        "mandarin": "zh",
+        "portuguese": "pt",
+        "punjabi": "pa",
+        "russian": "ru",
+        "spanish": "es"
+    }
+    # if youtube provides a language value
+    if yt_lang != None:
+        # if the language value is a value and not a key
+        if len((yt_lang).split("-")[0]) < 3:
+            key_list = list(YOUTUBE_LANGUAGE.keys())
+            val_list =list(YOUTUBE_LANGUAGE.values())
+            yt_lang = key_list[val_list.index(yt_lang)]
+        else:
+            pass
+        # now set the language to the peertube value using the key
+        try:
+            lang = PEERTUBE_LANGUAGE[yt_lang]
+        except:
+            # in the event that no key exists for the youtube language, use the conf value
+            if len(conf_lang) > 2:
+                conf_lang = PEERTUBE_LANGUAGE[conf_lang]
+            lang = conf_lang
+    else:
+        if len(conf_lang) > 2:
+            conf_lang = PEERTUBE_LANGUAGE[conf_lang]
+        lang = conf_lang
+    return lang
+
+def set_pt_category(category_str):
+    print(category_str)
+    PEERTUBE_CATEGORY = {
+        "music": 1,
+        "films": 2,
+        "vehicles": 3,
+        "sport": 5,
+        "travels": 6,
+        "gaming": 7,
+        "people": 8,
+        "comedy": 9,
+        "entertainment": 10,
+        "news": 11,
+        "how to": 12,
+        "education": 13,
+        "activism": 14,
+        "science & technology": 15,
+        "science": 15,
+        "technology": 15,
+        "animals": 16
+    }
+    category = str(PEERTUBE_CATEGORY[category_str])
+    return category