48 lines
2.1 KiB
Python
48 lines
2.1 KiB
Python
import requests
|
|
|
|
from stream_fusion.utils.metdata.metadata_provider_base import MetadataProvider
|
|
from stream_fusion.utils.models.movie import Movie
|
|
from stream_fusion.utils.models.series import Series
|
|
from stream_fusion.settings import settings
|
|
from stream_fusion.logging_config import logger
|
|
|
|
class TMDB(MetadataProvider):
|
|
def get_metadata(self, id, type):
|
|
self.logger.info("Getting metadata for " + type + " with id " + id)
|
|
|
|
full_id = id.split(":")
|
|
|
|
result = None
|
|
|
|
for lang in self.config['languages']:
|
|
url = f"https://api.themoviedb.org/3/find/{full_id[0]}?api_key={settings.tmdb_api_key}&external_source=imdb_id&language={lang}"
|
|
response = requests.get(url)
|
|
data = response.json()
|
|
logger.trace(data)
|
|
|
|
if lang == self.config['languages'][0]:
|
|
if type == "movie":
|
|
result = Movie(
|
|
id=id,
|
|
tmdb_id=data["movie_results"][0]["id"],
|
|
titles=[self.replace_weird_characters(data["movie_results"][0]["title"])],
|
|
year=data["movie_results"][0]["release_date"][:4],
|
|
languages=self.config['languages']
|
|
)
|
|
else:
|
|
result = Series(
|
|
id=id,
|
|
tmdb_id = data["tv_results"][0]["id"],
|
|
titles=[self.replace_weird_characters(data["tv_results"][0]["name"])],
|
|
season="S{:02d}".format(int(full_id[1])),
|
|
episode="E{:02d}".format(int(full_id[2])),
|
|
languages=self.config['languages']
|
|
)
|
|
else:
|
|
if type == "movie":
|
|
result.titles.append(self.replace_weird_characters(data["movie_results"][0]["title"]))
|
|
else:
|
|
result.titles.append(self.replace_weird_characters(data["tv_results"][0]["name"]))
|
|
|
|
self.logger.info("Got metadata for " + type + " with id " + id)
|
|
return result
|