def subtitleMissedPP(self):
t = PageTemplate(rh=self, filename='manage_subtitleMissedPP.mako')
app.RELEASES_IN_PP = []
for root, _, files in os.walk(app.TV_DOWNLOAD_DIR, topdown=False):
# Skip folders that are being used for unpacking
if u'_UNPACK' in root.upper():
continue
for filename in sorted(files):
if not is_media_file(filename):
continue
video_path = os.path.join(root, filename)
video_date = datetime.datetime.fromtimestamp(os.stat(video_path).st_ctime)
video_age = datetime.datetime.today() - video_date
tv_episode = Episode.from_filepath(video_path)
if not tv_episode:
logger.log(u"Filename '{0}' cannot be parsed to an episode".format(filename), logger.DEBUG)
continue
ep_status = tv_episode.status
if ep_status in (SNATCHED, SNATCHED_PROPER, SNATCHED_BEST):
status = 'snatched'
elif ep_status == DOWNLOADED:
status = 'downloaded'
else:
continue
if not tv_episode.series.subtitles:
continue
related_files = PostProcessor(video_path).list_associated_files(video_path, subtitles_only=True)
if related_files:
continue
age_hours = divmod(video_age.seconds, 3600)[0]
age_minutes = divmod(video_age.seconds, 60)[0]
if video_age.days > 0:
age_unit = 'd'
age_value = video_age.days
elif age_hours > 0:
age_unit = 'h'
age_value = age_hours
else:
age_unit = 'm'
age_value = age_minutes
app.RELEASES_IN_PP.append({'release': video_path, 'seriesid': tv_episode.series.indexerid,
'show_name': tv_episode.series.name, 'season': tv_episode.season,
'episode': tv_episode.episode, 'status': status, 'age': age_value,
'age_unit': age_unit, 'date': video_date,
'indexername': tv_episode.series.indexer_name})
return t.render(releases_in_pp=app.RELEASES_IN_PP,
controller='manage', action='subtitleMissedPP')
def subtitles_download_in_pp(): # pylint: disable=too-many-locals, too-many-branches, too-many-statements
"""Check for needed subtitles in the post process folder."""
from medusa import process_tv
from medusa.tv import Episode
logger.info(u'Checking for needed subtitles in Post-Process folder')
# Check if PP folder is set
if not app.TV_DOWNLOAD_DIR or not os.path.isdir(app.TV_DOWNLOAD_DIR):
logger.warning(
u'You must set a valid post-process folder in "Post Processing" settings'
)
return
# Search for all wanted languages
if not wanted_languages():
return
SubtitlesFinder.unpack_rar_files(app.TV_DOWNLOAD_DIR)
run_post_process = False
for root, _, files in os.walk(app.TV_DOWNLOAD_DIR, topdown=False):
# Skip folders that are being used for unpacking
if u'_UNPACK' in root.upper():
continue
for filename in sorted(files):
# Delete unwanted subtitles before downloading new ones
delete_unwanted_subtitles(root, filename)
if not is_media_file(filename):
continue
video_path = os.path.join(root, filename)
tv_episode = Episode.from_filepath(video_path)
if not tv_episode:
logger.debug(u'%s cannot be parsed to an episode',
filename)
continue
if tv_episode.status not in (SNATCHED, SNATCHED_PROPER,
SNATCHED_BEST):
continue
if not tv_episode.series.subtitles:
logger.debug(
u'Subtitle disabled for show: %s. Running post-process to PP it',
filename)
run_post_process = True
continue
# Should not consider existing subtitles from db if it's a replacement
new_release_name = remove_extension(filename)
if tv_episode.release_name and new_release_name != tv_episode.release_name:
logger.debug(
u"As this is a release replacement I'm not going to consider existing "
u'subtitles or release name from database to refine the new release'
)
logger.debug(
u"Replacing old release name '%s' with new release name '%s'",
tv_episode.release_name, new_release_name)
tv_episode.subtitles = []
tv_episode.release_name = new_release_name
embedded_subtitles = bool(not app.IGNORE_EMBEDDED_SUBS
and video_path.endswith('.mkv'))
downloaded_languages = download_subtitles(
tv_episode,
video_path=video_path,
subtitles=False,
embedded_subtitles=embedded_subtitles)
# Don't run post processor unless at least one file has all of the needed subtitles OR
# if user don't want to ignore embedded subtitles and wants to consider 'unknown' as wanted sub,
# and .mkv has one.
if not app.PROCESS_AUTOMATICALLY and not run_post_process:
if not needs_subtitles(downloaded_languages):
run_post_process = True
elif not app.IGNORE_EMBEDDED_SUBS:
embedded_subs = get_embedded_subtitles(video_path)
run_post_process = accept_unknown(
embedded_subs) or accept_any(embedded_subs)
if run_post_process:
logger.info(
u'Starting post-process with default settings now that we found subtitles'
)
process_tv.ProcessResult(app.TV_DOWNLOAD_DIR,
app.PROCESS_METHOD).process()
def subtitles_download_in_pp(): # pylint: disable=too-many-locals, too-many-branches, too-many-statements
"""Check for needed subtitles in the post process folder."""
from medusa import process_tv
from medusa.tv import Episode
logger.info(u'Checking for needed subtitles in Post-Process folder')
# Check if PP folder is set
if not app.TV_DOWNLOAD_DIR or not os.path.isdir(app.TV_DOWNLOAD_DIR):
logger.warning(u'You must set a valid post-process folder in "Post Processing" settings')
return
# Search for all wanted languages
if not wanted_languages():
return
SubtitlesFinder.unpack_rar_files(app.TV_DOWNLOAD_DIR)
run_post_process = False
for root, _, files in os.walk(app.TV_DOWNLOAD_DIR, topdown=False):
# Skip folders that are being used for unpacking
if u'_UNPACK' in root.upper():
continue
for filename in sorted(files):
# Delete unwanted subtitles before downloading new ones
delete_unwanted_subtitles(root, filename)
if not is_media_file(filename):
continue
video_path = os.path.join(root, filename)
tv_episode = Episode.from_filepath(video_path)
if not tv_episode:
logger.debug(u'%s cannot be parsed to an episode', filename)
continue
if tv_episode.status not in (SNATCHED, SNATCHED_PROPER, SNATCHED_BEST):
continue
if not tv_episode.series.subtitles:
logger.debug(u'Subtitle disabled for show: %s. Running post-process to PP it', filename)
run_post_process = True
continue
# Should not consider existing subtitles from db if it's a replacement
new_release_name = remove_extension(filename)
if tv_episode.release_name and new_release_name != tv_episode.release_name:
logger.debug(u"As this is a release replacement I'm not going to consider existing "
u'subtitles or release name from database to refine the new release')
logger.debug(u"Replacing old release name '%s' with new release name '%s'",
tv_episode.release_name, new_release_name)
tv_episode.subtitles = []
tv_episode.release_name = new_release_name
embedded_subtitles = bool(not app.IGNORE_EMBEDDED_SUBS and video_path.endswith('.mkv'))
downloaded_languages = download_subtitles(tv_episode, video_path=video_path,
subtitles=False, embedded_subtitles=embedded_subtitles)
# Don't run post processor unless at least one file has all of the needed subtitles OR
# if user don't want to ignore embedded subtitles and wants to consider 'unknown' as wanted sub,
# and .mkv has one.
if not app.PROCESS_AUTOMATICALLY and not run_post_process:
if not needs_subtitles(downloaded_languages):
run_post_process = True
elif not app.IGNORE_EMBEDDED_SUBS:
embedded_subs = get_embedded_subtitles(video_path)
run_post_process = accept_unknown(embedded_subs) or accept_any(embedded_subs)
if run_post_process:
logger.info(u'Starting post-process with default settings now that we found subtitles')
process_tv.ProcessResult(app.TV_DOWNLOAD_DIR, app.PROCESS_METHOD).process()