Example #1
0
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)
Example #2
0
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)