diff --git a/comet/scrapers/stremthru.py b/comet/scrapers/stremthru.py index e922311..a433222 100644 --- a/comet/scrapers/stremthru.py +++ b/comet/scrapers/stremthru.py @@ -5,6 +5,7 @@ import aiohttp from comet.core.logger import log_scraper_error, logger from comet.scrapers.base import BaseScraper from comet.scrapers.models import ScrapeRequest +from comet.services.anime import anime_mapper class StremthruScraper(BaseScraper): @@ -15,8 +16,14 @@ class StremthruScraper(BaseScraper): torrents = [] try: + media_id = request.media_only_id + if "kitsu" in request.media_id: + imdb_id = anime_mapper.get_imdb_from_kitsu(int(media_id)) + if imdb_id: + media_id = imdb_id + data = await self.session.get( - f"{self.url}/v0/torznab/api?t=search&imdbid={request.media_only_id}" + f"{self.url}/v0/torznab/api?t=search&imdbid={media_id}" ) data_text = await data.text() diff --git a/comet/services/anime.py b/comet/services/anime.py index a9a9ddd..87f1fde 100644 --- a/comet/services/anime.py +++ b/comet/services/anime.py @@ -237,7 +237,9 @@ class AnimeMapper: except asyncio.CancelledError: raise except Exception as exc: - logger.warning(f"Anime mapping refresh loop encountered an error: {exc}") + logger.warning( + f"Anime mapping refresh loop encountered an error: {exc}" + ) async def stop(self): if self._background_task: