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
