Ejemplo n.º 1
0
def load_movies(data_path, repo):
    filename = os.path.join(data_path, "movies.csv")
    with open(filename, mode='r', encoding='utf-8-sig') as csvfile:
        movie_file_reader = csv.DictReader(csvfile)

        for row in movie_file_reader:
            media_id = row["Id"]
            title = row['Title']
            release_year = int(row['Year'])
            description = row["Description"]
            runtime_minutes = row["Runtime (Minutes)"]
            director = Director(row["Director"])
            genres = set([Genre(x) for x in row["Genre"].split(',')])
            actors = set([Actor(x) for x in row["Actors"].split(',')])

            movie = Movie(title, release_year)
            movie.description = description
            movie.runtime_minutes = runtime_minutes
            movie.director = director
            movie.media_id = media_id

            for genre in genres:
                movie.add_genre(genre)

            for actor in actors:
                movie.add_actor(actor)

            repo.add_media(movie)
Ejemplo n.º 2
0
def test_description():
    movie1 = Movie("Moana", 2000)
    assert movie1.description == ""
    movie1.description = "test description"
    assert movie1.description == "test description"
    movie1.description = "test description "
    assert movie1.description == "test description"
    movie1.description = "test description  "
    assert movie1.description == "test description"
    movie1.description = " test description"
    assert movie1.description == "test description"
    movie1.description = "  test description"
    assert movie1.description == "test description"
    movie1.description = " test description "
    assert movie1.description == "test description"
    movie1.description = "  test description  "
    assert movie1.description == "test description"