mirror of https://github.com/morpheus65535/bazarr
289 lines
14 KiB
Python
289 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, TableEpisodes, TableShows
|
|
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(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 = TableEpisodes.select(TableEpisodes.episode_file_id, TableEpisodes.file_size)\
|
|
.where(TableEpisodes.path == original_path)\
|
|
.dicts()\
|
|
.get_or_none()
|
|
if not item:
|
|
logging.exception(f"BAZARR error when trying to select this episode from database: {reversed_path}")
|
|
else:
|
|
try:
|
|
subtitle_languages = embedded_subs_reader(reversed_path,
|
|
file_size=item['file_size'],
|
|
episode_file_id=item['episode_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 as error:
|
|
logging.debug("BAZARR unable to index this unrecognized language: %s (%s)", subtitle_language, error)
|
|
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()
|
|
core.CUSTOM_PATHS = [dest_folder] if dest_folder else []
|
|
|
|
# get previously indexed subtitles that haven't changed:
|
|
item = TableEpisodes.select(TableEpisodes.subtitles) \
|
|
.where(TableEpisodes.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(),
|
|
only_one=settings.general.getboolean('single_language'))
|
|
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, "series",
|
|
previously_indexed_subtitles_to_exclude)
|
|
except Exception:
|
|
logging.exception("BAZARR unable to index external subtitles.")
|
|
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(subtitle_path)])
|
|
|
|
elif str(language) != '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(subtitle_path),
|
|
os.stat(subtitle_path).st_size])
|
|
|
|
TableEpisodes.update({TableEpisodes.subtitles: str(actual_subtitles)})\
|
|
.where(TableEpisodes.path == original_path)\
|
|
.execute()
|
|
matching_episodes = TableEpisodes.select(TableEpisodes.sonarrEpisodeId, TableEpisodes.sonarrSeriesId)\
|
|
.where(TableEpisodes.path == original_path)\
|
|
.dicts()
|
|
|
|
for episode in matching_episodes:
|
|
if episode:
|
|
logging.debug("BAZARR storing those languages to DB: " + str(actual_subtitles))
|
|
list_missing_subtitles(epno=episode['sonarrEpisodeId'])
|
|
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(no=None, epno=None, send_event=True):
|
|
if epno is not None:
|
|
episodes_subtitles_clause = (TableEpisodes.sonarrEpisodeId == epno)
|
|
elif no is not None:
|
|
episodes_subtitles_clause = (TableEpisodes.sonarrSeriesId == no)
|
|
else:
|
|
episodes_subtitles_clause = None
|
|
episodes_subtitles = TableEpisodes.select(TableShows.sonarrSeriesId,
|
|
TableEpisodes.sonarrEpisodeId,
|
|
TableEpisodes.subtitles,
|
|
TableShows.profileId,
|
|
TableEpisodes.audio_language)\
|
|
.join(TableShows, on=(TableEpisodes.sonarrSeriesId == TableShows.sonarrSeriesId))\
|
|
.where(episodes_subtitles_clause)\
|
|
.dicts()
|
|
if isinstance(episodes_subtitles, str):
|
|
logging.error("BAZARR list missing subtitles query to DB returned this instead of rows: " + episodes_subtitles)
|
|
return
|
|
|
|
use_embedded_subs = settings.general.getboolean('use_embedded_subs')
|
|
|
|
for episode_subtitles in episodes_subtitles:
|
|
missing_subtitles_text = '[]'
|
|
if episode_subtitles['profileId']:
|
|
# get desired subtitles
|
|
desired_subtitles_temp = get_profiles_list(profile_id=episode_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(
|
|
episode_subtitles['audio_language']):
|
|
continue
|
|
desired_subtitles_list.append([language['language'], language['forced'], language['hi']])
|
|
|
|
# get existing subtitles
|
|
actual_subtitles_list = []
|
|
if episode_subtitles['subtitles'] is not None:
|
|
if use_embedded_subs:
|
|
actual_subtitles_temp = ast.literal_eval(episode_subtitles['subtitles'])
|
|
else:
|
|
actual_subtitles_temp = [x for x in ast.literal_eval(episode_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=episode_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(episode_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[0], 'False', 'True'] in actual_subtitles_list:
|
|
cutoff_met = True
|
|
|
|
if cutoff_met:
|
|
missing_subtitles_text = str([])
|
|
else:
|
|
# if cutoff isn't met or None, we continue
|
|
|
|
# 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 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)
|
|
|
|
TableEpisodes.update({TableEpisodes.missing_subtitles: missing_subtitles_text})\
|
|
.where(TableEpisodes.sonarrEpisodeId == episode_subtitles['sonarrEpisodeId'])\
|
|
.execute()
|
|
|
|
if send_event:
|
|
event_stream(type='episode', payload=episode_subtitles['sonarrEpisodeId'])
|
|
event_stream(type='badges')
|
|
|
|
|
|
def series_full_scan_subtitles():
|
|
use_ffprobe_cache = settings.sonarr.getboolean('use_ffprobe_cache')
|
|
|
|
episodes = TableEpisodes.select(TableEpisodes.path).dicts()
|
|
|
|
count_episodes = len(episodes)
|
|
for i, episode in enumerate(episodes):
|
|
show_progress(id='episodes_disk_scan',
|
|
header='Full disk scan...',
|
|
name='Episodes subtitles',
|
|
value=i,
|
|
count=count_episodes)
|
|
store_subtitles(episode['path'], path_mappings.path_replace(episode['path']), use_cache=use_ffprobe_cache)
|
|
|
|
hide_progress(id='episodes_disk_scan')
|
|
|
|
gc.collect()
|
|
|
|
|
|
def series_scan_subtitles(no):
|
|
episodes = TableEpisodes.select(TableEpisodes.path)\
|
|
.where(TableEpisodes.sonarrSeriesId == no)\
|
|
.order_by(TableEpisodes.sonarrEpisodeId)\
|
|
.dicts()
|
|
|
|
for episode in episodes:
|
|
store_subtitles(episode['path'], path_mappings.path_replace(episode['path']), use_cache=False)
|