[O] Adjustable imdb source

This commit is contained in:
2026-03-10 17:03:05 -04:00
parent 9d1f995b8a
commit 862dc69b28
2 changed files with 49 additions and 8 deletions
+31
View File
@@ -0,0 +1,31 @@
import json
import urllib.request
import urllib.parse
from utils import with_disk_cache
@with_disk_cache('imdbapi_info')
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}"
req = urllib.request.Request(url, headers={
'accept': 'application/json',
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:148.0) Gecko/20100101 Firefox/148.0'
})
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]}
except Exception as e:
print(f"Error fetching from imdbapi for {imdb_id}: {e}")
return {"code": "1", "message": "NOT_FOUND", "data": {}}