def test_refine_episode():
episode = Episode(name=os.path.join(resources_dir, 'Refine.Episode.mkv'),
series='Refine',
season=0,
episode=0)
refine(episode)
assert episode.alternative_series == ['series']
assert episode.series_tvdb_id == 123456
def test_refine_episode_original_series(episodes):
episode = Episode(episodes['dallas_s01e03'].name,
episodes['dallas_s01e03'].series.lower(),
episodes['dallas_s01e03'].season,
episodes['dallas_s01e03'].episode)
refine(episode)
assert episode.series == episodes['dallas_s01e03'].series
assert episode.year == 1978
assert episode.series_imdb_id == 'tt0077000'
def test_refine_episode(episodes):
episode = Episode(episodes['bbt_s07e05'].name,
episodes['bbt_s07e05'].series.lower(),
episodes['bbt_s07e05'].season,
episodes['bbt_s07e05'].episode)
refine(episode)
assert episode.series == episodes['bbt_s07e05'].series
assert episode.year == episodes['bbt_s07e05'].year
assert episode.series_imdb_id == episodes['bbt_s07e05'].series_imdb_id
def test_refine_episode_year(episodes):
episode = Episode(episodes['dallas_2012_s01e03'].name,
episodes['dallas_2012_s01e03'].series.lower(),
episodes['dallas_2012_s01e03'].season,
episodes['dallas_2012_s01e03'].episode,
year=episodes['dallas_2012_s01e03'].year,
original_series=False)
refine(episode)
assert episode.series == episodes['dallas_2012_s01e03'].series
assert episode.year == episodes['dallas_2012_s01e03'].year
assert episode.series_imdb_id == 'tt1723760'
def test_refine_episode_no_year(episodes):
video = episodes['dallas_s01e03']
episode = Episode(video.name.lower(), video.series.lower(), video.season,
video.episode)
refine(episode)
assert episode.series == video.series
assert episode.year == video.year
assert episode.original_series == video.original_series
assert episode.title == video.title
assert episode.imdb_id == video.imdb_id
assert episode.series_imdb_id == video.series_imdb_id
assert episode.tvdb_id == video.tvdb_id
assert episode.series_tvdb_id == video.series_tvdb_id
def test_refine_ambiguous_2(episodes):
video = episodes['the_100_s03e09']
episode = Episode(video.name.lower(), video.series.lower(), video.season,
video.episode)
refine(episode)
assert episode.series == video.series
assert episode.year == video.year
assert episode.original_series == video.original_series
assert episode.title == video.title
assert episode.imdb_id == video.imdb_id
assert episode.series_imdb_id == video.series_imdb_id
assert episode.tvdb_id == video.tvdb_id
assert episode.series_tvdb_id == video.series_tvdb_id
def test_refine(episodes):
video = episodes['bbt_s07e05']
episode = Episode(video.name.lower(), video.series.lower(), video.season,
video.episode)
refine(episode)
assert episode.series == video.series
assert episode.year == video.year
assert episode.original_series == video.original_series
assert episode.title == video.title
assert episode.imdb_id == video.imdb_id
assert episode.series_imdb_id == video.series_imdb_id
assert episode.tvdb_id == video.tvdb_id
assert episode.series_tvdb_id == video.series_tvdb_id
def test_refine_episode():
episode = Episode(name=os.path.join(resources_dir, 'Refine.Episode.mkv'), series='Refine', season=0, episode=0)
wanted_item = WantedItem(type='episode', title='Series', year=2018, season=1, episode=[1, 2], source='Source',
quality='Resolution', codec='Codec', releasegrp='Group')
refine(episode, wanted_item)
assert episode.series == 'Series'
assert episode.year == 2018
assert episode.season == 1
assert episode.episode == 1 # Only first episode is kept when refining
assert episode.source == 'Source'
assert episode.resolution == 'Resolution'
assert episode.video_codec == 'Codec'
assert episode.release_group == 'Group'
def test_refine_episode_alternative_series(episodes):
video = episodes['turn_s04e03']
episode = Episode(video.name.lower(), video.series.lower(), video.season,
video.episode)
refine(episode)
assert episode.series == video.series
assert episode.year == video.year
assert episode.original_series == video.original_series
assert episode.title == video.title
assert episode.imdb_id == video.imdb_id
assert episode.series_imdb_id == video.series_imdb_id
assert episode.tvdb_id == video.tvdb_id
assert episode.series_tvdb_id == video.series_tvdb_id
assert episode.alternative_series == video.alternative_series
def test_refine_episode_partial(episodes):
video = episodes['csi_s15e18']
episode = Episode(video.name.lower(),
video.series.lower().split(':')[0], video.season,
video.episode)
refine(episode)
assert episode.series == video.series
assert episode.year == video.year
assert episode.original_series == video.original_series
assert episode.title == video.title
assert episode.imdb_id == video.imdb_id
assert episode.series_imdb_id == video.series_imdb_id
assert episode.tvdb_id == video.tvdb_id
assert episode.series_tvdb_id == video.series_tvdb_id
def usage():
print "Usage: " + sys.argv[0] + " lang series season episode"
sys.exit(1)
numargs = len(sys.argv)
if numargs != 5:
usage()
lang = sys.argv[1]
series = sys.argv[2]
season = int(sys.argv[3])
episode = int(sys.argv[4])
e = Episode(series + '.mp4', series, season, episode)
l = Language(lang)
pathname = None
subtitles = download_best_subtitles(set((e, )), set((l, )))
for video, video_subtitles in subtitles.items():
for video_subtitle in video_subtitles:
filename = get_subtitle_path(video.name, video_subtitle.language)
pathname = os.path.join(OUT_DIR, filename)
break
save_subtitles(subtitles, directory=OUT_DIR, encoding='utf-8')
if pathname is not None:
print('{"filename":"' + pathname + '"}')
else:
print("{}")
