From 33792da5740be63d97b6f786e0e60f82ffcd8871 Mon Sep 17 00:00:00 2001 From: localhost_frssoft Date: Sun, 11 Sep 2022 23:55:35 +0300 Subject: [PATCH] Oops, sqlite not support SELECT in executemany... --- src/imdb_datasets_worker.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) 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