def test_repository_can_add_a_review(in_memory_repo):
user = in_memory_repo.get_user('thorke')
movie = in_memory_repo.get_movie(1)
timestamp= datetime.today()
review = make_review("Trump's onto it!", user, movie, timestamp,5)
in_memory_repo.add_review(review)
assert review in in_memory_repo.get_reviews(1)
def load_reviews(data_path: str, repo: MemoryRepository, users):
for line in read_csv_file(os.path.join(data_path, 'reviews.csv')):
if line[4] is None:
review = make_review(
review_text=line[3],
user=users[line[1]],
movie=repo.get_movie(int(line[2])),
timestamp=line[4]
)
repo.add_review(review)
def load_reviews(data_path: str, repo: MemoryRepository, users):
ignore_row = 0
for data_row in read_csv_file(os.path.join(data_path, 'reviews.csv')):
if ignore_row == 0:
ignore_row += 1
continue
review = make_review(review_text=data_row[3],
user=users[data_row[1]],
movie=repo.get_movie(int(data_row[2])),
timestamp=datetime.fromisoformat(data_row[4]),
rating=int(data_row[5]))
repo.add_review(review)
def add_review(movie_id: int, review_text: str, username: str,
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.
review = make_review(review_text, user, movie)
# Update the repository.
repo.add_review(review)
def add_review(rate: int, rank: int, review_text: str, username: str,
repo: AbstractRepository):
# Check that the article exists.
movie = repo.get_movie(int(rank))
print(movie)
if movie is None:
raise NonExistentMovieException
user = repo.get_user(username)
if user is None:
raise UnknownUserException
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
print(timestamp)
# Create comment.
review = make_review(review_text, user, movie, timestamp, rate)
# Update the repository.
repo.add_review(review)