def test_list_associated_files(p, create_structure, monkeypatch):
"""Run the test."""
# Given
test_path = create_structure(p['path'], structure=p['structure'])
path = os.path.join(test_path, os.path.normcase(p['path']))
file_path = os.path.join(path, p['structure'][0])
expected_associated_files = p['expected_associated_files']
subtitles_only = p.get('subtitles_only', False)
subfolders = p.get('subfolders', False)
refine = p.get('refine', False)
monkeypatch.setattr(app, 'ALLOWED_EXTENSIONS', p['allowed_extensions'])
monkeypatch.setattr(app, 'MOVE_ASSOCIATED_FILES', 1)
processor = PostProcessor(file_path)
processor._rar_basename = _rar_basename
# When
found_associated_files = processor.list_associated_files(
file_path,
subfolders=subfolders,
subtitles_only=subtitles_only,
refine=refine)
associated_files_basenames = [
os.path.basename(i) for i in found_associated_files
]
# Then
assert set(associated_files_basenames) == set(expected_associated_files)
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 test_process(self):
show = TVShow(1, 3)
show.name = test.SHOW_NAME
show.location = test.SHOW_DIR
show.save_to_db()
app.showList = [show]
episode = TVEpisode(show, test.SEASON, test.EPISODE)
episode.name = "some episode name"
episode.save_to_db()
addNameToCache('show name', 3)
app.PROCESS_METHOD = 'move'
post_processor = PostProcessor(test.FILE_PATH)
self.assertTrue(post_processor.process())
def test__process_postponed(monkeypatch, p, create_structure):
"""Run the test."""
# Given
test_path = create_structure(p['path'], structure=p['structure'])
path = os.path.join(test_path, os.path.normcase(p['path']))
video_path = os.path.join(path, p['video'])
processor = PostProcessor(path)
sut = ProcessResult(path)
# Overwrite internal method
sut.subtitles_enabled = lambda path, resource_name: p['subtitles_enabled']
# When
result = sut._process_postponed(processor, video_path, p['video'], p['ignore_subs'])
# Then
assert p['expected'] == result
def test_parse_info(p, monkeypatch, parse_method):
"""Run the test."""
# Given
monkeypatch.setattr(NameParser, 'parse', parse_method)
sut = PostProcessor(file_path=p['file_path'], nzb_name=p['nzb_name'])
# When
show, season, episodes, quality, version, airdate = sut._parse_info()
# Then
assert show is not None
assert p['expected'] == {
'show': show.name,
'season': season,
'episodes': episodes,
'quality': quality,
'version': version,
}
def setUp(self):
"""Set up tests."""
self.post_processor = PostProcessor(test.FILE_PATH)