def get_reviews_for_movie(movie_id, repo: AbstractRepository):
movie = repo.get_movie(movie_id)
if movie is None:
raise NonExistentMovieException
return reviews_to_dict(movie.reviews)
def get_comments_for_movie(movie_rank, repo: AbstractRepository):
movie = repo.get_movie(movie_rank)
if movie is None:
raise NonExistentMovieException
return comments_to_dict(movie.comments)
def get_movie(movie_rank: int, repo: AbstractRepository):
movie = repo.get_movie(movie_rank)
if movie is None:
raise NonExistentMovieException
return movie_to_dict(movie, repo)
def get_movie(movie_title: str, repo: AbstractRepository):
movie = repo.get_movie(movie_title)
if movie is None:
raise NonExistentMovieException
return movie_to_dict(movie)
def get_movie(movie_id: int, repo: AbstractRepository):
try:
movie = repo.get_movie(movie_id)
except:
raise NonExistentMovieException
if movie is None:
raise NonExistentMovieException
return movie_to_dict(movie)
def add_review(title: str, release_year: int, username: str, review_text: str, rating: int, repo: AbstractRepository):
movie = repo.get_movie(title=title, release_year=release_year)
if movie is None:
raise NonExistentMovieException
user = repo.get_user(username=username)
if user is None:
raise UnknownUserException
review = Review(movie, review_text, rating)
user.add_review(review)
repo.add_review(review)
def set_rating(movie_id: int, rating: int, username: str,
repo: AbstractRepository):
user = repo.get_user(username)
if user is None:
user = User('Guest account', 'defaultpass')
movie = repo.get_movie(movie_id)
if movie is None:
raise NonExistentMovieException
repo.set_rating(rating, user, movie)
def add_review(movie_title: str, review_text: str, username: str, rating: int,
repo: AbstractRepository):
movie = repo.get_movie(movie_title)
if movie is None:
raise NonExistentMovieException
user = repo.get_user(username)
if user is None:
raise UnknownUserException
review = make_review(review_text, user, movie, rating)
repo.add_review(review)
def add_review(movie_id: int, review_text: str, rating, username: str, repo: AbstractRepository):
movie = repo.get_movie(movie_id)
if movie is None:
raise NonExistentMovieException
user = repo.get_user(username)
if user is None:
raise UnknownUserException
# Create Review
review = make_review(review_text, user, movie, rating)
# Update Repo
repo.add_review(review)
def add_review(movie_id: int, review_text: str, username: str, rating, repo: AbstractRepository):
# Check that the movie exists.
movie = repo.get_movie(movie_id)
if movie is None:
raise NonExistentMovieException
user = repo.get_user(username)
if user is None:
raise UnknownUserException
# Create review.
print(movie)
review = make_review(user, movie, review_text, rating)
# Update the repository.
repo.add_review(review)
def add_comment(movie_rank: int, comment_text: str, username: str,
repo: AbstractRepository):
# Check that the movie exists.
movie = repo.get_movie(movie_rank)
if movie is None:
raise NonExistentMovieException
user = repo.get_user(username)
if user is None:
user = User("Guest account", "Abcd1234")
# Create comment.
comment = make_comment(comment_text, user, movie)
# Update the repository.
repo.add_comment(comment)
def get_movie(title: str, release_year: int, repo: AbstractRepository):
movie = repo.get_movie(title=title, release_year=release_year)
return movie
def remove_from_watchlist(movie_id: int, repo: AbstractRepository):
if 'username' not in session:
return
user = repo.get_user(session['username'])
movie = repo.get_movie(movie_id)
user.remove_from_watchlist(movie)
def get_movie(rank, repo: AbstractRepository):
movie = repo.get_movie(rank)
print(movie)
return movie_to_dict(movie)
