fix site id

This commit is contained in:
jxxghp
2023-06-21 10:48:42 +08:00
parent c05a80239c
commit 957222ddbd
6 changed files with 64 additions and 9 deletions

View File

@ -148,12 +148,13 @@ class ChainBase(AbstractSingleton, metaclass=Singleton):
"""
return self.run_module("search_medias", meta=meta)
def search_torrents(self, mediainfo: Optional[MediaInfo], site: CommentedMap,
def search_torrents(self, site: CommentedMap,
mediainfo: Optional[MediaInfo] = None,
keyword: str = None) -> List[TorrentInfo]:
"""
搜索一个站点的种子资源
:param mediainfo: 识别的媒体信息
:param site: 站点
:param mediainfo: 识别的媒体信息
:param keyword: 搜索关键词,如有按关键词搜索,否则按媒体信息名称搜索
:reutrn: 资源列表
"""

View File

@ -41,11 +41,28 @@ class SearchChain(ChainBase):
def search_by_title(self, title: str) -> List[TorrentInfo]:
"""
根据标题搜索资源,不识别不过滤,直接返回站点内容
:param title: 标题,为空时返回所有站点首页内容
"""
logger.info(f'开始搜索资源,关键词:{title} ...')
# 搜索
return self.__search_all_sites(keyword=title)
def browse(self, domain: str, keyword: str = None) -> List[TorrentInfo]:
"""
浏览站点首页内容
:param domain: 站点域名
:param keyword: 关键词,有值时为搜索
"""
if not keyword:
logger.info(f'开始浏览站点首页内容,站点:{domain} ...')
else:
logger.info(f'开始搜索资源,关键词:{keyword},站点:{domain} ...')
site = self.siteshelper.get_indexer(domain)
if not site:
logger.error(f'站点 {domain} 不存在!')
return []
return self.search_torrents(site=site, keyword=keyword)
def process(self, mediainfo: MediaInfo,
keyword: str = None,
no_exists: Dict[int, Dict[int, NotExistMediaInfo]] = None) -> Optional[List[Context]]: