def make_movie():
movie = Movie('Avengers: Endgame', 2019)
movie.set_director(Director("Anthony Russo"))
movie.set_description(
"After Thanos, an intergalactic warlord, disintegrates half of the universe, the Avengers must reunite and assemble again to reinvigorate their trounced allies and restore balance."
)
movie.set_runtime_minutes(181)
return movie
def load_movies_actors_directors_genre_description(data_path: str, repo: MemoryRepository):
genres = dict()
actors = dict()
directors = dict()
for data_row in read_csv_file(os.path.join(data_path, "movies.csv")):
movie_index = int(data_row[0])
title = data_row[1]
try:
movie_revenue = float(data_row[10])
except ValueError:
movie_revenue = 0
list_of_genre_names = data_row[2].split(",")
list_of_genres = [Genre(genre_name) for genre_name in list_of_genre_names] # can have duplicate
movie_description = data_row[3]
director = Director(data_row[4]) # can have duplicate
list_of_actor_names = data_row[5].split(",")
list_of_actor_names = [name.strip() for name in list_of_actor_names]
list_of_actors = [Actor(actor_full_name) for actor_full_name in list_of_actor_names] # can have duplicate
release_year = int(data_row[6])
runtime = int(data_row[7])
# Create Movie object
movie = Movie(
title=title,
release_year=release_year,
id=movie_index
)
movie.set_revenue(revenue=movie_revenue)
# Add movie to repo
repo.add_movie(movie)
# Add any new genres to repo
for genre in list_of_genres:
if genre not in genres.keys():
genres[genre] = list()
genres[genre].append(movie_index)
for actor in list_of_actors:
if actor not in actors.keys():
actors[actor] = list()
actors[actor].append(movie_index)
# Add any new directors to dict
if director not in directors:
directors[director] = list()
directors[director].append(movie_index)
# Connect the current movie with its attributes
add_movie_attributes(movie=movie,
list_of_genres=list_of_genres,
description=movie_description,
list_of_actors=list_of_actors,
director=director,
runtime=runtime)
# Associate Genres with Movies and add them to the repository
for genre in genres.keys():
for current_movie_index in genres[genre]:
movie = repo.get_movie_by_index(current_movie_index)
movie.add_genre(genre)
genre.add_Movie(movie)
repo.add_genre(genre)
# Associate Actors with Movies and add them to the repository
for actor in actors.keys():
for current_movie_index in actors[actor]:
movie = repo.get_movie_by_index(current_movie_index)
movie.add_actor(actor)
actor.add_played_movies(movie)
repo.add_actor(actor)
# Associate Directors with Movies and add them to the repositor
for director in directors.keys():
for current_movie_index in directors[director]:
movie = repo.get_movie_by_index(current_movie_index)
movie.set_director(director)
director.add_directed_movies(movie)
repo.add_director(director)
def test_director(self):
movie1 = Movie("Moana", 2016, 1)
movie1.set_director(Director("Ron Clements"))
assert repr(movie1.director) == "<Director Ron Clements>"
