def _search(self, title, year=None, language='en'): """ Search the api for a show. :param title: the title to search for :type title: str :param year: the year (optional) :type year: int or None :param language: the language to search for :type language: str :return: the search result or None if not found :rtype : tvdb_api_v2.models.series_search_result.SeriesSearchResult or None """ name = title if year: name += ' (' + text_type(year) + ')' log.info('Searching tvdb api for %s', name) # Make sure to convert it to native string before searching (to prevent unicode encoding error) search_results = self._client.search_series_by_name(s2n(name), language=language) # Only return 1 result or None for search_result in search_results.data: if sanitize(search_result.series_name) == sanitize(name): return search_result elif search_result.aliases: # If no match, fallback to aliases (if aliases are available) for alias in search_result.aliases: if sanitize(alias) == sanitize(name): return search_result else: continue return None
def skip_movie(self, wanted_item_index, title, year): if not wanted_item_index: raise cherrypy.HTTPError(400, 'No wanted_item index supplied') if not title: raise cherrypy.HTTPError(400, 'No title supplied') movie = title if year: movie += ' (' + year + ')' # Check if already skipped movie_sanitized = sanitize(movie) for x in autosubliminal.SKIPMOVIE: if movie_sanitized == sanitize(x): send_websocket_notification('Already skipped movie %s.' % movie) redirect('/home') # Skip movie if subchecker.skip_movie(wanted_item_index): config.write_config_property('skipmovie', movie, '00') config.apply_skipmovie() send_websocket_notification('Skipped movie %s.' % movie) else: send_websocket_notification( 'Could not skip movie! Please check the log file!', type='error') redirect('/home')
def _search(self, title, year=None, language='en'): """ Search the api for a show. :param title: the title to search for :type title: str :param year: the year (optional) :type year: int or None :param language: the language to search for :type language: str :return: the search result or None if not found :rtype : tvdb_api_v2.models.series_search_result.SeriesSearchResult or None """ name = title if year: name += ' (' + text_type(year) + ')' log.info('Searching tvdb api for %s', name) # Make sure to convert it to native string before searching (to prevent unicode encoding error) search_results = self._client.search_series_by_name(s2n(name), language=language) # Only return 1 result or None for search_result in search_results.data: if sanitize(search_result.series_name) == sanitize(name): return search_result elif search_result.aliases: # If no match, fallback to aliases (if aliases are available) for alias in search_result.aliases: if sanitize(alias) == sanitize(name): return search_result else: continue return None
def skip_movie(title, year): """Check if a movie should be skipped.""" movie = title if year: movie += ' (' + text_type(year) + ')' movie_sanitized = sanitize(movie) for x in autosubliminal.SKIPMOVIE: if movie_sanitized == sanitize(x): log.debug('Found match in skipmovie, skipping movie %s', movie) return True
def skip_movie(title, year): """Check if a movie should be skipped.""" movie = title if year: movie += ' (' + text_type(year) + ')' movie_sanitized = sanitize(movie) for x in autosubliminal.SKIPMOVIE: if movie_sanitized == sanitize(x): log.debug('Found match in skipmovie, skipping movie %s', movie) return True
def get_addic7ed_show_name_mapping(show_name): """ Get the addic7ed show name mapping for a show. :param show_name: the show name to get the addic7ed id for :return: the addic7ed show id or None :rtype: int | None """ show_name_sanitized = sanitize(show_name) for x in autosubliminal.ADDIC7EDSHOWNAMEMAPPING: if show_name_sanitized == sanitize(x): log.debug('Found match in addic7edshownamemapping for %s', show_name) return int(autosubliminal.ADDIC7EDSHOWNAMEMAPPING[x])
def skip_show(show_name, season): """Check if a show should be skipped.""" show_name_sanitized = sanitize(show_name) for x in autosubliminal.SKIPSHOW: if show_name_sanitized == sanitize(x): log.debug('Found match in skipshow for %s', show_name) for s in autosubliminal.SKIPSHOW[x].split(','): if s == '00': log.debug('Found all season match in skipshow, skipping all seasons for %s', show_name) return True elif int(s) == int(season): log.debug('Found season match in skipshow, skipping season %s for %s', season, show_name) return True
def get_alternative_show_name_mapping(show_name): """ Get the list of alternative show names for a show. :param show_name: the show name to get the alternatives for :return: a list of alternative show names or None :rtype: list | None """ show_name_sanitized = sanitize(show_name) for x in autosubliminal.ALTERNATIVESHOWNAMEMAPPING: if show_name_sanitized == sanitize(x): log.debug('Found match in alternativeshownamemapping for %s', show_name) alternatives = autosubliminal.ALTERNATIVESHOWNAMEMAPPING[x] return [sanitize(x) for x in alternatives.split(',')] # Needs to return a list
def get_movie_name_mapping(title, year): """ Get the imdb movie name mapping for a movie. :param title: the title of the movie :param year: the year of the movie :return: the imdb id of the movie or None :rtype: str | None """ movie = title if year: movie += ' (' + text_type(year) + ')' movie_sanitized = sanitize(movie) for x in autosubliminal.MOVIENAMEMAPPING: if movie_sanitized == sanitize(x): log.debug('Found match in movienamemapping for %s', movie) return autosubliminal.MOVIENAMEMAPPING[x]
def skip_show(show_name, season): """Check if a show should be skipped.""" show_name_sanitized = sanitize(show_name) for x in autosubliminal.SKIPSHOW: if show_name_sanitized == sanitize(x): log.debug('Found match in skipshow for %s', show_name) for s in autosubliminal.SKIPSHOW[x].split(','): if s == '00': log.debug( 'Found all season match in skipshow, skipping all seasons for %s', show_name) return True elif int(s) == int(season): log.debug( 'Found season match in skipshow, skipping season %s for %s', season, show_name) return True
def skip_show(self, wanted_item_index, title, season=None): if not season: return PageTemplate('/home/home-skipshow.mako').render( wanted_item_index=wanted_item_index, title=title) else: if not wanted_item_index: raise cherrypy.HTTPError(400, 'No wanted_item index supplied') if not title: raise cherrypy.HTTPError(400, 'No show supplied') # Check if season is a number to be sure if not season == '00': season = text_type(int(season)) config_season = season # Check if already skipped title_sanitized = sanitize(title) for x in autosubliminal.SKIPSHOW: if title_sanitized == sanitize(x): for s in autosubliminal.SKIPSHOW[x].split(','): if s == season or s == '00': send_websocket_notification( 'Already skipped show %s season %s.' % (title, season)) redirect('/home') # Not skipped yet, skip all or append season the seasons to skip if season == '00': config_season = '00' else: seasons = autosubliminal.SKIPSHOW[x].split(',') seasons.append(season) config_season = ','.join(sorted(seasons)) # Skip show if subchecker.skip_show(wanted_item_index, season): config.write_config_property('skipshow', title, config_season) config.apply_skipshow() if season == '00': send_websocket_notification( 'Skipped show %s all seasons.' % title) else: send_websocket_notification('Skipped show %s season %s.' % (title, season)) else: send_websocket_notification( 'Could not skip show! Please check the log file!', type='error') redirect('/home')
def get_alternative_movie_name_mapping(title, year): """ Get the list of alternative movie titles (without year). :param title: the title of the movie :param year: the year of the movie :return: a list of alternative titles (without year) or None :rtype: list | None """ movie = title if year: movie += ' (' + text_type(year) + ')' movie_sanitized = sanitize(movie) for x in autosubliminal.ALTERNATIVEMOVIENAMEMAPPING: if movie_sanitized == sanitize(x): log.debug('Found match in alternativemovienamemapping for %s', movie) alternatives = autosubliminal.ALTERNATIVEMOVIENAMEMAPPING[x] return [sanitize(x) for x in alternatives.split(',')] # Needs to return a list
def sanitize_imdb_title(string_value, ignore_characters=None): # Remove (I), (II), ... from imdb titles (this is added when there are multiple titles with the same name) # Example response from imdb: see http://www.imdb.com/find?q=Aftermath&s=tt&mx=20 string_value = re.sub('^(.+)(\(\w+\))$', r'\1', string_value) # Sanitize on ascii level (replaces à by a) string_value = unidecode(string_value) # Return sanitized title return sanitize(string_value, ignore_characters)
def sanitize_imdb_title(string_value, ignore_characters=None): # Remove (I), (II), ... from imdb titles (this is added when there are multiple titles with the same name) # Example response from imdb: see http://www.imdb.com/find?q=Aftermath&s=tt&mx=20 string_value = re.sub('^(.+)(\(\w+\))$', r'\1', string_value) # Sanitize on ascii level (replaces à by a) string_value = unidecode(string_value) # Return sanitized title return sanitize(string_value, ignore_characters)
def skip_show(self, wanted_item_index, title, season=None): if not season: return PageTemplate('/home/home-skipshow.mako').render(wanted_item_index=wanted_item_index, title=title) else: if not wanted_item_index: raise cherrypy.HTTPError(400, 'No wanted_item index supplied') if not title: raise cherrypy.HTTPError(400, 'No show supplied') # Check if season is a number to be sure if not season == '00': season = text_type(int(season)) config_season = season # Check if already skipped title_sanitized = sanitize(title) for x in autosubliminal.SKIPSHOW: if title_sanitized == sanitize(x): for s in autosubliminal.SKIPSHOW[x].split(','): if s == season or s == '00': send_websocket_notification('Already skipped show %s season %s.' % (title, season)) redirect('/home') # Not skipped yet, skip all or append season the seasons to skip if season == '00': config_season = '00' else: seasons = autosubliminal.SKIPSHOW[x].split(',') seasons.append(season) config_season = ','.join(sorted(seasons)) # Skip show if subchecker.skip_show(wanted_item_index, season): config.write_config_property('skipshow', title, config_season) config.apply_skipshow() if season == '00': send_websocket_notification('Skipped show %s all seasons.' % title) else: send_websocket_notification('Skipped show %s season %s.' % (title, season)) else: send_websocket_notification('Could not skip show! Please check the log file!', type='error') redirect('/home')
def skip_movie(self, wanted_item_index, title, year): if not wanted_item_index: raise cherrypy.HTTPError(400, 'No wanted_item index supplied') if not title: raise cherrypy.HTTPError(400, 'No title supplied') movie = title if year: movie += ' (' + year + ')' # Check if already skipped movie_sanitized = sanitize(movie) for x in autosubliminal.SKIPMOVIE: if movie_sanitized == sanitize(x): send_websocket_notification('Already skipped movie %s.' % movie) redirect('/home') # Skip movie if subchecker.skip_movie(wanted_item_index): config.write_config_property('skipmovie', movie, '00') config.apply_skipmovie() send_websocket_notification('Skipped movie %s.' % movie) else: send_websocket_notification('Could not skip movie! Please check the log file!', type='error') redirect('/home')
def test_sanitize(): assert sanitize(None) is None assert sanitize('(Mr.-Robot! / :),') == 'mr robot'