mirror of https://github.com/morpheus65535/bazarr
280 lines
14 KiB
Python
280 lines
14 KiB
Python
# coding=utf-8
|
|
|
|
import gc
|
|
import os
|
|
import logging
|
|
import ast
|
|
|
|
from subliminal_patch import core, search_external_subtitles
|
|
|
|
from languages.custom_lang import CustomLanguage
|
|
from app.database import get_profiles_list, get_profile_cutoff, TableMovies
|
|
from languages.get_languages import alpha2_from_alpha3, language_from_alpha2, get_language_set
|
|
from app.config import settings
|
|
from utilities.helper import get_subtitle_destination_folder
|
|
from utilities.path_mappings import path_mappings
|
|
from subtitles.tools.embedded_subs_reader import embedded_subs_reader
|
|
from app.event_handler import event_stream, show_progress, hide_progress
|
|
from subtitles.indexer.utils import guess_external_subtitles, get_external_subtitles_path
|
|
|
|
gc.enable()
|
|
|
|
|
|
def store_subtitles_movie(original_path, reversed_path, use_cache=True):
|
|
logging.debug('BAZARR started subtitles indexing for this file: ' + reversed_path)
|
|
actual_subtitles = []
|
|
if os.path.exists(reversed_path):
|
|
if settings.general.getboolean('use_embedded_subs'):
|
|
logging.debug("BAZARR is trying to index embedded subtitles.")
|
|
item = TableMovies.select(TableMovies.movie_file_id, TableMovies.file_size)\
|
|
.where(TableMovies.path == original_path)\
|
|
.dicts()\
|
|
.get_or_none()
|
|
if not item:
|
|
logging.exception(f"BAZARR error when trying to select this movie from database: {reversed_path}")
|
|
else:
|
|
try:
|
|
subtitle_languages = embedded_subs_reader(reversed_path,
|
|
file_size=item['file_size'],
|
|
movie_file_id=item['movie_file_id'],
|
|
use_cache=use_cache)
|
|
for subtitle_language, subtitle_forced, subtitle_hi, subtitle_codec in subtitle_languages:
|
|
try:
|
|
if (settings.general.getboolean("ignore_pgs_subs") and subtitle_codec.lower() == "pgs") or \
|
|
(settings.general.getboolean("ignore_vobsub_subs") and subtitle_codec.lower() ==
|
|
"vobsub") or \
|
|
(settings.general.getboolean("ignore_ass_subs") and subtitle_codec.lower() ==
|
|
"ass"):
|
|
logging.debug("BAZARR skipping %s sub for language: %s" % (subtitle_codec, alpha2_from_alpha3(subtitle_language)))
|
|
continue
|
|
|
|
if alpha2_from_alpha3(subtitle_language) is not None:
|
|
lang = str(alpha2_from_alpha3(subtitle_language))
|
|
if subtitle_forced:
|
|
lang = lang + ':forced'
|
|
if subtitle_hi:
|
|
lang = lang + ':hi'
|
|
logging.debug("BAZARR embedded subtitles detected: " + lang)
|
|
actual_subtitles.append([lang, None, None])
|
|
except Exception:
|
|
logging.debug("BAZARR unable to index this unrecognized language: " + subtitle_language)
|
|
pass
|
|
except Exception:
|
|
logging.exception(
|
|
"BAZARR error when trying to analyze this %s file: %s" % (os.path.splitext(reversed_path)[1],
|
|
reversed_path))
|
|
pass
|
|
|
|
try:
|
|
dest_folder = get_subtitle_destination_folder() or ''
|
|
core.CUSTOM_PATHS = [dest_folder] if dest_folder else []
|
|
|
|
# get previously indexed subtitles that haven't changed:
|
|
item = TableMovies.select(TableMovies.subtitles) \
|
|
.where(TableMovies.path == original_path) \
|
|
.dicts() \
|
|
.get_or_none()
|
|
if not item:
|
|
previously_indexed_subtitles_to_exclude = []
|
|
else:
|
|
previously_indexed_subtitles = ast.literal_eval(item['subtitles']) if item['subtitles'] else []
|
|
previously_indexed_subtitles_to_exclude = [x for x in previously_indexed_subtitles
|
|
if len(x) == 3 and
|
|
x[1] and
|
|
os.path.isfile(path_mappings.path_replace(x[1])) and
|
|
os.stat(path_mappings.path_replace(x[1])).st_size == x[2]]
|
|
|
|
subtitles = search_external_subtitles(reversed_path, languages=get_language_set())
|
|
full_dest_folder_path = os.path.dirname(reversed_path)
|
|
if dest_folder:
|
|
if settings.general.subfolder == "absolute":
|
|
full_dest_folder_path = dest_folder
|
|
elif settings.general.subfolder == "relative":
|
|
full_dest_folder_path = os.path.join(os.path.dirname(reversed_path), dest_folder)
|
|
subtitles = guess_external_subtitles(full_dest_folder_path, subtitles, "movie",
|
|
previously_indexed_subtitles_to_exclude)
|
|
except Exception:
|
|
logging.exception("BAZARR unable to index external subtitles.")
|
|
pass
|
|
else:
|
|
for subtitle, language in subtitles.items():
|
|
valid_language = False
|
|
if language:
|
|
if hasattr(language, 'alpha3'):
|
|
valid_language = alpha2_from_alpha3(language.alpha3)
|
|
else:
|
|
logging.debug(f"Skipping subtitles because we are unable to define language: {subtitle}")
|
|
continue
|
|
|
|
if not valid_language:
|
|
logging.debug(f'{language.alpha3} is an unsupported language code.')
|
|
continue
|
|
|
|
subtitle_path = get_external_subtitles_path(reversed_path, subtitle)
|
|
custom = CustomLanguage.found_external(subtitle, subtitle_path)
|
|
|
|
if custom is not None:
|
|
actual_subtitles.append([custom, path_mappings.path_replace_reverse_movie(subtitle_path)])
|
|
|
|
elif str(language.basename) != 'und':
|
|
if language.forced:
|
|
language_str = str(language)
|
|
elif language.hi:
|
|
language_str = str(language) + ':hi'
|
|
else:
|
|
language_str = str(language)
|
|
logging.debug("BAZARR external subtitles detected: " + language_str)
|
|
actual_subtitles.append([language_str, path_mappings.path_replace_reverse_movie(subtitle_path),
|
|
os.stat(subtitle_path).st_size])
|
|
|
|
TableMovies.update({TableMovies.subtitles: str(actual_subtitles)})\
|
|
.where(TableMovies.path == original_path)\
|
|
.execute()
|
|
matching_movies = TableMovies.select(TableMovies.radarrId).where(TableMovies.path == original_path).dicts()
|
|
|
|
for movie in matching_movies:
|
|
if movie:
|
|
logging.debug("BAZARR storing those languages to DB: " + str(actual_subtitles))
|
|
list_missing_subtitles_movies(no=movie['radarrId'])
|
|
else:
|
|
logging.debug("BAZARR haven't been able to update existing subtitles to DB : " + str(actual_subtitles))
|
|
else:
|
|
logging.debug("BAZARR this file doesn't seems to exist or isn't accessible.")
|
|
|
|
logging.debug('BAZARR ended subtitles indexing for this file: ' + reversed_path)
|
|
|
|
return actual_subtitles
|
|
|
|
|
|
def list_missing_subtitles_movies(no=None, send_event=True):
|
|
movies_subtitles = TableMovies.select(TableMovies.radarrId,
|
|
TableMovies.subtitles,
|
|
TableMovies.profileId,
|
|
TableMovies.audio_language)\
|
|
.where((TableMovies.radarrId == no) if no else None)\
|
|
.dicts()
|
|
if isinstance(movies_subtitles, str):
|
|
logging.error("BAZARR list missing subtitles query to DB returned this instead of rows: " + movies_subtitles)
|
|
return
|
|
|
|
use_embedded_subs = settings.general.getboolean('use_embedded_subs')
|
|
|
|
for movie_subtitles in movies_subtitles:
|
|
missing_subtitles_text = '[]'
|
|
if movie_subtitles['profileId']:
|
|
# get desired subtitles
|
|
desired_subtitles_temp = get_profiles_list(profile_id=movie_subtitles['profileId'])
|
|
desired_subtitles_list = []
|
|
if desired_subtitles_temp:
|
|
for language in desired_subtitles_temp['items']:
|
|
if language['audio_exclude'] == "True":
|
|
if language_from_alpha2(language['language']) in ast.literal_eval(
|
|
movie_subtitles['audio_language']):
|
|
continue
|
|
desired_subtitles_list.append([language['language'], language['forced'], language['hi']])
|
|
|
|
# get existing subtitles
|
|
actual_subtitles_list = []
|
|
if movie_subtitles['subtitles'] is not None:
|
|
if use_embedded_subs:
|
|
actual_subtitles_temp = ast.literal_eval(movie_subtitles['subtitles'])
|
|
else:
|
|
actual_subtitles_temp = [x for x in ast.literal_eval(movie_subtitles['subtitles']) if x[1]]
|
|
|
|
for subtitles in actual_subtitles_temp:
|
|
subtitles = subtitles[0].split(':')
|
|
lang = subtitles[0]
|
|
forced = False
|
|
hi = False
|
|
if len(subtitles) > 1:
|
|
if subtitles[1] == 'forced':
|
|
forced = True
|
|
hi = False
|
|
elif subtitles[1] == 'hi':
|
|
forced = False
|
|
hi = True
|
|
actual_subtitles_list.append([lang, str(forced), str(hi)])
|
|
|
|
# check if cutoff is reached and skip any further check
|
|
cutoff_met = False
|
|
cutoff_temp_list = get_profile_cutoff(profile_id=movie_subtitles['profileId'])
|
|
|
|
if cutoff_temp_list:
|
|
for cutoff_temp in cutoff_temp_list:
|
|
cutoff_language = [cutoff_temp['language'], cutoff_temp['forced'], cutoff_temp['hi']]
|
|
if cutoff_temp['audio_exclude'] == 'True' and language_from_alpha2(cutoff_temp['language']) in \
|
|
ast.literal_eval(movie_subtitles['audio_language']):
|
|
cutoff_met = True
|
|
elif cutoff_language in actual_subtitles_list:
|
|
cutoff_met = True
|
|
# HI is considered as good as normal
|
|
elif cutoff_language and [cutoff_language[0], 'False', 'True'] in actual_subtitles_list:
|
|
cutoff_met = True
|
|
|
|
if cutoff_met:
|
|
missing_subtitles_text = str([])
|
|
else:
|
|
# get difference between desired and existing subtitles
|
|
missing_subtitles_list = []
|
|
for item in desired_subtitles_list:
|
|
if item not in actual_subtitles_list:
|
|
missing_subtitles_list.append(item)
|
|
|
|
# remove missing that have forced or hi subtitles for this language in existing
|
|
for item in actual_subtitles_list:
|
|
if item[2] == 'True':
|
|
try:
|
|
missing_subtitles_list.remove([item[0], 'False', 'False'])
|
|
except ValueError:
|
|
pass
|
|
|
|
# make the missing languages list looks like expected
|
|
missing_subtitles_output_list = []
|
|
for item in missing_subtitles_list:
|
|
lang = item[0]
|
|
if item[1] == 'True':
|
|
lang += ':forced'
|
|
elif item[2] == 'True':
|
|
lang += ':hi'
|
|
missing_subtitles_output_list.append(lang)
|
|
|
|
missing_subtitles_text = str(missing_subtitles_output_list)
|
|
|
|
TableMovies.update({TableMovies.missing_subtitles: missing_subtitles_text})\
|
|
.where(TableMovies.radarrId == movie_subtitles['radarrId'])\
|
|
.execute()
|
|
|
|
if send_event:
|
|
event_stream(type='movie', payload=movie_subtitles['radarrId'])
|
|
event_stream(type='badges')
|
|
|
|
|
|
def movies_full_scan_subtitles():
|
|
use_ffprobe_cache = settings.radarr.getboolean('use_ffprobe_cache')
|
|
|
|
movies = TableMovies.select(TableMovies.path).dicts()
|
|
|
|
count_movies = len(movies)
|
|
for i, movie in enumerate(movies):
|
|
show_progress(id='movies_disk_scan',
|
|
header='Full disk scan...',
|
|
name='Movies subtitles',
|
|
value=i,
|
|
count=count_movies)
|
|
store_subtitles_movie(movie['path'], path_mappings.path_replace_movie(movie['path']),
|
|
use_cache=use_ffprobe_cache)
|
|
|
|
hide_progress(id='movies_disk_scan')
|
|
|
|
gc.collect()
|
|
|
|
|
|
def movies_scan_subtitles(no):
|
|
movies = TableMovies.select(TableMovies.path)\
|
|
.where(TableMovies.radarrId == no)\
|
|
.order_by(TableMovies.radarrId)\
|
|
.dicts()
|
|
|
|
for movie in movies:
|
|
store_subtitles_movie(movie['path'], path_mappings.path_replace_movie(movie['path']), use_cache=False)
|