mirror of
https://github.com/mister-monster/YouTube2PeerTube.git
synced 2024-11-21 19:44:17 -05:00
95 lines
2.5 KiB
Python
95 lines
2.5 KiB
Python
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
|