localhost_frssoft/ FMN_bot
1
0
Fork 0
mirror of https://gitea.phreedom.club/localhost_frssoft/FMN_bot synced 2024-05-15 17:29:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Correct fetching IMDB database

This commit is contained in:
localhost_frssoft 2022-09-07 02:10:14 +03:00
parent e782cacf5a
commit 4d02fde9aa
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/imdb_datasets_worker.py
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() return tt_list
tt_list.append(tt_film)
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