def save(data): connect = Connect() cursor = connect.initConnect() query = ("INSERT INTO movies " "(title" ", original_title" ", year" ", release_date" ", duration" ", imdb_rating" ", imdb_id" ", imdb_poster)" "VALUES (%s, %s, %s, %s, %s, %s, %s, %s)") data = (data['title'] , data['original_title'] , data['year'] , data['release_date'] , data['duration'] , data['rating'] , data['imdb_id'] , data['poster'] ) cursor.execute(query, data) connect.commit() id_movie = cursor.lastrowid connect.closeConnect() return id_movie
def addDirector(id_movie, id_director): connect = Connect() cursor = connect.initConnect() query = ("INSERT INTO movies_directors " "(id_movie, id_director)" "VALUES (%s, %s)") data = (id_movie, id_director) cursor.execute(query, data) connect.commit() connect.closeConnect()
def addGenre(id_movie, id_genre): connect = Connect() cursor = connect.initConnect() query = ("INSERT INTO movies_genres " "(id_movie, id_genre)" "VALUES (%s, %s)") data = (id_movie, id_genre) cursor.execute(query, data) connect.commit() connect.closeConnect()
def getMoviesWithoutDirector(): connect = Connect() cursor = connect.initConnect() query = ("SELECT id_movie, imdb_id " " FROM movies " " WHERE id_movie NOT IN (SELECT id_movie FROM movies_directors) AND id_movie > 11844") cursor.execute(query) result = {} for (id_movie, imdb_id) in cursor: result[id_movie] = imdb_id connect.closeConnect() return result
def save(id_imdb): connect = Connect() cursor = connect.initConnect() query = ("INSERT INTO imdb " "(id_imdb)" "VALUES (%s)") data = (id_imdb,) try: cursor.execute(query, data) except MySQLdb.Error: raise connect.commit() connect.closeConnect()
def save(name): connect = Connect() cursor = connect.initConnect() query = ("INSERT INTO directors " "(name)" "VALUES (%s)") data = (name,) cursor.execute(query, data) connect.commit() id_director = cursor.lastrowid connect.closeConnect() return id_director
def find(name): connect = Connect() cursor = connect.initConnect() query = ("SELECT id_director, name " "FROM directors " "WHERE name = %s " "ORDER BY id_director ASC") data = (name,) cursor.execute(query, data) result = False for (id_director, name) in cursor: result = id_director break connect.closeConnect() return result
def getLastImdbID(): connect = Connect() cursor = connect.initConnect() query = ("SELECT MAX(id_imdb) AS last_imdb_id " "FROM imdb ") cursor.execute(query) result = False for (last_imdb_id) in cursor: if(last_imdb_id[0] == None): last_imdb_id = 0 else: last_imdb_id = int(last_imdb_id[0]) break nid = last_imdb_id + 1 try: Imdb.save(nid) connect.closeConnect() except MySQLdb.Error: connect.closeConnect() nid = Imdb.getLastImdbID() return nid