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("{}")