mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
255 lines
9.3 KiB
Python
255 lines
9.3 KiB
Python
import base64
|
||
import json
|
||
from typing import List, Optional, Tuple
|
||
|
||
from app.core.config import settings
|
||
from app.db.systemconfig_oper import SystemConfigOper
|
||
from app.log import logger
|
||
from app.schemas import MediaType
|
||
from app.utils.http import RequestUtils, AsyncRequestUtils
|
||
from app.utils.singleton import SingletonClass
|
||
from app.utils.string import StringUtils
|
||
|
||
|
||
class RousiSpider(metaclass=SingletonClass):
|
||
"""
|
||
Rousi.pro API v1 Spider
|
||
|
||
使用 API v1 接口进行种子搜索
|
||
- 认证方式：Bearer Token (Passkey)
|
||
- 搜索接口：/api/v1/torrents
|
||
- 详情接口：/api/v1/torrents/:id
|
||
"""
|
||
_indexerid = None
|
||
_domain = None
|
||
_url = None
|
||
_name = ""
|
||
_proxy = None
|
||
_cookie = None
|
||
_ua = None
|
||
_size = 20
|
||
_searchurl = "https://%s/api/v1/torrents"
|
||
_downloadurl = "https://%s/api/v1/torrents/%s"
|
||
_timeout = 15
|
||
_apikey = None
|
||
|
||
def __init__(self, indexer: dict):
|
||
self.systemconfig = SystemConfigOper()
|
||
if indexer:
|
||
self._indexerid = indexer.get('id')
|
||
self._url = indexer.get('domain')
|
||
self._domain = StringUtils.get_url_domain(self._url)
|
||
self._searchurl = self._searchurl % self._domain
|
||
self._downloadurl = self._downloadurl % (self._domain, "%s")
|
||
self._name = indexer.get('name')
|
||
if indexer.get('proxy'):
|
||
self._proxy = settings.PROXY
|
||
self._cookie = indexer.get('cookie')
|
||
self._ua = indexer.get('ua')
|
||
self._apikey = indexer.get('apikey')
|
||
self._timeout = indexer.get('timeout') or 15
|
||
|
||
def __get_params(self, keyword: str, mtype: MediaType = None, page: Optional[int] = 0) -> dict:
|
||
"""
|
||
构建 API 请求参数
|
||
|
||
:param keyword: 搜索关键词
|
||
:param mtype: 媒体类型 (MOVIE/TV)
|
||
:param page: 页码（从 0 开始，API 需要从 1 开始）
|
||
:return: 请求参数字典
|
||
"""
|
||
params = {
|
||
"page": int(page) + 1,
|
||
"page_size": self._size
|
||
}
|
||
if keyword:
|
||
params["keyword"] = keyword
|
||
|
||
# 注意：API 的 category 参数为字符串类型，仅支持单个分类
|
||
# 这里优先选择主分类，如需多分类搜索需要分别请求
|
||
if mtype:
|
||
if mtype == MediaType.MOVIE:
|
||
params["category"] = "movie"
|
||
elif mtype == MediaType.TV:
|
||
params["category"] = "tv"
|
||
|
||
return params
|
||
|
||
def __parse_result(self, results: List[dict]) -> List[dict]:
|
||
"""
|
||
解析搜索结果
|
||
|
||
将 API 返回的种子数据转换为 MoviePilot 标准格式
|
||
|
||
:param results: API 返回的种子列表
|
||
:return: 标准化的种子信息列表
|
||
"""
|
||
torrents = []
|
||
if not results:
|
||
return torrents
|
||
|
||
for result in results:
|
||
# 解析分类信息
|
||
raw_cat = result.get('category')
|
||
cat_val = None
|
||
|
||
category = MediaType.UNKNOWN.value
|
||
|
||
if isinstance(raw_cat, dict):
|
||
cat_val = raw_cat.get('slug') or raw_cat.get('name')
|
||
elif isinstance(raw_cat, str):
|
||
cat_val = raw_cat
|
||
|
||
if cat_val:
|
||
cat_val = str(cat_val).lower()
|
||
if cat_val in ['movie', 'documentary']:
|
||
category = MediaType.MOVIE.value
|
||
elif cat_val in ['tv', 'animation', 'variety', 'sports']:
|
||
category = MediaType.TV.value
|
||
|
||
# 解析促销信息
|
||
# API 后端已处理全站促销优先级，直接使用返回的 promotion 数据
|
||
downloadvolumefactor = 1.0
|
||
uploadvolumefactor = 1.0
|
||
freedate = None
|
||
|
||
promotion = result.get('promotion')
|
||
if promotion and promotion.get('is_active'):
|
||
downloadvolumefactor = float(promotion.get('down_multiplier', 1.0))
|
||
uploadvolumefactor = float(promotion.get('up_multiplier', 1.0))
|
||
# 促销到期时间，格式化为 YYYY-MM-DD HH:MM:SS
|
||
if promotion.get('until'):
|
||
freedate = StringUtils.unify_datetime_str(promotion.get('until'))
|
||
|
||
torrent = {
|
||
'title': result.get('title'),
|
||
'description': result.get('subtitle'),
|
||
'enclosure': self.__get_download_url(result.get('id')),
|
||
'pubdate': StringUtils.unify_datetime_str(result.get('created_at')),
|
||
'size': int(result.get('size') or 0),
|
||
'seeders': int(result.get('seeders') or 0),
|
||
'peers': int(result.get('leechers') or 0),
|
||
'grabs': int(result.get('downloads') or 0),
|
||
'downloadvolumefactor': downloadvolumefactor,
|
||
'uploadvolumefactor': uploadvolumefactor,
|
||
'freedate': freedate,
|
||
'page_url': f"https://{self._domain}/torrent/{result.get('uuid')}",
|
||
'labels': [],
|
||
'category': category
|
||
}
|
||
torrents.append(torrent)
|
||
return torrents
|
||
|
||
def search(self, keyword: str, mtype: MediaType = None, page: Optional[int] = 0) -> Tuple[bool, List[dict]]:
|
||
"""
|
||
同步搜索种子
|
||
|
||
:param keyword: 搜索关键词
|
||
:param mtype: 媒体类型 (MOVIE/TV)
|
||
:param page: 页码（从 0 开始）
|
||
:return: (是否发生错误, 种子列表)
|
||
"""
|
||
if not self._apikey:
|
||
logger.warn(f"{self._name} 未配置 API Key (Passkey)")
|
||
return True, []
|
||
|
||
params = self.__get_params(keyword, mtype, page)
|
||
headers = {
|
||
"Authorization": f"Bearer {self._apikey}",
|
||
"Accept": "application/json"
|
||
}
|
||
|
||
res = RequestUtils(
|
||
headers=headers,
|
||
proxies=self._proxy,
|
||
timeout=self._timeout
|
||
).get_res(url=self._searchurl, params=params)
|
||
|
||
if res and res.status_code == 200:
|
||
try:
|
||
data = res.json()
|
||
if data.get('code') == 0:
|
||
results = data.get('data', {}).get('torrents', [])
|
||
return False, self.__parse_result(results)
|
||
else:
|
||
logger.warn(f"{self._name} 搜索失败，错误信息：{data.get('message')}")
|
||
return True, []
|
||
except Exception as e:
|
||
logger.warn(f"{self._name} 解析响应失败：{e}")
|
||
return True, []
|
||
elif res is not None:
|
||
logger.warn(f"{self._name} 搜索失败，HTTP 错误码：{res.status_code}")
|
||
return True, []
|
||
else:
|
||
logger.warn(f"{self._name} 搜索失败，无法连接 {self._domain}")
|
||
return True, []
|
||
|
||
async def async_search(self, keyword: str, mtype: MediaType = None, page: Optional[int] = 0) -> Tuple[bool, List[dict]]:
|
||
"""
|
||
异步搜索种子
|
||
|
||
:param keyword: 搜索关键词
|
||
:param mtype: 媒体类型 (MOVIE/TV)
|
||
:param page: 页码（从 0 开始）
|
||
:return: (是否发生错误, 种子列表)
|
||
"""
|
||
if not self._apikey:
|
||
logger.warn(f"{self._name} 未配置 API Key (Passkey)")
|
||
return True, []
|
||
|
||
params = self.__get_params(keyword, mtype, page)
|
||
headers = {
|
||
"Authorization": f"Bearer {self._apikey}",
|
||
"Accept": "application/json"
|
||
}
|
||
|
||
res = await AsyncRequestUtils(
|
||
headers=headers,
|
||
proxies=self._proxy,
|
||
timeout=self._timeout
|
||
).get_res(url=self._searchurl, params=params)
|
||
|
||
if res and res.status_code == 200:
|
||
try:
|
||
data = res.json()
|
||
if data.get('code') == 0:
|
||
results = data.get('data', {}).get('torrents', [])
|
||
return False, self.__parse_result(results)
|
||
else:
|
||
logger.warn(f"{self._name} 搜索失败，错误信息：{data.get('message')}")
|
||
return True, []
|
||
except Exception as e:
|
||
logger.warn(f"{self._name} 解析响应失败：{e}")
|
||
return True, []
|
||
elif res is not None:
|
||
logger.warn(f"{self._name} 搜索失败，HTTP 错误码：{res.status_code}")
|
||
return True, []
|
||
else:
|
||
logger.warn(f"{self._name} 搜索失败，无法连接 {self._domain}")
|
||
return True, []
|
||
|
||
def __get_download_url(self, torrent_id: int) -> str:
|
||
"""
|
||
构建种子下载链接
|
||
|
||
使用 base64 编码的方式告诉 MoviePilot 如何获取真实下载地址
|
||
MoviePilot 会先请求详情接口，然后从响应中提取 data.download_url
|
||
|
||
:param torrent_id: 种子 ID
|
||
:return: base64 编码的请求配置字符串 + 详情接口 URL
|
||
"""
|
||
url = self._downloadurl % torrent_id
|
||
# MoviePilot 会解析这个特殊格式的 URL：
|
||
# 1. 使用指定的 method 和 header 请求 URL
|
||
# 2. 从 JSON 响应中提取 result 指定的字段值作为真实下载地址
|
||
params = {
|
||
'method': 'get',
|
||
'header': {
|
||
'Authorization': f'Bearer {self._apikey}',
|
||
'Accept': 'application/json'
|
||
},
|
||
'result': 'data.download_url'
|
||
}
|
||
base64_str = base64.b64encode(json.dumps(params).encode('utf-8')).decode('utf-8')
|
||
return f"[{base64_str}]{url}"
|