159 lines
7.1 KiB
Python
159 lines
7.1 KiB
Python
from typing import Optional, List, Dict
|
||
|
||
from app.chain import ChainBase
|
||
from app.core.config import settings
|
||
from app.core.context import Context, MediaInfo, TorrentInfo
|
||
from app.core.metainfo import MetaInfo
|
||
from app.helper.sites import SitesHelper
|
||
from app.log import logger
|
||
from app.schemas import NotExistMediaInfo
|
||
from app.schemas.types import MediaType
|
||
from app.utils.string import StringUtils
|
||
|
||
|
||
class SearchChain(ChainBase):
|
||
"""
|
||
站点资源搜索处理链
|
||
"""
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.siteshelper = SitesHelper()
|
||
|
||
def search_by_tmdbid(self, tmdbid: int, mtype: str = None) -> Optional[List[Context]]:
|
||
"""
|
||
根据TMDB ID搜索资源，精确匹配，但不不过滤本地存在的资源
|
||
:param tmdbid: TMDB ID
|
||
:param mtype: 媒体，电影 or 电视剧
|
||
"""
|
||
mediainfo = self.recognize_media(tmdbid=tmdbid, mtype=mtype)
|
||
if not mediainfo:
|
||
logger.error(f'{tmdbid} 媒体信息识别失败！')
|
||
return None
|
||
return self.process(mediainfo=mediainfo)
|
||
|
||
def search_by_title(self, title: str, site_ids: List[int] = None) -> List[TorrentInfo]:
|
||
"""
|
||
根据标题搜索资源，不识别不过滤，直接返回站点内容
|
||
"""
|
||
logger.info(f'开始搜索资源，关键词：{title} ...')
|
||
# 未开启的站点不搜索
|
||
indexer_sites = []
|
||
for indexer in self.siteshelper.get_indexers():
|
||
if not settings.INDEXER_SITES \
|
||
or any([s in indexer.get("domain") for s in settings.INDEXER_SITES.split(',')]):
|
||
if site_ids and indexer.get("id") not in site_ids:
|
||
continue
|
||
indexer_sites.append(indexer)
|
||
if not indexer_sites:
|
||
logger.warn('未开启任何有效站点，无法搜索资源')
|
||
return []
|
||
# 搜索
|
||
return self.search_torrents(mediainfo=None, sites=indexer_sites, keyword=title)
|
||
|
||
def process(self, mediainfo: MediaInfo,
|
||
keyword: str = None,
|
||
no_exists: Dict[int, Dict[int, NotExistMediaInfo]] = None) -> Optional[List[Context]]:
|
||
"""
|
||
根据媒体信息搜索种子资源，精确匹配，应用过滤规则，同时根据no_exists过滤本地已存在的资源
|
||
:param mediainfo: 媒体信息
|
||
:param keyword: 搜索关键词
|
||
:param no_exists: 缺失的媒体信息
|
||
"""
|
||
logger.info(f'开始搜索资源，关键词：{keyword or mediainfo.title} ...')
|
||
# 未开启的站点不搜索
|
||
indexer_sites = []
|
||
for indexer in self.siteshelper.get_indexers():
|
||
if not settings.INDEXER_SITES \
|
||
or any([s in indexer.get("domain") for s in settings.INDEXER_SITES.split(',')]):
|
||
# 站点流控
|
||
state, msg = self.siteshelper.check(indexer.get("domain"))
|
||
if not state:
|
||
logger.warn(msg)
|
||
continue
|
||
indexer_sites.append(indexer)
|
||
if not indexer_sites:
|
||
logger.warn('未开启任何有效站点，无法搜索资源')
|
||
return []
|
||
# 补充媒体信息
|
||
if not mediainfo.names:
|
||
mediainfo: MediaInfo = self.recognize_media(mtype=mediainfo.type,
|
||
tmdbid=mediainfo.tmdb_id)
|
||
if not mediainfo:
|
||
logger.error(f'媒体信息识别失败！')
|
||
return []
|
||
# 缺失的媒体信息
|
||
if no_exists and no_exists.get(mediainfo.tmdb_id):
|
||
# 过滤剧集
|
||
season_episodes = {sea: info.episodes
|
||
for sea, info in no_exists[mediainfo.tmdb_id].items()}
|
||
else:
|
||
season_episodes = None
|
||
# 执行搜索
|
||
torrents: List[TorrentInfo] = self.search_torrents(
|
||
mediainfo=mediainfo,
|
||
keyword=keyword,
|
||
sites=indexer_sites
|
||
)
|
||
if not torrents:
|
||
logger.warn(f'{keyword or mediainfo.title} 未搜索到资源')
|
||
return []
|
||
# 过滤种子
|
||
logger.info(f'开始过滤资源，当前规则：{settings.FILTER_RULE} ...')
|
||
result: List[TorrentInfo] = self.filter_torrents(torrent_list=torrents,
|
||
season_episodes=season_episodes)
|
||
if result is not None:
|
||
torrents = result
|
||
if not torrents:
|
||
logger.warn(f'{keyword or mediainfo.title} 没有符合过滤条件的资源')
|
||
return []
|
||
# 过滤不匹配的资源
|
||
logger.info(f'开始匹配，总 {len(torrents)} 个资源 ...')
|
||
_match_torrents = []
|
||
if mediainfo:
|
||
for torrent in torrents:
|
||
# 比对IMDBID
|
||
if torrent.imdbid \
|
||
and mediainfo.imdb_id \
|
||
and torrent.imdbid == mediainfo.imdb_id:
|
||
logger.info(f'{mediainfo.title} 匹配到资源：{torrent.site_name} - {torrent.title}')
|
||
_match_torrents.append(torrent)
|
||
continue
|
||
# 识别前预处理
|
||
result: Optional[tuple] = self.prepare_recognize(title=torrent.title, subtitle=torrent.description)
|
||
if result:
|
||
title, subtitle = result
|
||
else:
|
||
title, subtitle = torrent.title, torrent.description
|
||
# 识别
|
||
torrent_meta = MetaInfo(title=title, subtitle=subtitle)
|
||
# 比对年份
|
||
if torrent_meta.year and mediainfo.year:
|
||
if mediainfo.type == MediaType.TV:
|
||
# 剧集
|
||
if torrent_meta.year not in [year for year in mediainfo.season_years.values()]:
|
||
continue
|
||
else:
|
||
# 没有季的剧集或者电影
|
||
if torrent_meta.year != mediainfo.year:
|
||
continue
|
||
# 比对标题
|
||
if torrent_meta.name in [mediainfo.title, mediainfo.original_title]:
|
||
logger.info(f'{mediainfo.title} 匹配到资源：{torrent.site_name} - {torrent.title}')
|
||
_match_torrents.append(torrent)
|
||
continue
|
||
# 比对别名和译名
|
||
for name in mediainfo.names:
|
||
if StringUtils.clear(name).strip().upper() == \
|
||
StringUtils.clear(torrent_meta.name).strip().upper():
|
||
logger.info(f'{mediainfo.title} 匹配到资源：{torrent.site_name} - {torrent.title}')
|
||
_match_torrents.append(torrent)
|
||
break
|
||
else:
|
||
_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]
|