def repository(): repo = MemoryRepository() user1 = User("Myles Kennedy", "123") user2 = User("Mark Tremonti", "456") director1 = Director("Taika Waititi") director2 = Director("Steven Speilberg") actor1 = Actor("Chris Pratt") actor2 = Actor("Emma Watson") genre1 = Genre("Comedy") genre2 = Genre("Animation") movie1 = Movie("Moana", 2016) movie2 = Movie("Ice Age", 2002) movie1.director = director1 movie2.director = director2 movie1.add_actor(actor1) movie2.add_actor(actor2) movie1.add_genre(genre1) movie2.add_genre(genre2) review1 = Review(movie1, "very nice", 9) review2 = Review(movie2, "incredible", 10) user1.add_review(review1) user2.add_review(review2) repo.add_movie(movie1) repo.add_movie(movie2) repo.add_director(director1) repo.add_director(director2) repo.add_actor(actor1) repo.add_actor(actor2) repo.add_genre(genre1) repo.add_genre(genre2) repo.add_review(review1) repo.add_review(review2) repo.add_user(user1) repo.add_user(user2) return repo
def load_data(data_path: str, repo: MemoryRepository): reader = MovieFileCSVReader(os.path.join(data_path, 'data1000Movies.csv')) reader.read_csv_file() for actor in reader.dataset_of_actors: repo.add_actor(actor) for movie in reader.dataset_of_movies: repo.add_movie(movie) for actor in movie.actors: for colleague in movie.actors: if not actor.check_if_this_actor_worked_with( colleague) and actor != colleague: actor.add_actor_colleague(colleague) for director in reader.dataset_of_directors: repo.add_director(director) for genre in reader.dataset_of_genres: repo.add_genre(genre) user = User(user_name="Myles Kennedy", password=generate_password_hash("123")) review = Review(repo.get_movie("Inception", 2010), "Absolutely incredible movie!", 10) repo.add_review(review) user.add_review(review) user.watchlist.add_movie(repo.get_movie("The Da Vinci Code", 2006)) user.watchlist.add_movie(repo.get_movie("Moana", 2016)) repo.add_user(user)
def test_user_review_property(): movie1 = Movie("Princess and The Frog", 2009) user1 = User("Jenna da Cruz", "testing123") review1 = Review(movie1, "I loved it", 10) user1.add_review(review1) assert user1.reviews == [review1]
def test_user_add_review(): movie1 = Movie("Princess and The Frog", 2009) review1 = Review(movie1, "I really liked it", 9) user1 = User("Jenna da Cruz", "testing123") user1.add_review(review1) assert user1.reviews == [review1]