def add_movie(self, movie: Movie):
if movie.id is None:
movie.add_id(len(self._movies) + 1)
insort_left(self._movies, movie)
self._movies_index[int(movie.id)] = movie
for genre in movie.genres:
if genre not in self._genres:
self._genres.append(genre)
if movie.director is not None:
if movie.director not in self._directors:
self._directors.append(movie.director)
for actor in movie.actors:
if actor not in self._actors:
self._actors.append(actor)
def load_movies_and_tags(data_path: str, repository: MemoryRepository):
for row in read_csv_file(
os.path.join(data_path, 'Data1000MoviesWithPics.csv')):
movie = Movie(row[1], int(row[6]))
movie.description = row[3]
genre_list = row[2].split(",")
for genre in genre_list:
movie.add_genre(Genre(genre.strip()))
movie.add_director(Director(row[4]))
actors_list = row[5].split(",")
for actor in actors_list:
movie.add_actor(Actor(actor.strip()))
movie.add_id(int(row[0]))
path = "adapters/data/movie_posters_static/" + row[0] + ".jpg"
movie.add_imagelink(row[12])
repository.add_movie(movie)
def load_movies_and_tags(data_path: str, repo: MemoryRepository):
#Rank,Title,Genre,Description,Director,Actors,Year,Runtime (Minutes),Rating,Votes,Revenue (Millions),Metascore
for row in read_csv_file(os.path.join(data_path, 'Data1000Movies.csv')):
movie = Movie(row[1], int(row[6]))
movie.description = row[3]
# genres
genre_list = row[2].split(",")
for genre in genre_list:
movie.add_genre(Genre(genre.strip()))
# director
movie.add_director(Director(row[4]))
# actors
actors_list = row[5].split(",")
for actor in actors_list:
movie.add_actor(Actor(actor.strip()))
movie.add_id(int(row[0]))
path = "adapters/data/movie_posters_static/" + row[0] + ".jpg"
movie.add_imagelink(row[12])
# Add the movie to the repository.
repo.add_movie(movie)
def dict_to_movie(dict):
movie = Movie(dict.title, dict.year)
movie.add_id(dict.id)
# Note there's no reviews or tags.
return movie