YouTube2PeerTube/utils.py

import toml

def read_conf(conf_file):
    conf_file = open(conf_file)
    conf = conf_file.read()
    conf = toml.loads(conf)
    conf_file.close()
    return conf

def convert_timestamp(timestamp):
    timestamp = timestamp.split('T')
    date = timestamp[0].split('-')
    time = timestamp[1].split('+')
    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