From ba1674c6bf665e7868882381973a473a11a7aa21 Mon Sep 17 00:00:00 2001 From: EstrellaXD Date: Mon, 25 Sep 2023 19:17:31 +0800 Subject: [PATCH] fix: collector season limit. Fix #470 --- backend/src/module/manager/collector.py | 2 +- backend/src/module/network/request_contents.py | 5 +++-- backend/src/module/searcher/searcher.py | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/backend/src/module/manager/collector.py b/backend/src/module/manager/collector.py index 927f5919..22e11456 100644 --- a/backend/src/module/manager/collector.py +++ b/backend/src/module/manager/collector.py @@ -9,7 +9,7 @@ logger = logging.getLogger(__name__) class SeasonCollector(DownloadClient): - def collect_season(self, bangumi: Bangumi, link: str = None, limit: int = 50): + def collect_season(self, bangumi: Bangumi, link: str = None, limit: int = None): logger.info( f"Start collecting {bangumi.official_title} Season {bangumi.season}..." ) diff --git a/backend/src/module/network/request_contents.py b/backend/src/module/network/request_contents.py index bde907a4..869fce62 100644 --- a/backend/src/module/network/request_contents.py +++ b/backend/src/module/network/request_contents.py @@ -30,8 +30,9 @@ class RequestContent(RequestURL): torrents.append( Torrent(name=_title, url=torrent_url, homepage=homepage) ) - if len(torrents) >= limit: - break + if isinstance(limit, int): + if len(torrents) >= limit: + break return torrents else: logger.warning(f"[Network] Failed to get torrents: {_url}") diff --git a/backend/src/module/searcher/searcher.py b/backend/src/module/searcher/searcher.py index 2dacbe40..6bc8beb1 100644 --- a/backend/src/module/searcher/searcher.py +++ b/backend/src/module/searcher/searcher.py @@ -44,7 +44,7 @@ class SearchTorrent(RequestContent, RSSAnalyser): url = search_url(site, keywords) return url - def search_season(self, data: Bangumi, site: str = "mikan", limit: int = 5) -> list[Torrent]: + def search_season(self, data: Bangumi, site: str = "mikan", limit: int = None) -> list[Torrent]: rss_item = self.special_url(data, site) torrents = self.search_torrents(rss_item, limit=limit) return [torrent for torrent in torrents if data.title_raw in torrent.name] \ No newline at end of file