def _do_search_subtitles(self, videos_hashcode, languages):
""" Search list of subtitles. """
subtitles_infos = []
# Search subtitles
hashcodes_sizes = [
{'moviehash': hash_code, 'moviebytesize': video.size}
for hash_code, video in videos_hashcode.items()
]
response = self._proxy.SearchSubtitles(
self._session_string, hashcodes_sizes)
if self.status_ok(response):
if 'data' in response and response['data']:
for data_subtitle in response['data']:
# Retrieve important info
sub_video_hashcode = data_subtitle['MovieHash']
sub_video = videos_hashcode[sub_video_hashcode]
sub_lang = Language.fromopensubtitles(
data_subtitle['SubLanguageID'])
if sub_lang in sub_video.languages_to_download \
and sub_lang in languages:
# Subtitle infos
sub_id = data_subtitle['IDSubtitleFile']
sub_rating = float(data_subtitle['SubRating'])
sub_format = data_subtitle['SubFormat']
# Video infos
sub_video_name = data_subtitle['MovieName']
if data_subtitle['MovieKind'] == "movie":
sub_video = VideoFactory.make_from_type(
sub_video, Movie)
elif data_subtitle['MovieKind'] == "episode":
sub_video = VideoFactory.make_from_type(
sub_video, Episode)
videos_hashcode[sub_video_hashcode] = sub_video
if isinstance(sub_video, Movie):
sub_video.name = sub_video_name
elif isinstance(sub_video, Episode):
# Retrieves serie name and episode name
match_result = re.match(
self._series_regexp, sub_video_name)
sub_video.name = match_result.group("serie_name")
sub_video.episode_name = match_result.group(
"episode_name")
sub_video.season = int(
data_subtitle['SeriesSeason'])
sub_video.episode = int(
data_subtitle['SeriesEpisode'])
subtitle = Subtitle(
sub_id, sub_lang, sub_video,
sub_rating, sub_format)
subtitles_infos.append(subtitle)
else:
raise SubtitleServerError(
self, "There is no result when searching for subtitles.")
else:
raise SubtitleServerError(self, self.get_status_reason(response))
return subtitles_infos
