Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-05-24 02:02:36 -04:00
Code Issues Packages Projects Releases Wiki Activity

Provider cleanup

Browse Source
This commit is contained in:
Zoe Roux 2025-05-16 12:57:31 +02:00
parent 418edb6c85
commit a3ae9fb64e
No known key found for this signature in database
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scanner/scanner/__init__.py
View File

@ -31,8 +31,9 @@ async def lifespan(_):
if is_master: if is_master:
await migrate(); await migrate();
# creating the processor makes it listen to requests event in pg # creating the processor makes it listen to requests event in pg
provider = tmdb #CompositeProvider(tmdb)
async with ( async with (
RequestProcessor(db, client, CompositeProvider(tmdb)) as processor, RequestProcessor(db, client, provider) as processor,
get_db() as db, get_db() as db,
): ):
_ = asyncio.create_task(processor.process_all()) _ = asyncio.create_task(processor.process_all())

10
scanner/scanner/requests.py
View File

@ -8,8 +8,8 @@ from asyncpg import Connection
from pydantic import Field, TypeAdapter from pydantic import Field, TypeAdapter
from .client import KyooClient from .client import KyooClient
from .models.videos import Guess, Resource, Video from .models.videos import Guess, Resource
from .providers.composite import CompositeProvider from .providers.provider import Provider
from .utils import Model from .utils import Model
logger = getLogger(__name__) logger = getLogger(__name__)
@ -54,7 +54,7 @@ class RequestProcessor:
self, self,
database: Connection, database: Connection,
client: KyooClient, client: KyooClient,
providers: CompositeProvider, providers: Provider,
): ):
self._database = database self._database = database
self._client = client self._client = client
@ -79,7 +79,9 @@ class RequestProcessor:
try: try:
found = await self.process_request() found = await self.process_request()
except Exception as e: 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): async def process_request(self):
cur = await self._database.fetchrow( cur = await self._database.fetchrow(