mirror of
https://github.com/thsrite/MoviePilot-Plugins.git
synced 2026-05-21 07:26:44 +00:00
349 lines
15 KiB
Python
349 lines
15 KiB
Python
import json
|
||
import re
|
||
|
||
from app.db.downloadhistory_oper import DownloadHistoryOper
|
||
from app.db.subscribe_oper import SubscribeOper
|
||
from app.plugins import _PluginBase
|
||
from typing import Any, List, Dict, Tuple
|
||
from app.log import logger
|
||
from app.core.event import eventmanager, Event
|
||
from app.schemas.types import EventType
|
||
|
||
|
||
class SubscribeGroup(_PluginBase):
|
||
# 插件名称
|
||
plugin_name = "订阅规则自动填充"
|
||
# 插件描述
|
||
plugin_desc = "电视剧订阅或搜索下载后自动添加官组和站点等信息到订阅，以保证后续订阅资源的统一性。"
|
||
# 插件图标
|
||
plugin_icon = "teamwork.png"
|
||
# 插件版本
|
||
plugin_version = "1.3"
|
||
# 插件作者
|
||
plugin_author = "thsrite"
|
||
# 作者主页
|
||
author_url = "https://github.com/thsrite"
|
||
# 插件配置项ID前缀
|
||
plugin_config_prefix = "subscribegroup_"
|
||
# 加载顺序
|
||
plugin_order = 26
|
||
# 可使用的用户级别
|
||
auth_level = 2
|
||
|
||
# 私有属性
|
||
_enabled: bool = False
|
||
_clear = False
|
||
_update_details = []
|
||
_subscribeoper = None
|
||
_downloadhistoryoper = None
|
||
|
||
def init_plugin(self, config: dict = None):
|
||
self._downloadhistoryoper = DownloadHistoryOper()
|
||
self._subscribeoper = SubscribeOper()
|
||
|
||
if config:
|
||
self._enabled = config.get("enabled")
|
||
self._clear = config.get("clear")
|
||
self._update_details = config.get("update_details")
|
||
|
||
# 清理已处理历史
|
||
if self._clear:
|
||
self.del_data(key="history")
|
||
|
||
self._clear = False
|
||
self.__update_config()
|
||
logger.info("已处理历史清理完成")
|
||
|
||
def __update_config(self):
|
||
self.update_config({
|
||
"enabled": self._enabled,
|
||
"clear": self._clear,
|
||
"update_details": self._update_details,
|
||
})
|
||
|
||
@eventmanager.register(EventType.DownloadAdded)
|
||
def subscribe_update(self, event: Event = None):
|
||
"""
|
||
填充订阅
|
||
"""
|
||
if not self._enabled:
|
||
logger.error("插件未开启")
|
||
return
|
||
|
||
if not self._update_details:
|
||
logger.error("插件未开启更新填充内容")
|
||
return
|
||
|
||
history: List[str] = self.get_data('history') or []
|
||
|
||
if event:
|
||
event_data = event.event_data
|
||
if not event_data or not event_data.get("hash") or not event_data.get("context"):
|
||
return
|
||
download_hash = event_data.get("hash")
|
||
# 根据hash查询下载记录
|
||
download_history = self._downloadhistoryoper.get_by_hash(download_hash)
|
||
if not download_history:
|
||
logger.warning(f"种子hash:{download_hash} 对应下载记录不存在")
|
||
return
|
||
|
||
if f"{download_history.type}:{download_history.tmdbid}" in history:
|
||
logger.warning(f"下载历史:{download_history.title} 已处理过，不再重复处理")
|
||
return
|
||
|
||
# 保存已处理历史
|
||
history.append(f"{download_history.type}:{download_history.tmdbid}")
|
||
self.save_data('history', history)
|
||
|
||
if download_history.type != '电视剧':
|
||
logger.warning(f"下载历史:{download_history.title} 不是电视剧，不进行官组填充")
|
||
return
|
||
|
||
# 根据下载历史查询订阅记录
|
||
subscribes = self._subscribeoper.list_by_tmdbid(tmdbid=download_history.tmdbid,
|
||
season=int(download_history.seasons.replace('S', ''))
|
||
if download_history.seasons and
|
||
download_history.seasons.count('-') == 0 else None)
|
||
if not subscribes or len(subscribes) == 0:
|
||
logger.warning(f"下载历史:{download_history.title} tmdbid:{download_history.tmdbid} 对应订阅记录不存在")
|
||
return
|
||
for subscribe in subscribes:
|
||
if subscribe.type != '电视剧':
|
||
logger.warning(f"订阅记录:{subscribe.name} 不是电视剧，不进行官组填充")
|
||
return
|
||
# 开始填充官组和站点
|
||
context = event_data.get("context")
|
||
_torrent = context.torrent_info
|
||
_meta = context.meta_info
|
||
|
||
# 分辨率
|
||
resource_pix = None
|
||
if "resource_pix" in self._update_details and not subscribe.resolution:
|
||
resource_pix = _meta.resource_pix if _meta else None
|
||
if resource_pix:
|
||
# 识别1080或者4k或720
|
||
if re.match(r"1080[pi]|x1080", resource_pix):
|
||
resource_pix = "1080[pi]|x1080"
|
||
if re.match(r"4K|2160p|x2160", resource_pix):
|
||
resource_pix = "4K|2160p|x2160"
|
||
if re.match(r"720[pi]|x720", resource_pix):
|
||
resource_pix = "720[pi]|x720"
|
||
|
||
# 资源类型
|
||
resource_type = None
|
||
if "resource_type" in self._update_details and not subscribe.quality:
|
||
resource_type = _meta.resource_type if _meta else None
|
||
if resource_type:
|
||
if re.match(r"Blu-?Ray.+VC-?1|Blu-?Ray.+AVC|UHD.+blu-?ray.+HEVC|MiniBD", resource_type):
|
||
resource_type = "Blu-?Ray.+VC-?1|Blu-?Ray.+AVC|UHD.+blu-?ray.+HEVC|MiniBD"
|
||
if re.match(r"Remux", resource_type):
|
||
resource_type = "Remux"
|
||
if re.match(r"Blu-?Ray", resource_type):
|
||
resource_type = "Blu-?Ray"
|
||
if re.match(r"UHD|UltraHD", resource_type):
|
||
resource_type = "UHD|UltraHD"
|
||
if re.match(r"WEB-?DL|WEB-?RIP", resource_type):
|
||
resource_type = "WEB-?DL|WEB-?RIP"
|
||
if re.match(r"HDTV", resource_type):
|
||
resource_type = "HDTV"
|
||
if re.match(r"[Hx].?265|HEVC", resource_type):
|
||
resource_type = "[Hx].?265|HEVC"
|
||
if re.match(r"[Hx].?264|AVC", resource_type):
|
||
resource_type = "[Hx].?264|AVC"
|
||
|
||
# 特效
|
||
resource_effect = None
|
||
if "resource_effect" in self._update_details and not subscribe.effect:
|
||
resource_effect = _meta.resource_effect if _meta else None
|
||
if resource_effect:
|
||
if re.match(r"Dolby[\\s.]+Vision|DOVI|[\\s.]+DV[\\s.]+", resource_effect):
|
||
resource_effect = "Dolby[\\s.]+Vision|DOVI|[\\s.]+DV[\\s.]+"
|
||
if re.match(r"Dolby[\\s.]*\\+?Atmos|Atmos", resource_effect):
|
||
resource_effect = "Dolby[\\s.]*\\+?Atmos|Atmos"
|
||
if re.match(r"[\\s.]+HDR[\\s.]+|HDR10|HDR10\\+", resource_effect):
|
||
resource_effect = "[\\s.]+HDR[\\s.]+|HDR10|HDR10\\+"
|
||
if re.match(r"[\\s.]+SDR[\\s.]+", resource_effect):
|
||
resource_effect = "[\\s.]+SDR[\\s.]+"
|
||
|
||
resource_team = None
|
||
sites = None
|
||
if ("group" in self._update_details and not subscribe.include
|
||
and (not subscribe.sites or len(json.loads(subscribe.sites)) == 0)):
|
||
# 官组
|
||
resource_team = _meta.resource_team if _meta else None
|
||
# 站点
|
||
sites = json.dumps([_torrent.site]) if _torrent and _torrent.site else None
|
||
|
||
# 更新订阅记录
|
||
if resource_pix or resource_type or resource_effect or resource_team or sites:
|
||
self._subscribeoper.update(subscribe.id, {
|
||
'include': resource_team,
|
||
'sites': sites,
|
||
'quality': resource_type,
|
||
'resolution': resource_pix,
|
||
'effect': resource_effect,
|
||
})
|
||
logger.info(f"订阅记录:{subscribe.name} 填充成功\n"
|
||
f"官组 {resource_team} \n"
|
||
f"站点 {sites} \n"
|
||
f"分辨率 {resource_pix} \n"
|
||
f"质量 {resource_type} \n"
|
||
f"特效 {resource_effect}")
|
||
else:
|
||
logger.warning(f"订阅记录:{subscribe.name} 已配置相关参数，无需自动填充")
|
||
|
||
def get_state(self) -> bool:
|
||
return self._enabled
|
||
|
||
@staticmethod
|
||
def get_command() -> List[Dict[str, Any]]:
|
||
pass
|
||
|
||
def get_api(self) -> List[Dict[str, Any]]:
|
||
pass
|
||
|
||
def get_form(self) -> Tuple[List[dict], Dict[str, Any]]:
|
||
"""
|
||
拼装插件配置页面，需要返回两块数据：1、页面配置；2、数据结构
|
||
"""
|
||
return [
|
||
{
|
||
'component': 'VForm',
|
||
'content': [
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'enabled',
|
||
'label': '启用插件',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'clear',
|
||
'label': '清理已处理记录',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSelect',
|
||
'props': {
|
||
'multiple': True,
|
||
'chips': True,
|
||
'model': 'update_details',
|
||
'label': '填充内容',
|
||
'items': [
|
||
{
|
||
"title": "资源质量",
|
||
"vale": "resource_type"
|
||
},
|
||
{
|
||
"title": "分辨率",
|
||
"vale": "resource_pix"
|
||
},
|
||
{
|
||
"title": "特效",
|
||
"vale": "resource_effect"
|
||
},
|
||
{
|
||
"title": "制作组",
|
||
"vale": "group"
|
||
}
|
||
]
|
||
}
|
||
}
|
||
]
|
||
},
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '电视剧订阅未配置包含关键词和订阅站点等配置时，订阅或搜索下载后，将下载种子的制作组和站点等信息填充到订阅信息中，以保证后续订阅资源的统一性。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '适用于订阅新出的电视剧。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
]
|
||
}
|
||
], {
|
||
"enabled": False,
|
||
"clear": False,
|
||
"update_details": []
|
||
}
|
||
|
||
def get_page(self) -> List[dict]:
|
||
pass
|
||
|
||
def stop_service(self):
|
||
"""
|
||
退出插件
|
||
"""
|
||
pass
|