Correct fetching IMDB database

This commit is contained in:
localhost_frssoft 2022-09-07 02:10:14 +03:00
parent e782cacf5a
commit 4d02fde9aa

View File

@ -84,20 +84,14 @@ def convert_datasets_to_db():
def get_title_by_id(films_ids=list): def get_title_by_id(films_ids=list):
tt_list = [] tt_list = c.executemany("SELECT * FROM titles WHERE tt_id = (?)", films_ids).fetchall()
for i in films_ids: if tt_list != []:
tt_film = c.execute("SELECT * FROM titles WHERE tt_id = (?)", (i,)).fetchone()
tt_list.append(tt_film)
return tt_list return tt_list
def get_title_by_names_and_year(film_names=list): def get_title_by_names_and_year(film_names=list):
tt_list = [] tt_list = c.executemany('''SELECT * FROM titles
for i in film_names:
tt_film = c.execute('''SELECT * FROM titles
WHERE (original_name = (?) OR ru_name = (?)) WHERE (original_name = (?) OR ru_name = (?))
AND year = (?)''', i).fetchone() AND year = (?)''', film_names).fetchall()
if tt_film:
tt_list.append(tt_film)
if tt_list != []: if tt_list != []:
return tt_list return tt_list