Exemplo n.º 1
0
 def add_movie(self, title, year, description, director, actor, genre,
               runtime, rating, revenue, meta, vote):
     movie = Movie(title, year)
     movie.description = description
     movie.director = Director(director)
     actors = actor.split(",")
     for a in actors:
         movie.add_actor(Actor(a.strip()))
     genres = genre.split(",")
     for g in genres:
         movie.add_genre(Genre(g))
     movie.runtime = runtime
     movie.rating = rating
     if revenue != "N/A":
         movie.revenue = float(revenue)
     if meta != "N/A":
         movie.meta = int(meta)
     movie.vote = vote
     self.dataset_of_movies.append(movie)
Exemplo n.º 2
0
    def add_movie(self, id, title, year, description, director, actor, genre,
                  runtime, rating, revenue, meta, vote):
        movie = Movie(title, year)
        movie.description = description
        movie.director = [Director(director)]
        actors = actor.split(",")
        for a in actors:
            movie.add_actor(Actor(a.strip()))
        genres = genre.split(",")
        for g in genres:
            movie.add_genre(Genre(g))
        movie.runtime = runtime
        movie.rating = rating
        if revenue != "N/A":
            movie.revenue = float(revenue)
        if meta != "N/A":
            movie.meta = int(meta)
        movie.vote = vote
        movie.id = id

        with self._session_cm as scm:
            scm.session.add(movie)
            scm.commit()