mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
264 lines
8.6 KiB
Python
264 lines
8.6 KiB
Python
from typing import Tuple, List, Optional
|
||
|
||
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.string import StringUtils
|
||
|
||
|
||
class HddolbySpider:
|
||
"""
|
||
HDDolby API
|
||
"""
|
||
_indexerid = None
|
||
_domain = None
|
||
_domain_host = None
|
||
_name = ""
|
||
_proxy = None
|
||
_cookie = None
|
||
_ua = None
|
||
_apikey = None
|
||
_size = 40
|
||
_pageurl = None
|
||
_timeout = 15
|
||
_searchurl = None
|
||
|
||
# 分类
|
||
_movie_category = [401, 405]
|
||
_tv_category = [402, 403, 404, 405]
|
||
|
||
# 标签
|
||
_labels = {
|
||
"gf": "官方",
|
||
"gy": "国语",
|
||
"yy": "粤语",
|
||
"ja": "日语",
|
||
"ko": "韩语",
|
||
"zz": "中文字幕",
|
||
"jz": "禁转",
|
||
"xz": "限转",
|
||
"diy": "DIY",
|
||
"sf": "首发",
|
||
"yq": "应求",
|
||
"m0": "零魔",
|
||
"yc": "原创",
|
||
"gz": "官字",
|
||
"db": "Dolby Vision",
|
||
"hdr10": "HDR10",
|
||
"hdrm": "HDR10+",
|
||
"tx": "特效",
|
||
"lz": "连载",
|
||
"wj": "完结",
|
||
"hdrv": "HDR Vivid",
|
||
"hlg": "HLG",
|
||
"hq": "高码率",
|
||
"hfr": "高帧率",
|
||
}
|
||
|
||
def __init__(self, indexer: dict):
|
||
self.systemconfig = SystemConfigOper()
|
||
if indexer:
|
||
self._indexerid = indexer.get('id')
|
||
self._domain = indexer.get('domain')
|
||
self._domain_host = StringUtils.get_url_domain(self._domain)
|
||
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
|
||
self._searchurl = f"https://api.{self._domain_host}/api/v1/torrent/search"
|
||
self._pageurl = f"{self._domain}details.php?id=%s&hit=1"
|
||
|
||
def __get_params(self, keyword: str, mtype: MediaType = None, page: Optional[int] = 0) -> dict:
|
||
"""
|
||
获取请求参数
|
||
"""
|
||
if mtype == MediaType.TV:
|
||
categories = self._tv_category
|
||
elif mtype == MediaType.MOVIE:
|
||
categories = self._movie_category
|
||
else:
|
||
categories = list(set(self._movie_category + self._tv_category))
|
||
|
||
# 输入参数
|
||
return {
|
||
"keyword": keyword,
|
||
"page_number": page,
|
||
"page_size": 100,
|
||
"categories": categories,
|
||
"visible": 1,
|
||
}
|
||
|
||
def __parse_result(self, results: List[dict]) -> List[dict]:
|
||
"""
|
||
解析搜索结果
|
||
"""
|
||
torrents = []
|
||
if not results:
|
||
return []
|
||
|
||
for result in results:
|
||
"""
|
||
{
|
||
"id": 120202,
|
||
"promotion_time_type": 0,
|
||
"promotion_until": "0000-00-00 00:00:00",
|
||
"category": 402,
|
||
"medium": 6,
|
||
"codec": 1,
|
||
"standard": 2,
|
||
"team": 10,
|
||
"audiocodec": 14,
|
||
"leechers": 0,
|
||
"seeders": 1,
|
||
"name": "[DBY] Lost S06 2010 Complete 1080p Netflix WEB-DL AVC DDP5.1-DBTV",
|
||
"small_descr": "lost ",
|
||
"times_completed": 0,
|
||
"size": 33665425886,
|
||
"added": "2025-02-18 19:47:56",
|
||
"url": 0,
|
||
"hr": 0,
|
||
"tmdb_type": "tv",
|
||
"tmdb_id": 4607,
|
||
"imdb_id": null,
|
||
"tags": "gf"
|
||
}
|
||
"""
|
||
# 类别
|
||
category_value = result.get('category')
|
||
if category_value in self._tv_category:
|
||
category = MediaType.TV.value
|
||
elif category_value in self._movie_category:
|
||
category = MediaType.MOVIE.value
|
||
else:
|
||
category = MediaType.UNKNOWN.value
|
||
# 标签
|
||
torrentLabelIds = result.get('tags', "").split(";") or []
|
||
torrentLabels = []
|
||
for labelId in torrentLabelIds:
|
||
if self._labels.get(labelId) is not None:
|
||
torrentLabels.append(self._labels.get(labelId))
|
||
# 种子信息
|
||
torrent = {
|
||
'title': result.get('name'),
|
||
'description': result.get('small_descr'),
|
||
'enclosure': self.__get_download_url(result.get('id'), result.get('downhash')),
|
||
'pubdate': result.get('added'),
|
||
'size': result.get('size'),
|
||
'seeders': result.get('seeders'),
|
||
'peers': result.get('leechers'),
|
||
'grabs': result.get('times_completed'),
|
||
'downloadvolumefactor': self.__get_downloadvolumefactor(result.get('promotion_time_type')),
|
||
'uploadvolumefactor': self.__get_uploadvolumefactor(result.get('promotion_time_type')),
|
||
'freedate': result.get('promotion_until'),
|
||
'page_url': self._pageurl % result.get('id'),
|
||
'labels': torrentLabels,
|
||
'category': category
|
||
}
|
||
torrents.append(torrent)
|
||
return torrents
|
||
|
||
def search(self, keyword: str, mtype: MediaType = None, page: Optional[int] = 0) -> Tuple[bool, List[dict]]:
|
||
"""
|
||
搜索
|
||
"""
|
||
|
||
# 准备参数
|
||
params = self.__get_params(keyword, mtype, page)
|
||
|
||
# 发送请求
|
||
res = RequestUtils(
|
||
headers={
|
||
"Content-Type": "application/json",
|
||
"Accept": "application/json, text/plain, */*",
|
||
"x-api-key": self._apikey
|
||
},
|
||
cookies=self._cookie,
|
||
proxies=self._proxy,
|
||
referer=f"{self._domain}",
|
||
timeout=self._timeout
|
||
).post_res(url=self._searchurl, json=params)
|
||
if res and res.status_code == 200:
|
||
result = res.json()
|
||
if result.get("error"):
|
||
logger.warn(f"{self._name} 搜索失败，错误信息：{result.get('error').get('message')}")
|
||
return True, []
|
||
return False, self.__parse_result(result.get('data'))
|
||
elif res is not None:
|
||
logger.warn(f"{self._name} 搜索失败，错误码：{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]]:
|
||
"""
|
||
异步搜索
|
||
"""
|
||
# 准备参数
|
||
params = self.__get_params(keyword, mtype, page)
|
||
|
||
# 发送请求
|
||
res = await AsyncRequestUtils(
|
||
headers={
|
||
"Content-Type": "application/json",
|
||
"Accept": "application/json, text/plain, */*",
|
||
"x-api-key": self._apikey
|
||
},
|
||
cookies=self._cookie,
|
||
proxies=self._proxy,
|
||
referer=f"{self._domain}",
|
||
timeout=self._timeout
|
||
).post_res(url=self._searchurl, json=params)
|
||
if res and res.status_code == 200:
|
||
result = res.json()
|
||
if result.get("error"):
|
||
logger.warn(f"{self._name} 搜索失败，错误信息：{result.get('error').get('message')}")
|
||
return True, []
|
||
return False, self.__parse_result(result.get('data'))
|
||
elif res is not None:
|
||
logger.warn(f"{self._name} 搜索失败，错误码：{res.status_code}")
|
||
return True, []
|
||
else:
|
||
logger.warn(f"{self._name} 搜索失败，无法连接 {self._domain}")
|
||
return True, []
|
||
|
||
@staticmethod
|
||
def __get_downloadvolumefactor(discount: int) -> float:
|
||
"""
|
||
获取下载系数
|
||
"""
|
||
discount_dict = {
|
||
2: 0,
|
||
5: 0.5,
|
||
6: 1,
|
||
7: 0.3
|
||
}
|
||
if discount:
|
||
return discount_dict.get(discount, 1)
|
||
return 1
|
||
|
||
@staticmethod
|
||
def __get_uploadvolumefactor(discount: int) -> float:
|
||
"""
|
||
获取上传系数
|
||
"""
|
||
discount_dict = {
|
||
3: 2,
|
||
4: 2,
|
||
6: 2
|
||
}
|
||
if discount:
|
||
return discount_dict.get(discount, 1)
|
||
return 1
|
||
|
||
def __get_download_url(self, torrent_id: int, downhash: str) -> str:
|
||
"""
|
||
获取下载链接，返回base64编码的json字符串及URL
|
||
"""
|
||
return f"{self._domain}download.php?id={torrent_id}&downhash={downhash}"
|