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]