azalea/MTfin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[F] Fix imdb info

Browse Source
This commit is contained in:
Azalea Gui
2026-03-10 17:17:25 -04:00
parent 862dc69b28
commit 4fdd9b2ff2
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils_imdb.py
+3 -7
View File
@@ -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. Fetch IMDb info from imdbapi.dev as a fallback or replacement for M-Team.
""" """
query = urllib.parse.quote(imdb_id) 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={ req = urllib.request.Request(url, headers={
'accept': 'application/json', 'accept': 'application/json',
@@ -19,12 +19,8 @@ def get_imdb_info(imdb_id: str) -> dict:
try: try:
with urllib.request.urlopen(req) as response: with urllib.request.urlopen(req) as response:
data = json.loads(response.read().decode()) data = json.loads(response.read().decode())
if data and "titles" in data and len(data["titles"]) > 0: if data and data.get("id") == imdb_id:
# Return the matching title info return {"code": "0", "message": "SUCCESS", "data": data}
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]}
except Exception as e: except Exception as e:
print(f"Error fetching from imdbapi for {imdb_id}: {e}") print(f"Error fetching from imdbapi for {imdb_id}: {e}")