fix: collector season limit. Fix #470

EstrellaXD · Sep 25, 2023 · ba1674c · ba1674c
1 parent a5f9f85
commit ba1674c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/backend/src/module/manager/collector.py b/backend/src/module/manager/collector.py
@@ -9,7 +9,7 @@
 
 
 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
@@ -30,8 +30,9 @@ def get_torrents(
                     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
@@ -44,7 +44,7 @@ def special_url(data: Bangumi, site: str) -> RSSItem:
         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]