From 58e84812d079771e7590a69b2639138b98e255e1 Mon Sep 17 00:00:00 2001 From: EstrellaXD Date: Wed, 4 Oct 2023 17:01:39 +0800 Subject: [PATCH] Fix: test. --- backend/src/module/parser/analyser/tmdb_parser.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/backend/src/module/parser/analyser/tmdb_parser.py b/backend/src/module/parser/analyser/tmdb_parser.py index 7dfd7b72..63de3f2e 100644 --- a/backend/src/module/parser/analyser/tmdb_parser.py +++ b/backend/src/module/parser/analyser/tmdb_parser.py @@ -56,7 +56,7 @@ def get_season(seasons: list) -> tuple[int, str]: return len(ss), ss[-1].get("poster_path") -def tmdb_parser(title, language) -> TMDBInfo | None: +def tmdb_parser(title, language, test: bool = False) -> TMDBInfo | None: with RequestContent() as req: url = search_url(title) contents = req.get_json(url).get("results") @@ -86,8 +86,9 @@ def tmdb_parser(title, language) -> TMDBInfo | None: official_title = info_content.get("name") year_number = info_content.get("first_air_date").split("-")[0] if poster_path: - img = req.get_content(f"https://image.tmdb.org/t/p/w780{poster_path}") - poster_link = save_image(img, "jpg") + if not test: + img = req.get_content(f"https://image.tmdb.org/t/p/w780{poster_path}") + poster_link = save_image(img, "jpg") else: poster_link = None return TMDBInfo(