diff --git a/src/imdb_datasets_worker.py b/src/imdb_datasets_worker.py index b380c1f..9352bb8 100644 --- a/src/imdb_datasets_worker.py +++ b/src/imdb_datasets_worker.py @@ -84,14 +84,22 @@ def convert_datasets_to_db(): def get_title_by_id(films_ids=list): - tt_list = c.executemany("SELECT * FROM titles WHERE tt_id = (?)", films_ids).fetchall() + tt_list = [] + for i in films_ids: + tt_film = c.execute("SELECT * FROM titles WHERE tt_id = (?)", (i,)).fetchone() + if tt_film: + tt_list.append(tt_film) if tt_list != []: return tt_list def get_title_by_names_and_year(film_names=list): - tt_list = c.executemany('''SELECT * FROM titles + tt_list = [] + for i in film_names: + tt_film = c.execute('''SELECT * FROM titles WHERE (original_name = (?) OR ru_name = (?)) - AND year = (?)''', film_names).fetchall() + AND year = (?)''', i).fetchone() + if tt_film: + tt_list.append(tt_film) if tt_list != []: return tt_list