def test_search_results(self, execute_task): task = execute_task('test_search_result') entry = task.entries[0] print entry['trakt_series_name'].lower() assert entry['trakt_series_name'].lower() == 'Shameless'.lower( ), 'lookup failed' with Session() as session: assert task.entries[1]['trakt_series_name'].lower( ) == 'Shameless'.lower(), 'second lookup failed' assert len(session.query(TraktShowSearchResult).all() ) == 1, 'should have added 1 show to search result' assert len(session.query(TraktShow).all() ) == 1, 'should only have added one show to show table' assert session.query(TraktShow).first().title == 'Shameless', 'should have added Shameless and' \ 'not Shameless (2011)' # change the search query session.query(TraktShowSearchResult).update( {'search': "Shameless.S01E03.HDTV-FlexGet"}) session.commit() lookupargs = {'title': "Shameless.S01E03.HDTV-FlexGet"} series = ApiTrakt.lookup_series(**lookupargs) assert series.tvdb_id == entry[ 'tvdb_id'], 'tvdb id should be the same as the first entry' assert series.id == entry[ 'trakt_show_id'], 'trakt id should be the same as the first entry' assert series.title.lower() == entry['trakt_series_name'].lower( ), 'series name should match first entry'
def lazy_watched_lookup(self, config, style, entry): """Does the lookup for this entry and populates the entry fields.""" if style == 'show' or style == 'episode': lookup = lookup_series trakt_id = entry.get('trakt_show_id', eval_lazy=True) else: lookup = lookup_movie trakt_id = entry.get('trakt_movie_id', eval_lazy=True) with Session() as session: lookupargs = {'trakt_id': trakt_id, 'session': session} try: item = lookup(**lookupargs) if style == 'episode': item = item.get_episode(entry['series_season'], entry['series_episode'], session) watched = ApiTrakt.watched(style, item, entry.get('title'), username=config.get('username'), account=config.get('account')) except LookupError as e: log.debug(e.args[0]) else: entry['trakt_watched'] = watched return entry
def get(self, session=None): args = lookup_parser.parse_args() try: result = at.lookup_series(session=session, **args) except LookupError as e: return {'status': 'error', 'message': e.args[0] }, 404 return jsonify(result.to_dict())
def get(self, session=None): args = lookup_parser.parse_args() try: result = at.lookup_series(session=session, **args) except LookupError as e: return {'status': 'error', 'message': e.args[0] }, 404 return jsonify(result.to_dict())
def get(self, session=None): args = lookup_parser.parse_args() include_actors = args.pop('include_actors') try: series = at.lookup_series(session=session, **args) except LookupError as e: return {'status': 'error', 'message': e.args[0] }, 404 result = series.to_dict() if include_actors: result["actors"] = list_actors(series.actors), return jsonify(result)
def test_search_results(self, execute_task): task = execute_task('test_search_results') entry = task.entries[0] assert entry['movie_name'].lower() == 'Harry Potter and The Philosopher\'s Stone'.lower(), 'lookup failed' with Session() as session: assert len(session.query(TraktMovieSearchResult).all()) == 1, 'should have added one movie to search result' # change the search query session.query(TraktMovieSearchResult).update({'search': "harry.potter.and.the.philosopher's"}) session.commit() lookupargs = {'title': "harry.potter.and.the.philosopher's"} movie = ApiTrakt.lookup_movie(**lookupargs) assert movie.imdb_id == entry['imdb_id'] assert movie.title.lower() == entry['movie_name'].lower()
def test_search_results(self, execute_task): task = execute_task('test_search_results') entry = task.entries[0] assert entry['movie_name'].lower() == 'Harry Potter and The Philosopher\'s Stone'.lower(), 'lookup failed' with Session() as session: assert len(session.query(TraktMovieSearchResult).all()) == 1, 'should have added one movie to search result' # change the search query session.query(TraktMovieSearchResult).update({'search': "harry.potter.and.the.philosopher's"}) session.commit() lookupargs = {'title': "harry.potter.and.the.philosopher's"} movie = ApiTrakt.lookup_movie(**lookupargs) assert movie.imdb_id == entry['imdb_id'] assert movie.title.lower() == entry['movie_name'].lower()
def get(self, title, session=None): args = lookup_parser.parse_args() include_actors = args.pop('include_actors') include_translations = args.pop('include_translations') kwargs = args kwargs['title'] = title try: movie = at.lookup_movie(session=session, **kwargs) except LookupError as e: return {'status': 'error', 'message': e.args[0] }, 404 result = movie.to_dict() if include_actors: result["actors"] = list_actors(movie.actors) if include_translations: result["translations"] = get_translations(movie.translate) return jsonify(result)
def lazy_watched_lookup(self, config, style, entry): """Does the lookup for this entry and populates the entry fields.""" if style == 'episode': lookup = lookup_series id = entry.get('trakt_show_id', eval_lazy=True) else: lookup = lookup_movie id = entry.get('trakt_movie_id', eval_lazy=True) with Session() as session: lookupargs = {'trakt_id': id, 'session': session} try: item = lookup(**lookupargs) if style == 'episode': item = item.get_episode(entry['series_season'], entry['series_episode']) watched = ApiTrakt.watched(config['username'], style, item, entry.get('title'), account=config.get('account')) except LookupError as e: log.debug(e.args[0]) else: entry['trakt_watched'] = watched return entry
def test_search_results(self, execute_task): task = execute_task('test_search_result') entry = task.entries[0] print(entry['trakt_series_name'].lower()) assert entry['trakt_series_name'].lower() == 'Shameless'.lower(), 'lookup failed' with Session() as session: assert task.entries[1]['trakt_series_name'].lower() == 'Shameless'.lower(), 'second lookup failed' assert len(session.query(TraktShowSearchResult).all()) == 1, 'should have added 1 show to search result' assert len(session.query(TraktShow).all()) == 1, 'should only have added one show to show table' assert session.query(TraktShow).first().title == 'Shameless', 'should have added Shameless and' \ 'not Shameless (2011)' # change the search query session.query(TraktShowSearchResult).update({'search': "Shameless.S01E03.HDTV-FlexGet"}) session.commit() lookupargs = {'title': "Shameless.S01E03.HDTV-FlexGet"} series = ApiTrakt.lookup_series(**lookupargs) assert series.tvdb_id == entry['tvdb_id'], 'tvdb id should be the same as the first entry' assert series.id == entry['trakt_id'], 'trakt id should be the same as the first entry' assert series.title.lower() == entry['trakt_series_name'].lower(), 'series name should match first entry'