Add utilities to manage thumbnail

This commit is contained in:
LecygneNoir 2018-07-28 12:50:53 +02:00
parent 7dd84e1f8d
commit c822a10d0e

View file

@ -98,6 +98,32 @@ def getLanguage(language, platform):
return PEERTUBE_LANGUAGE[language.lower()] return PEERTUBE_LANGUAGE[language.lower()]
def remove_empty_kwargs(**kwargs):
good_kwargs = {}
if kwargs is not None:
for key, value in kwargs.iteritems():
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"
# 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"
return options
# return the nfo as a RawConfigParser object # return the nfo as a RawConfigParser object
def loadNFO(options): def loadNFO(options):
video_directory = dirname(options.get('--file')) + "/" video_directory = dirname(options.get('--file')) + "/"
@ -117,7 +143,6 @@ def loadNFO(options):
else: else:
if options.get('--name'): if options.get('--name'):
nfo_file = video_directory + options.get('--name') + ".txt" nfo_file = video_directory + options.get('--name') + ".txt"
print nfo_file
if isfile(nfo_file): if isfile(nfo_file):
try: try:
logging.info("Using " + nfo_file + " as NFO, loading...") logging.info("Using " + nfo_file + " as NFO, loading...")