diff --git a/scanner/scanner/__init__.py b/scanner/scanner/__init__.py index 5167937d..25ab41f9 100644 --- a/scanner/scanner/__init__.py +++ b/scanner/scanner/__init__.py @@ -31,8 +31,9 @@ async def lifespan(_): if is_master: await migrate(); # creating the processor makes it listen to requests event in pg + provider = tmdb #CompositeProvider(tmdb) async with ( - RequestProcessor(db, client, CompositeProvider(tmdb)) as processor, + RequestProcessor(db, client, provider) as processor, get_db() as db, ): _ = asyncio.create_task(processor.process_all()) diff --git a/scanner/scanner/requests.py b/scanner/scanner/requests.py index e2bb5d70..89d7afa2 100644 --- a/scanner/scanner/requests.py +++ b/scanner/scanner/requests.py @@ -8,8 +8,8 @@ from asyncpg import Connection from pydantic import Field, TypeAdapter from .client import KyooClient -from .models.videos import Guess, Resource, Video -from .providers.composite import CompositeProvider +from .models.videos import Guess, Resource +from .providers.provider import Provider from .utils import Model logger = getLogger(__name__) @@ -54,7 +54,7 @@ class RequestProcessor: self, database: Connection, client: KyooClient, - providers: CompositeProvider, + providers: Provider, ): self._database = database self._client = client @@ -79,7 +79,9 @@ class RequestProcessor: try: found = await self.process_request() except Exception as e: - logger.error("Failed to process one of the metadata request", exc_info=e) + logger.error( + "Failed to process one of the metadata request", exc_info=e + ) async def process_request(self): cur = await self._database.fetchrow(