From 4fdd9b2ff28a4c9039aeb6923df43a6d8e921e1b Mon Sep 17 00:00:00 2001 From: Azalea <22280294+hykilpikonna@users.noreply.github.com> Date: Tue, 10 Mar 2026 17:17:25 -0400 Subject: [PATCH] [F] Fix imdb info --- utils_imdb.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/utils_imdb.py b/utils_imdb.py index 98055de..9178cbc 100644 --- a/utils_imdb.py +++ b/utils_imdb.py @@ -9,7 +9,7 @@ def get_imdb_info(imdb_id: str) -> dict: Fetch IMDb info from imdbapi.dev as a fallback or replacement for M-Team. """ query = urllib.parse.quote(imdb_id) - url = f"https://api.imdbapi.dev/search/titles?query={query}" + url = f"https://api.imdbapi.dev/titles/{query}" req = urllib.request.Request(url, headers={ 'accept': 'application/json', @@ -19,12 +19,8 @@ def get_imdb_info(imdb_id: str) -> dict: try: with urllib.request.urlopen(req) as response: data = json.loads(response.read().decode()) - if data and "titles" in data and len(data["titles"]) > 0: - # Return the matching title info - for title in data["titles"]: - if title.get("id") == imdb_id: - return {"code": "0", "message": "SUCCESS", "data": title} - return {"code": "0", "message": "SUCCESS", "data": data["titles"][0]} + if data and data.get("id") == imdb_id: + return {"code": "0", "message": "SUCCESS", "data": data} except Exception as e: print(f"Error fetching from imdbapi for {imdb_id}: {e}")