You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
240 lines
7.4 KiB
240 lines
7.4 KiB
#!/usr/bin/python
|
|
# coding: utf-8
|
|
|
|
from configparser import RawConfigParser, NoOptionError, NoSectionError
|
|
from os.path import dirname, splitext, basename, isfile, getmtime
|
|
import re
|
|
import unidecode
|
|
import logging
|
|
import datetime
|
|
|
|
logger = logging.getLogger('Prismedia')
|
|
|
|
### CATEGORIES ###
|
|
YOUTUBE_CATEGORY = {
|
|
"music": 10,
|
|
"films": 1,
|
|
"vehicles": 2,
|
|
"sport": 17,
|
|
"travels": 19,
|
|
"gaming": 20,
|
|
"people": 22,
|
|
"comedy": 23,
|
|
"entertainment": 24,
|
|
"news": 25,
|
|
"how to": 26,
|
|
"education": 27,
|
|
"activism": 29,
|
|
"science & technology": 28,
|
|
"science": 28,
|
|
"technology": 28,
|
|
"animals": 15
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
### LANGUAGES ###
|
|
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"
|
|
}
|
|
######################
|
|
|
|
|
|
def getCategory(category, platform):
|
|
if platform == "youtube":
|
|
return YOUTUBE_CATEGORY[category.lower()]
|
|
else:
|
|
return PEERTUBE_CATEGORY[category.lower()]
|
|
|
|
|
|
def getLanguage(language, platform):
|
|
if platform == "youtube":
|
|
return YOUTUBE_LANGUAGE[language.lower()]
|
|
else:
|
|
return PEERTUBE_LANGUAGE[language.lower()]
|
|
|
|
|
|
def ask_overwrite(question):
|
|
while True:
|
|
reply = str(input(question + ' (Yes/[No]): ') or "No").lower().strip()
|
|
if reply[:1] == 'y':
|
|
return True
|
|
if reply[:1] == 'n':
|
|
return False
|
|
|
|
|
|
def remove_empty_kwargs(**kwargs):
|
|
good_kwargs = {}
|
|
if kwargs is not None:
|
|
for key, value in kwargs.items():
|
|
if value:
|
|
good_kwargs[key] = value
|
|
return good_kwargs
|
|
|
|
|
|
def searchThumbnail(options):
|
|
video_directory = dirname(options.get('--file')) + "/"
|
|
# First, check for thumbnail based on videoname
|
|
if options.get('--name'):
|
|
if isfile(video_directory + options.get('--name') + ".jpg"):
|
|
options['--thumbnail'] = video_directory + options.get('--name') + ".jpg"
|
|
elif isfile(video_directory + options.get('--name') + ".jpeg"):
|
|
options['--thumbnail'] = video_directory + options.get('--name') + ".jpeg"
|
|
elif isfile(video_directory + options.get('--name') + ".png"):
|
|
options['--thumbnail'] = video_directory + options.get('--name') + ".png"
|
|
# Then, if we still not have thumbnail, check for thumbnail based on videofile name
|
|
if not options.get('--thumbnail'):
|
|
video_file = splitext(basename(options.get('--file')))[0]
|
|
if isfile(video_directory + video_file + ".jpg"):
|
|
options['--thumbnail'] = video_directory + video_file + ".jpg"
|
|
elif isfile(video_directory + video_file + ".jpeg"):
|
|
options['--thumbnail'] = video_directory + video_file + ".jpeg"
|
|
elif isfile(video_directory + video_file + ".png"):
|
|
options['--thumbnail'] = video_directory + video_file + ".png"
|
|
|
|
# Display some info after research
|
|
if not options.get('--thumbnail'):
|
|
logger.debug("No thumbnail has been found, continuing")
|
|
else:
|
|
logger.info("Using " + options.get('--thumbnail') + " as thumbnail")
|
|
|
|
return options
|
|
|
|
|
|
def searchOriginalDate(options):
|
|
fileModificationDate = str(getmtime(options.get('--file'))).split('.')
|
|
return datetime.datetime.fromtimestamp(int(fileModificationDate[0])).isoformat()
|
|
|
|
|
|
# return the nfo as a RawConfigParser object
|
|
def loadNFO(filename):
|
|
try:
|
|
logger.info("Loading " + filename + " as NFO")
|
|
nfo = RawConfigParser()
|
|
nfo.read(filename, encoding='utf-8')
|
|
return nfo
|
|
except Exception as e:
|
|
logger.critical("Problem loading NFO file " + filename + ": " + str(e))
|
|
exit(1)
|
|
return False
|
|
|
|
|
|
def parseNFO(options):
|
|
video_directory = dirname(options.get('--file'))
|
|
directory_name = basename(video_directory)
|
|
nfo_txt = False
|
|
nfo_directory = False
|
|
nfo_videoname = False
|
|
nfo_file = False
|
|
nfo_cli = False
|
|
|
|
if isfile(video_directory + "/" + "nfo.txt"):
|
|
nfo_txt = loadNFO(video_directory + "/" + "nfo.txt")
|
|
elif isfile(video_directory + "/" + "NFO.txt"):
|
|
nfo_txt = loadNFO(video_directory + "/" + "NFO.txt")
|
|
|
|
if isfile(video_directory + "/" + directory_name + ".txt"):
|
|
nfo_directory = loadNFO(video_directory + "/" + directory_name + ".txt")
|
|
|
|
if options.get('--name'):
|
|
if isfile(video_directory + "/" + options.get('--name')):
|
|
nfo_videoname = loadNFO(video_directory + "/" + options.get('--name') + ".txt")
|
|
|
|
video_file = splitext(basename(options.get('--file')))[0]
|
|
if isfile(video_directory + "/" + video_file + ".txt"):
|
|
nfo_file = loadNFO(video_directory + "/" + video_file + ".txt")
|
|
|
|
if options.get('--nfo'):
|
|
if isfile(options.get('--nfo')):
|
|
nfo_cli = loadNFO(options.get('--nfo'))
|
|
else:
|
|
logger.critical("Given NFO file does not exist, please check your path.")
|
|
exit(1)
|
|
|
|
# If there is no NFO and strict option is enabled, then stop there
|
|
if options.get('--withNFO'):
|
|
if not isinstance(nfo_cli, RawConfigParser) and \
|
|
not isinstance(nfo_file, RawConfigParser) and \
|
|
not isinstance(nfo_videoname, RawConfigParser) and \
|
|
not isinstance(nfo_directory, RawConfigParser) and \
|
|
not isinstance(nfo_txt, RawConfigParser):
|
|
logger.critical("You have required the strict presence of NFO but none is found, please use a NFO.")
|
|
exit(1)
|
|
|
|
# We need to load NFO in this exact order to keep the priorities
|
|
# options in cli > nfo_cli > nfo_file > nfo_videoname > nfo_directory > nfo_txt
|
|
for nfo in [nfo_cli, nfo_file, nfo_videoname, nfo_directory, nfo_txt]:
|
|
if nfo:
|
|
# We need to check all options and replace it with the nfo value if not defined (None or False)
|
|
for key, value in options.items():
|
|
key = key.replace("--", "")
|
|
try:
|
|
# get string options
|
|
if value is None and nfo.get('video', key):
|
|
options['--' + key] = nfo.get('video', key)
|
|
# get boolean options
|
|
elif value is False and nfo.getboolean('video', key):
|
|
options['--' + key] = nfo.getboolean('video', key)
|
|
except NoOptionError:
|
|
continue
|
|
except NoSectionError:
|
|
logger.critical(nfo + " misses section [video], please check syntax of your NFO.")
|
|
exit(1)
|
|
return options
|
|
|
|
|
|
def upcaseFirstLetter(s):
|
|
return s[0].upper() + s[1:]
|
|
|
|
|
|
def cleanString(toclean):
|
|
toclean = unidecode.unidecode(toclean)
|
|
cleaned = re.sub('[^A-Za-z0-9]+', '', toclean)
|
|
|
|
return cleaned
|