def test_refine_movie(movies):
movie = Movie(movies['man_of_steel'].name,
movies['man_of_steel'].title.lower())
refine(movie)
assert movie.title == movies['man_of_steel'].title
assert movie.year == movies['man_of_steel'].year
assert movie.imdb_id == movies['man_of_steel'].imdb_id
def test_refine_movie():
movie = Movie(name=os.path.join(resources_dir, 'Refine.Movie.mkv'), title='Refine')
wanted_item = WantedItem(type='movie', title='Title', year=2018, source='Source', quality='Resolution',
codec='Codec', releasegrp='Group')
refine(movie, wanted_item)
assert movie.title == 'Title'
assert movie.year == 2018
assert movie.source == 'Source'
assert movie.resolution == 'Resolution'
assert movie.video_codec == 'Codec'
assert movie.release_group == 'Group'
def test_skip_refine():
movie = Movie(name=os.path.join(resources_dir,
'Refine.Movie.Not.Exists.mkv'),
title='Refine')
refine(movie)
assert movie.title == 'Refine'
def test_refine_movie():
movie = Movie(name=os.path.join(resources_dir, 'Refine.Movie.mkv'),
title='Refine')
refine(movie)
assert movie.alternative_titles == ['title']
assert movie.imdb_id == 'tt123456'
def test_skip_refine():
movie = Movie(name=os.path.join(resources_dir, 'Refine.Movie.Not.Exists.mkv'), title='Refine')
wanted_item = WantedItem(type='movie', title='Title')
refine(movie, wanted_item)
assert movie.title == 'Refine'
def usage():
print "Usage: " + sys.argv[0] + " lang title [year]"
sys.exit(1)
numargs = len(sys.argv)
if numargs < 3 or numargs > 4:
usage()
lang = sys.argv[1]
title = sys.argv[2]
year=None
if numargs == 4:
year = int(sys.argv[3])
m = Movie(title + '.mp4', title, year=year)
l = Language(lang)
pathname = None
subtitles = download_best_subtitles(set((m,)), 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("{}")
