def test_get_movies_by_multiple_id(in_memory_repo): target_movie_ids = [5, 6, 9990, 88888] movies_as_dict = movie_library_services.get_movies_by_id(target_movie_ids, in_memory_repo) assert len(movies_as_dict) == 2 movie_ids = [movie['id'] for movie in movies_as_dict] assert set([5, 6]).issubset(movie_ids)
def test_get_movies_by_multiple_id(in_memory_repo): target_movie_ids = [5, 6, 9990, 88888] movies_as_dict = movie_library_services.get_movies_by_id( target_movie_ids, in_memory_repo) # Check that 2 movies were returned from the query. assert len(movies_as_dict) == 2 # Check that the movie ids returned were 5 and 6. movie_ids = [movie['id'] for movie in movies_as_dict] assert set([5, 6]).issubset(movie_ids)
def movies_by_director(): movies_per_page = 3 # Read query parameters. director_name = request.args.get('director') cursor = request.args.get('cursor') movie_to_show_reviews = request.args.get('view_reviews_for') if movie_to_show_reviews is None: # No view-reviews query parameter, so set to a non-existent movie id. movie_to_show_reviews = -1 else: # Convert movie_to_show_reviews from string to int. movie_to_show_reviews = int(movie_to_show_reviews) if cursor is None: # No cursor query parameter, so initialise cursor to start at the beginning. cursor = 0 else: # Convert cursor from string to int. cursor = int(cursor) # Retrieve movie ids for movies that have director_name. movie_ids = services.get_movie_ids_for_director(director_name, repo.repo_instance) # Retrieve the batch of movies to display on the Web page. movies = services.get_movies_by_id( movie_ids[cursor:cursor + movies_per_page], repo.repo_instance) first_movie_url = None last_movie_url = None next_movie_url = None prev_movie_url = None if cursor > 0: # There are preceding movies, so generate URLs for the 'previous' and 'first' navigation buttons. prev_movie_url = url_for('movie_library_bp.movies_by_director', director=director_name, cursor=cursor - movies_per_page) first_movie_url = url_for('movie_library_bp.movies_by_director', director=director_name) if cursor + movies_per_page < len(movie_ids): # There are further movies, so generate URLs for the 'next' and 'last' navigation buttons. next_movie_url = url_for('movie_library_bp.movies_by_director', director=director_name, cursor=cursor + movies_per_page) last_cursor = movies_per_page * int(len(movie_ids) / movies_per_page) if len(movie_ids) % movies_per_page == 0: last_cursor -= movies_per_page last_movie_url = url_for('movie_library_bp.movies_by_director', director=director_name, cursor=last_cursor) # Construct urls for viewing movie reviews and adding reviews. for movie in movies: movie['view_review_url'] = url_for( 'movie_library_bp.movies_by_director', director=director_name, cursor=cursor, view_reviews_for=movie['id']) movie['add_review_url'] = url_for( 'movie_library_bp.write_review_on_movie', movie=movie['id']) movie['add_to_watchlist_url'] = url_for( 'movie_library_bp.add_to_watchlist', movie=movie['id']) # Generate the webpage to display the movies. return render_template('movie_library/movies.html', title='Movies', movies_title='Movies directed by ' + director_name, movies=movies, selected_movies=utilities.get_selected_movies(3), actor_urls=utilities.get_actors_and_urls(), director_urls=utilities.get_directors_and_urls(), genre_urls=utilities.get_genres_and_urls(), first_movie_url=first_movie_url, last_movie_url=last_movie_url, prev_movie_url=prev_movie_url, next_movie_url=next_movie_url, show_reviews_for_movie=movie_to_show_reviews, is_watchlist=False)