feat 订阅支持选择站点范围

app/chain/search.py

@@ -84,12 +84,14 @@ class SearchChain(ChainBase):
 
     def process(self, mediainfo: MediaInfo,
                 keyword: str = None,
-                no_exists: Dict[int, Dict[int, NotExistMediaInfo]] = None) -> List[Context]:
+                no_exists: Dict[int, Dict[int, NotExistMediaInfo]] = None,
+                sites: List[int] = None) -> List[Context]:
         """
         根据媒体信息搜索种子资源，精确匹配，应用过滤规则，同时根据no_exists过滤本地已存在的资源
         :param mediainfo: 媒体信息
         :param keyword: 搜索关键词
         :param no_exists: 缺失的媒体信息
+        :param sites: 站点ID列表，为空时搜索所有站点
         """
         logger.info(f'开始搜索资源，关键词：{keyword or mediainfo.title} ...')
         # 补充媒体信息
@@ -109,7 +111,8 @@ class SearchChain(ChainBase):
         # 执行搜索
         torrents: List[TorrentInfo] = self.__search_all_sites(
             mediainfo=mediainfo,
-            keyword=keyword
+            keyword=keyword,
+            sites=sites
         )
         if not torrents:
             logger.warn(f'{keyword or mediainfo.title} 未搜索到资源')
@@ -199,17 +202,22 @@ class SearchChain(ChainBase):
         return contexts
 
     def __search_all_sites(self, mediainfo: Optional[MediaInfo] = None,
-                           keyword: str = None) -> Optional[List[TorrentInfo]]:
+                           keyword: str = None,
+                           sites: List[int] = None) -> Optional[List[TorrentInfo]]:
         """
         多线程搜索多个站点
         :param mediainfo:  识别的媒体信息
         :param keyword:  搜索关键词，如有按关键词搜索，否则按媒体信息名称搜索
+        :param sites:  指定站点ID列表，如有则只搜索指定站点，否则搜索所有站点
         :reutrn: 资源列表
         """
         # 未开启的站点不搜索
         indexer_sites = []
         # 配置的索引站点
-        config_indexers = [str(sid) for sid in self.systemconfig.get(SystemConfigKey.IndexerSites) or []]
+        if sites:
+            config_indexers = [str(sid) for sid in sites]
+        else:
+            config_indexers = [str(sid) for sid in self.systemconfig.get(SystemConfigKey.IndexerSites) or []]
         for indexer in self.siteshelper.get_indexers():
             # 检查站点索引开关
             if not config_indexers or str(indexer.get("id")) in config_indexers:

app/chain/subscribe.py

@@ -201,10 +201,16 @@ class SubscribeChain(ChainBase):
                 start_episode=subscribe.start_episode,
 
             )
+            # 站点范围
+            if subscribe.sites:
+                sites = json.loads(subscribe.sites)
+            else:
+                sites = None
             # 搜索
             contexts = self.searchchain.process(mediainfo=mediainfo,
                                                 keyword=subscribe.keyword,
-                                                no_exists=no_exists)
+                                                no_exists=no_exists,
+                                                sites=sites)
             if not contexts:
                 logger.warn(f'订阅 {subscribe.keyword or subscribe.name} 未搜索到资源')
                 # 未搜索到资源，但本地缺失可能有变化，更新订阅剩余集数
@@ -357,6 +363,11 @@ class SubscribeChain(ChainBase):
                     torrent_meta = context.meta_info
                     torrent_mediainfo = context.media_info
                     torrent_info = context.torrent_info
+                    # 不在订阅站点范围的不处理
+                    if subscribe.sites:
+                        sub_sites = json.loads(subscribe.sites)
+                        if sub_sites and torrent_info.site not in sub_sites:
+                            continue
                     # 如果是电视剧过滤掉已经下载的集数
                     if torrent_mediainfo.type == MediaType.TV:
                         if self.__check_subscribe_note(subscribe, torrent_meta.episode_list):