Add the movie database as a provider

scanner/providers/implementations/themoviedatabase.py

@@ -0,0 +1,61 @@
+import asyncio
+import logging
+from aiohttp import ClientSession
+from typing import Optional, Any
+
+from ..provider import Provider
+from ..types.movie import Movie, MovieTranslation
+from ..types.status import Status
+
+
+class TheMovieDatabase(Provider):
+	def __init__(self, client: ClientSession, api_key: str) -> None:
+		super().__init__()
+		self._client = client
+		self.base = "https://api.themoviedb.org/3"
+		self.api_key = api_key
+
+
+	async def get(self, path: str, *, params: dict[str, Any] = {}):
+		params = {k:v for k,v in params.items() if v is not None}
+		async with self._client.get(f"{self.base}/{path}", params={ "api_key": self.api_key, **params }) as r:
+			r.raise_for_status()
+			return await r.json()
+
+
+	async def identify_movie(self, name: str, year: Optional[int], *, language: list[str]) -> Movie:
+		movie_id = (await self.get("search/movie", params={ "query": name, "year": year }))["results"][0]["id"]
+
+		async def for_language(lng: str) -> Movie:
+			movie = await self.get(f"/movie/{movie_id}", params={ "language": lng, "append_to_response": ""})
+			logging.debug("TMDb responded: %s", movie)
+
+			ret = (
+				Movie(
+					aliases=[],
+					release_date=None,
+					status=Status.UNKNOWN,
+					studio=None,
+					genres=[],
+					poster=[],
+					thumbnails=[],
+					logo=[],
+				)
+			)
+			translation = (
+				MovieTranslation(
+					name=movie["title"],
+					keywords=[],
+					overview=movie["overview"],
+				)
+			)
+			ret.translations = { lng: translation }
+			return ret
+
+		# TODO: make the folllowing generic
+		tasks = map(lambda lng: for_language(lng), language)
+		movies: list[Movie] = await asyncio.gather(*tasks)
+		movie = movies[0]
+		movie.translations = {k: v.translations[k] for k, v in zip(language, movies)}
+		return movie
+

scanner/providers/provider.py

@@ -0,0 +1,25 @@
+import os
+from aiohttp import ClientSession
+from abc import abstractmethod
+from typing import Optional, TypeVar
+
+from .types.movie import Movie
+
+
+Self = TypeVar("Self", bound="Provider")
+
+class Provider:
+	@classmethod
+	def get_all(cls: type[Self], client: ClientSession) -> list[Self]:
+		providers = []
+
+		from providers.implementations.themoviedatabase import TheMovieDatabase
+		tmdb =os.environ.get("THEMOVIEDB_APIKEY")
+		if tmdb:
+			providers.append(TheMovieDatabase(client, tmdb))
+
+		return providers
+
+	@abstractmethod
+	async def identify_movie(self, name: str, year: Optional[int], *, language: list[str]) -> Movie:
+		raise NotImplementedError