add 搜索结果保存与查询

app/chain/douban.py

@@ -46,10 +46,10 @@ class DoubanChain(ChainBase):
         mediainfo: MediaInfo = self.recognize_media(meta=meta)
         if not mediainfo:
             logger.warn(f'{meta.name} 未识别到TMDB媒体信息')
-            return Context(meta=meta, mediainfo=MediaInfo(douban_info=doubaninfo))
+            return Context(meta_info=meta, media_info=MediaInfo(douban_info=doubaninfo))
         logger.info(f'{doubanid} 识别到媒体信息：{mediainfo.type.value} {mediainfo.title_year}{meta.season}')
         mediainfo.set_douban_info(doubaninfo)
-        return Context(meta=meta, mediainfo=mediainfo)
+        return Context(meta_info=meta, media_info=mediainfo)
 
     def movie_top250(self, page: int = 1, count: int = 30) -> List[dict]:
         """

app/chain/media.py

@@ -28,12 +28,12 @@ class MediaChain(ChainBase):
         mediainfo: MediaInfo = self.recognize_media(meta=metainfo)
         if not mediainfo:
             logger.warn(f'{title} 未识别到媒体信息')
-            return Context(meta=metainfo)
+            return Context(meta_info=metainfo)
         logger.info(f'{title} 识别到媒体信息：{mediainfo.type.value} {mediainfo.title_year}')
         # 更新媒体图片
         self.obtain_images(mediainfo=mediainfo)
         # 返回上下文
-        return Context(meta=metainfo, mediainfo=mediainfo, title=title, subtitle=subtitle)
+        return Context(meta_info=metainfo, media_info=mediainfo)
 
     def search(self, title: str) -> Tuple[MetaBase, List[MediaInfo]]:
         """

app/chain/search.py

@@ -1,3 +1,4 @@
+import pickle
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime
 from typing import Dict
@@ -8,11 +9,12 @@ from app.core.config import settings
 from app.core.context import Context
 from app.core.context import MediaInfo, TorrentInfo
 from app.core.metainfo import MetaInfo
+from app.db.systemconfig_oper import SystemConfigOper
 from app.helper.progress import ProgressHelper
 from app.helper.sites import SitesHelper
 from app.log import logger
 from app.schemas import NotExistMediaInfo
-from app.schemas.types import MediaType, ProgressKey
+from app.schemas.types import MediaType, ProgressKey, SystemConfigKey
 from app.utils.string import StringUtils
 
 
@@ -25,6 +27,7 @@ class SearchChain(ChainBase):
         super().__init__()
         self.siteshelper = SitesHelper()
         self.progress = ProgressHelper()
+        self.systemconfig = SystemConfigOper()
 
     def search_by_tmdbid(self, tmdbid: int, mtype: MediaType = None) -> Optional[List[Context]]:
         """
@@ -36,7 +39,10 @@ class SearchChain(ChainBase):
         if not mediainfo:
             logger.error(f'{tmdbid} 媒体信息识别失败！')
             return None
-        return self.process(mediainfo=mediainfo)
+        results = self.process(mediainfo=mediainfo)
+        # 保存眲结果
+        self.systemconfig.set(SystemConfigKey.SearchResults, pickle.dumps(results))
+        return results
 
     def search_by_title(self, title: str) -> List[TorrentInfo]:
         """
@@ -47,6 +53,15 @@ class SearchChain(ChainBase):
         # 搜索
         return self.__search_all_sites(keyword=title)
 
+    def last_search_results(self) -> List[Context]:
+        """
+        获取上次搜索结果
+        """
+        results = self.systemconfig.get(SystemConfigKey.SearchResults)
+        if not results:
+            return []
+        return pickle.loads(results)
+
     def browse(self, domain: str, keyword: str = None) -> List[TorrentInfo]:
         """
         浏览站点首页内容
@@ -168,9 +183,9 @@ class SearchChain(ChainBase):
             _match_torrents = torrents
         logger.info(f"匹配完成，共匹配到 {len(_match_torrents)} 个资源")
         # 组装上下文返回
-        return [Context(meta=MetaInfo(title=torrent.title, subtitle=torrent.description),
-                        mediainfo=mediainfo,
-                        torrentinfo=torrent) for torrent in _match_torrents]
+        return [Context(meta_info=MetaInfo(title=torrent.title, subtitle=torrent.description),
+                        media_info=mediainfo,
+                        torrent_info=torrent) for torrent in _match_torrents]
 
     def __search_all_sites(self, mediainfo: Optional[MediaInfo] = None,
                            keyword: str = None) -> Optional[List[TorrentInfo]]:

app/chain/subscribe.py

@@ -304,7 +304,7 @@ class SubscribeChain(ChainBase):
                         logger.warn(f'未识别到媒体信息，标题：{torrent.title}')
                         continue
                     # 上下文
-                    context = Context(meta=meta, mediainfo=mediainfo, torrentinfo=torrent)
+                    context = Context(meta_info=meta, media_info=mediainfo, torrent_info=torrent)
                     self._torrents_cache[domain].append(context)
         # 从缓存中匹配订阅
         self.match()