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