def get_server_proxy(self, url, timeout=None): return ServerProxy( url, SubZeroRequestsTransport(use_https=self.use_ssl, timeout=timeout or self.timeout, user_agent=os.environ.get( "SZ_USER_AGENT", "Sub-Zero/2")))
def __init__(self, username=None, password=None, use_tag_search=False, only_foreign=False): if username is not None and password is None or username is None and password is not None: raise ConfigurationError('Username and password must be specified') self.username = username or '' self.password = password or '' self.use_tag_search = use_tag_search self.only_foreign = only_foreign if use_tag_search: logger.info("Using tag/exact filename search") if only_foreign: logger.info("Only searching for foreign/forced subtitles") super(PatchedOpenSubtitlesProvider, self).__init__() self.server = ServerProxy('http://api.opensubtitles.org/xml-rpc', TimeoutTransport(10))
def get_server_proxy(self, url, timeout=10): transport = TimeoutSafeTransport if url.startswith("https") else TimeoutTransport return ServerProxy(url, transport(timeout))
class PatchedOpenSubtitlesProvider(ProviderRetryMixin, OpenSubtitlesProvider): only_foreign = True def __init__(self, username=None, password=None, use_tag_search=False, only_foreign=False): if username is not None and password is None or username is None and password is not None: raise ConfigurationError('Username and password must be specified') self.username = username or '' self.password = password or '' self.use_tag_search = use_tag_search self.only_foreign = only_foreign if use_tag_search: logger.info("Using tag/exact filename search") if only_foreign: logger.info("Only searching for foreign/forced subtitles") super(PatchedOpenSubtitlesProvider, self).__init__() self.server = ServerProxy('http://api.opensubtitles.org/xml-rpc', TimeoutTransport(10)) def initialize(self): logger.info('Logging in') # fixme: retry on SSLError response = self.retry( lambda: checked( self.server.LogIn(self.username, self.password, 'eng', 'subliminal v%s' % get_version(__version__)) ) ) self.token = response['token'] logger.debug('Logged in with token %r', self.token) def list_subtitles(self, video, languages): """ :param video: :param languages: :return: patch: query movies even if hash is known; add tag parameter """ season = episode = None if isinstance(video, Episode): query = video.series season = video.season episode = video.episode # elif ('opensubtitles' not in video.hashes or not video.size) and not video.imdb_id: # query = video.name.split(os.sep)[-1] else: query = video.title return self.query(languages, hash=video.hashes.get('opensubtitles'), size=video.size, imdb_id=video.imdb_id, query=query, season=season, episode=episode, tag=os.path.basename(video.name), use_tag_search=self.use_tag_search, only_foreign=self.only_foreign) def query(self, languages, hash=None, size=None, imdb_id=None, query=None, season=None, episode=None, tag=None, use_tag_search=False, only_foreign=False): # fill the search criteria criteria = [] if hash and size: criteria.append({'moviehash': hash, 'moviebytesize': str(size)}) if use_tag_search and tag: criteria.append({'tag': tag}) if imdb_id: criteria.append({'imdbid': imdb_id}) if query and season and episode: criteria.append({'query': query, 'season': season, 'episode': episode}) elif query: criteria.append({'query': query}) if not criteria: raise ValueError('Not enough information') # add the language for criterion in criteria: criterion['sublanguageid'] = ','.join(sorted(l.opensubtitles for l in languages)) # query the server logger.info('Searching subtitles %r', criteria) response = self.retry(lambda: checked(self.server.SearchSubtitles(self.token, criteria))) subtitles = [] # exit if no data if not response['data']: logger.info('No subtitles found') return subtitles # loop over subtitle items for subtitle_item in response['data']: # read the item language = Language.fromopensubtitles(subtitle_item['SubLanguageID']) hearing_impaired = bool(int(subtitle_item['SubHearingImpaired'])) page_link = subtitle_item['SubtitlesLink'] subtitle_id = int(subtitle_item['IDSubtitleFile']) matched_by = subtitle_item['MatchedBy'] movie_kind = subtitle_item['MovieKind'] hash = subtitle_item['MovieHash'] movie_name = subtitle_item['MovieName'] movie_release_name = subtitle_item['MovieReleaseName'] movie_year = int(subtitle_item['MovieYear']) if subtitle_item['MovieYear'] else None movie_imdb_id = int(subtitle_item['IDMovieImdb']) movie_fps = subtitle_item.get('MovieFPS') series_season = int(subtitle_item['SeriesSeason']) if subtitle_item['SeriesSeason'] else None series_episode = int(subtitle_item['SeriesEpisode']) if subtitle_item['SeriesEpisode'] else None sub_file_name = subtitle_item.get('SubFileName') foreign_parts_only = bool(int(subtitle_item.get('SubForeignPartsOnly', 0))) # foreign/forced subtitles only wanted if only_foreign and not foreign_parts_only: continue # foreign/forced not wanted if not only_foreign and foreign_parts_only: continue query_parameters = subtitle_item.get("QueryParameters") subtitle = PatchedOpenSubtitlesSubtitle(language, hearing_impaired, page_link, subtitle_id, matched_by, movie_kind, hash, movie_name, movie_release_name, movie_year, movie_imdb_id, series_season, series_episode, query_parameters, fps=movie_fps) logger.debug('Found subtitle %r', subtitle) subtitles.append(subtitle) return subtitles
def get_server_proxy(self, url, timeout=10): return ServerProxy(url, SubZeroTransport(timeout, url))
def get_server_proxy(self, url, timeout=None): return ServerProxy(url, SubZeroRequestsTransport(use_https=self.use_ssl, timeout=timeout or self.timeout, user_agent="Bazarr/1"))