init

app/chain/search.py

@@ -0,0 +1,73 @@
+from typing import Optional, List
+
+from app.chain import _ChainBase
+from app.chain.common import CommonChain
+from app.core import Context, MetaInfo, MediaInfo, TorrentInfo
+from app.core.meta import MetaBase
+from app.helper.sites import SitesHelper
+from app.log import logger
+
+
+class SearchChain(_ChainBase):
+    """
+    站点资源搜索处理链
+    """
+
+    def __init__(self):
+        super().__init__()
+        self.common = CommonChain()
+        self.siteshelper = SitesHelper()
+
+    def process(self, meta: MetaBase, mediainfo: MediaInfo,
+                keyword: str = None) -> Optional[List[Context]]:
+        """
+        根据媒体信息，执行搜索
+        :param meta: 元数据
+        :param mediainfo: 媒体信息
+        :param keyword: 搜索关键词
+        """
+        # 执行搜索
+        logger.info(f'开始搜索资源，关键词：{keyword or mediainfo.title} ...')
+        torrents: List[TorrentInfo] = self.run_module(
+            'search_torrents',
+            mediainfo=mediainfo,
+            keyword=keyword,
+            sites=self.siteshelper.get_indexers()
+        )
+        if not torrents:
+            logger.warn(f'{keyword or mediainfo.title} 未搜索到资源')
+            return []
+        # 过滤不匹配的资源
+        _match_torrents = []
+        if mediainfo:
+            for torrent in torrents:
+                # 比对IMDBID
+                if torrent.imdbid \
+                        and mediainfo.imdb_id \
+                        and torrent.imdbid == mediainfo.imdb_id:
+                    _match_torrents.append(torrent)
+                    continue
+                # 识别
+                torrent_meta = MetaInfo(torrent.title, torrent.description)
+                # 识别媒体信息
+                torrent_mediainfo: MediaInfo = self.run_module('recognize_media', meta=torrent_meta)
+                if not torrent_mediainfo:
+                    logger.warn(f'未识别到媒体信息，标题：{torrent.title}')
+                    continue
+                # 过滤
+                if torrent_mediainfo.tmdb_id == mediainfo.tmdb_id \
+                        and torrent_mediainfo.type == mediainfo.type:
+                    _match_torrents.append(torrent)
+        else:
+            _match_torrents = torrents
+        # 过滤种子
+        result = self.run_module("filter_torrents", torrent_list=_match_torrents)
+        if result is not None:
+            _match_torrents = result
+        if not _match_torrents:
+            logger.warn(f'{keyword or mediainfo.title} 没有符合过滤条件的资源')
+            return []
+        # 组装上下文返回
+        return [Context(meta=MetaInfo(torrent.title),
+                        mediainfo=mediainfo,
+                        torrentinfo=torrent) for torrent in _match_torrents]