localhost_frssoft/ FMN_bot
1
0
Fork 0
mirror of https://gitea.phreedom.club/localhost_frssoft/FMN_bot synced 2024-11-22 11:39:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

Documentary movies will be dropped (Special for FMN, we not canceling this!)

This commit is contained in:
localhost_frssoft 2022-11-16 20:59:20 +03:00
parent 47f20fe4e8
commit eaf57df418
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/imdb_datasets_worker.py
View File

@ -27,6 +27,7 @@ def convert_tsv_to_db(title_basics_tsv):
original_name = line[3] original_name = line[3]
ru_name = None ru_name = None
year = line[5] year = line[5]
genres = line[-1].strip().split(',')
if year.startswith(r"\N"): if year.startswith(r"\N"):
year = None year = None
else: else:
@ -35,7 +36,11 @@ def convert_tsv_to_db(title_basics_tsv):
if tt_type not in ("movie", "tvMovie", "video"): if tt_type not in ("movie", "tvMovie", "video"):
original_name = None original_name = None
year = None year = None
if "Documentary" in genres:
logger.debug(f'Документальный {original_name} отсеян')
original_name = None
year = None
tt_type = "doc"
write_dataset.append((tt_id, tt_type, original_name, ru_name, year)) write_dataset.append((tt_id, tt_type, original_name, ru_name, year))
counter += 1 counter += 1
if counter >= chunk: if counter >= chunk: