This commit is contained in:
thsrite
2024-10-25 12:58:20 +08:00
parent 924622a975
commit 4afe0bf8d0
4 changed files with 782 additions and 2 deletions

View File

@@ -43,7 +43,7 @@ MoviePilot三方插件市场https://github.com/thsrite/MoviePilot-Plugins/
37. docker自定义任务 v1.3 `管理宿主机docker自定义容器定时任务。`
38. 插件彻底卸载 v1.2 `删除数据库中已安装插件记录、清理插件文件。`
39. 实时软连接 v2.0.1 `监控目录文件变化,媒体文件软连接,其他文件可选复制。`
40. 订阅规则自动填充 v2.7 `电视剧下载后自动添加官组等信息到订阅;添加订阅后根据二级分类名称自定义订阅规则。`
40. 订阅规则自动填充 v2.8 `电视剧下载后自动添加官组等信息到订阅;添加订阅后根据二级分类名称自定义订阅规则。`
41. Emby元数据刷新 v2.0.1 `定时刷新Emby媒体库元数据演职人员中文。`
42. Emby媒体标签 v1.2 `自动给媒体库媒体添加标签。`
43. 热门媒体订阅 v1.7 `自定添加热门媒体到订阅。`

View File

@@ -450,7 +450,6 @@
"icon": "teamwork.png",
"author": "thsrite",
"level": 2,
"v2": true,
"history": {
"v2.7": "下载填充判断当前站点是否在已选订阅站点范围内",
"v2.6": "兼容属性值包含:号",

View File

@@ -332,5 +332,31 @@
"v1.1": "bug修复",
"v1.0": "监控未上线影视作品,自动添加订阅。"
}
},
"SubscribeGroup": {
"name": "订阅规则自动填充",
"description": "电视剧下载后自动添加官组等信息到订阅;添加订阅后根据二级分类名称自定义订阅规则。",
"labels": "订阅",
"version": "2.8",
"icon": "teamwork.png",
"author": "thsrite",
"level": 2,
"history": {
"v2.8": "兼容v2",
"v2.7": "下载填充判断当前站点是否在已选订阅站点范围内",
"v2.6": "兼容属性值包含:号",
"v2.5": "操作历史Unicode编码转中文",
"v2.4": "保存路径支持变量{name} (订阅名称 (年份)",
"v2.3": "二级分类自定义填充支持保存路径",
"v2.1": "站点与官组分开,修复质量无填充",
"v2.0": "种子下载自定义填充支持自定义占位符",
"v1.8": "修复种子下载不填充bug",
"v1.7": "操作历史Unicode编码转中文",
"v1.6": "支持一行配置多个二级分类名称",
"v1.5": "支持操作历史",
"v1.4": "支持根据二级分类名称自定义订阅规则",
"v1.3": "增加质量、分辨率、特效信息填充",
"v1.2": "修复订阅已存在包含关键词和订阅站点"
}
}
}

View File

@@ -0,0 +1,755 @@
import json
import re
import time
from app.db.downloadhistory_oper import DownloadHistoryOper
from app.db.subscribe_oper import SubscribeOper
from app.db.site_oper import SiteOper
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, SystemConfigKey
class SubscribeGroup(_PluginBase):
# 插件名称
plugin_name = "订阅规则自动填充"
# 插件描述
plugin_desc = "电视剧下载后自动添加官组等信息到订阅;添加订阅后根据二级分类名称自定义订阅规则。"
# 插件图标
plugin_icon = "teamwork.png"
# 插件版本
plugin_version = "2.8"
# 插件作者
plugin_author = "thsrite"
# 作者主页
author_url = "https://github.com/thsrite"
# 插件配置项ID前缀
plugin_config_prefix = "subscribegroup_"
# 加载顺序
plugin_order = 26
# 可使用的用户级别
auth_level = 2
# 私有属性
_enabled: bool = False
_category: bool = False
_clear = False
_clear_handle = False
_update_details = []
_update_confs = None
_subscribe_confs = {}
_subscribeoper = None
_downloadhistoryoper = None
_siteoper = None
def init_plugin(self, config: dict = None):
self._downloadhistoryoper = DownloadHistoryOper()
self._subscribeoper = SubscribeOper()
self._siteoper = SiteOper()
if config:
self._enabled = config.get("enabled")
self._category = config.get("category")
self._clear = config.get("clear")
self._clear_handle = config.get("clear_handle")
self._update_details = config.get("update_details") or []
self._update_confs = config.get("update_confs")
if self._update_confs:
active_sites = self._siteoper.list_active()
for confs in str(self._update_confs).split("\n"):
category = None
resolution = None
quality = None
effect = None
include = None
exclude = None
savepath = None
sites = []
for conf in str(confs).split("#"):
if ":" in conf:
k = conf.split(":")[0]
v = ":".join(conf.split(":")[1:])
if k == "category":
category = v
if k == "resolution":
resolution = v
if k == "quality":
quality = v
if k == "effect":
effect = v
if k == "include":
include = v
if k == "exclude":
exclude = v
if k == "savepath":
savepath = v
if k == "sites":
for site_name in str(v).split(","):
for active_site in active_sites:
if str(site_name) == str(active_site.name):
sites.append(active_site.id)
break
if category:
for c in str(category).split(","):
self._subscribe_confs[c] = {
'resolution': resolution,
'quality': quality,
'effect': effect,
'include': include,
'exclude': exclude,
'savepath': savepath,
'sites': sites
}
logger.info(f"获取到二级分类自定义配置 {len(self._subscribe_confs.keys())}")
else:
self._subscribe_confs = {}
# 清理已处理历史
if self._clear_handle:
self.del_data(key="history_handle")
self._clear_handle = False
self.__update_config()
logger.info("已处理历史清理完成")
# 清理历史记录
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,
"category": self._category,
"clear": self._clear,
"clear_handle": self._clear_handle,
"update_details": self._update_details,
"update_confs": self._update_confs,
})
@eventmanager.register(EventType.SubscribeAdded)
def subscribe_notice(self, event: Event = None):
"""
添加订阅根据二级分类填充订阅
"""
if not event:
logger.error("订阅事件数据为空")
return
if not self._category:
logger.error("二级分类自定义填充未开启")
return
if len(self._subscribe_confs.keys()) == 0:
logger.error("插件未开启二级分类自定义填充")
return
if event:
event_data = event.event_data
if not event_data or not event_data.get("subscribe_id") or not event_data.get("mediainfo"):
logger.error(f"订阅事件数据不完整 {event_data}")
return
sid = event_data.get("subscribe_id")
category = event_data.get("mediainfo").get("category")
if not category:
logger.error(f"订阅ID:{sid} 未获取到二级分类")
return
if category not in self._subscribe_confs.keys():
logger.error(f"订阅ID:{sid} 二级分类:{category} 未配置自定义规则")
return
# 查询订阅
subscribe = self._subscribeoper.get(sid)
# 二级分类自定义配置
category_conf = self._subscribe_confs.get(category)
update_dict = {}
if category_conf.get('include'):
update_dict['include'] = category_conf.get('include')
if category_conf.get('exclude'):
update_dict['exclude'] = category_conf.get('exclude')
if category_conf.get('sites'):
update_dict['sites'] = category_conf.get('sites')
if category_conf.get('resolution'):
update_dict['resolution'] = self.__parse_pix(category_conf.get('resolution'))
if category_conf.get('quality'):
update_dict['quality'] = self.__parse_type(category_conf.get('quality'))
if category_conf.get('effect'):
update_dict['effect'] = self.__parse_effect(category_conf.get('effect'))
if category_conf.get('savepath'):
# 判断是否有变量{name}
if '{name}' in category_conf.get('savepath'):
savepath = category_conf.get('savepath').replace('{name}', f"{subscribe.name} ({subscribe.year})")
update_dict['save_path'] = savepath
else:
update_dict['save_path'] = category_conf.get('savepath')
# 更新订阅自定义配置
self._subscribeoper.update(sid, update_dict)
logger.info(f"订阅记录:{subscribe.name} 填充成功\n{update_dict}")
# 读取历史记录
history = self.get_data('history') or []
history.append({
'name': subscribe.name,
'type': f'二级分类自定义配置 {category}',
'content': json.dumps(update_dict),
"time": time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time()))
})
# 保存历史
self.save_data(key="history", value=history)
@eventmanager.register(EventType.DownloadAdded)
def download_notice(self, event: Event = None):
"""
添加下载填充订阅制作组等信息
"""
if not event:
logger.error("下载事件数据为空")
return
if not self._enabled:
# logger.error("种子下载自定义填充未开启")
return
if len(self._update_details) == 0:
# logger.error("插件未开启更新填充内容")
return
if event:
event_data = event.event_data
if not event_data or not event_data.get("hash") or not event_data.get("context"):
logger.error(f"下载事件数据不完整 {event_data}")
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
history_handle: List[str] = self.get_data('history_handle') or []
if f"{download_history.type}:{download_history.tmdbid}" in history_handle:
logger.warning(f"下载历史:{download_history.title} 已处理过,不再重复处理")
return
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
logger.info(
f"获取到tmdbid {download_history.tmdbid} season {int(download_history.seasons.replace('S', '')) if download_history.seasons and download_history.seasons.count('-') == 0 else None} 订阅记录:{len(subscribes)}")
for subscribe in subscribes:
if subscribe.type != '电视剧':
logger.warning(f"订阅记录:{subscribe.name} 不是电视剧,不进行官组填充")
continue
# 开始填充官组和站点
context = event_data.get("context")
_torrent = context.torrent_info
_meta = context.meta_info
# 填充数据
update_dict = {}
# 分辨率
if "分辨率" in self._update_details and not subscribe.resolution:
resource_pix = _meta.resource_pix if _meta else None
if resource_pix:
resource_pix = self.__parse_pix(resource_pix)
if resource_pix:
update_dict['resolution'] = resource_pix
else:
logger.warning(f"订阅记录:{subscribe.name} 未获取到分辨率信息")
# 资源质量
if "资源质量" in self._update_details and not subscribe.quality:
resource_type = _meta.resource_type if _meta else None
if resource_type:
resource_type = self.__parse_type(resource_type)
if resource_type:
update_dict['quality'] = resource_type
else:
logger.warning(f"订阅记录:{subscribe.name} 未获取到资源质量信息")
# 特效
if "特效" in self._update_details and not subscribe.effect:
resource_effect = _meta.resource_effect if _meta else None
if resource_effect:
resource_effect = self.__parse_effect(resource_effect)
if resource_effect:
update_dict['effect'] = resource_effect
else:
logger.warning(f"订阅记录:{subscribe.name} 未获取到特效信息")
# 制作组
if "制作组" in self._update_details and not subscribe.include:
# 官组
resource_team = _meta.resource_team if _meta else None
customization = _meta.customization if _meta else None
if resource_team and customization:
resource_team = f"{customization}.+{resource_team}"
if not resource_team and customization:
resource_team = customization
if resource_team:
update_dict['include'] = resource_team
# 站点
if "站点" in self._update_details and (
not subscribe.sites or (subscribe.sites and len(json.loads(subscribe.sites)) == 0)):
# 站点 判断是否在订阅站点范围内
rss_sites = self.systemconfig.get(SystemConfigKey.RssSites) or []
if _torrent and _torrent.site and int(_torrent.site) in rss_sites:
sites = json.dumps([_torrent.site])
update_dict['sites'] = sites
if len(update_dict.keys()) == 0:
logger.info(f"订阅记录:{subscribe.name} 无需填充")
continue
# 更新订阅记录
self._subscribeoper.update(subscribe.id, update_dict)
logger.info(f"订阅记录:{subscribe.name} 填充成功\n {update_dict}")
# 读取历史记录
history = self.get_data('history') or []
history.append({
'name': subscribe.name,
'type': '种子下载自定义配置',
'content': json.dumps(update_dict),
"time": time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time()))
})
# 保存历史
self.save_data(key="history", value=history)
# 保存已处理历史
history_handle.append(f"{download_history.type}:{download_history.tmdbid}")
self.save_data('history_handle', history_handle)
def __parse_pix(self, 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"
return resource_pix
def __parse_type(self, 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"
return resource_type
def __parse_effect(self, 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.]+"
return resource_effect
def get_state(self) -> bool:
return self._enabled or self._category
@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': 'category',
'label': '二级分类自定义填充',
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSwitch',
'props': {
'model': 'clear',
'label': '清理历史记录',
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSwitch',
'props': {
'model': 'clear_handle',
'label': '清理已处理记录',
}
}
]
},
]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
},
'content': [
{
'component': 'VSelect',
'props': {
'multiple': True,
'chips': True,
'model': 'update_details',
'label': '种子下载填充内容',
'items': [
{
"title": "资源质量",
"vale": "资源质量"
},
{
"title": "分辨率",
"vale": "分辨率"
},
{
"title": "特效",
"vale": "特效"
},
{
"title": "制作组",
"vale": "制作组"
},
{
"title": "站点",
"vale": "站点"
}
]
}
}
]
},
]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12
},
'content': [
{
'component': 'VTextarea',
'props': {
'model': 'update_confs',
'label': '二级分类自定义填充',
'rows': 3,
'placeholder': 'category:日番#include:.*(CR.*简繁|简繁英).RLWeb|ADWeb.#sites:观众,红叶PT\n'
'category:港台剧,日韩剧#include:国粤'
}
}
]
}
]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
},
'content': [
{
'component': 'VAlert',
'props': {
'type': 'error',
'variant': 'tonal',
'text': '种子下载自定义填充:需要下载种子才会填充订阅属性,且不会覆盖原有属性!'
}
}
]
}
]
},
{
'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': 'error',
'variant': 'tonal',
'text': '二级分类自定义填充:添加订阅才会填充订阅属性,会强制覆盖!用于根据二级分类自定义订阅规则,具体属性明细请查看电视剧订阅设置页面。'
}
}
]
}
]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
},
'content': [
{
'component': 'VAlert',
'props': {
'type': 'info',
'variant': 'tonal',
'text': 'category:二级分类名称(多个分类名称逗号拼接),resolution:分辨率,quality:质量,effect:特效,include:包含关键词,'
'exclude:排除关键词,sites:站点名称(多个站点用逗号拼接),savepath:保存路径/{name}{name}为当前订阅的名称和年份)。'
'category必填多组属性用#分割。例如category:动漫#resolution:1080p'
'添加的动漫订阅指定分辨率为1080p'
}
}
]
}
]
}
]
}
], {
"enabled": False,
"category": False,
"clear": False,
"clear_handle": False,
"update_details": [],
"update_confs": "",
}
def get_page(self) -> List[dict]:
historys = self.get_data('history')
if not historys:
return [
{
'component': 'div',
'text': '暂无数据',
'props': {
'class': 'text-center',
}
}
]
if not isinstance(historys, list):
historys = [historys]
# 按照时间倒序
historys = sorted(historys, key=lambda x: x.get("time") or 0, reverse=True)
contens = [
{
'component': 'tr',
'props': {
'class': 'text-sm'
},
'content': [
{
'component': 'td',
'props': {
'class': 'whitespace-nowrap break-keep text-high-emphasis'
},
'text': history.get("time")
},
{
'component': 'td',
'text': history.get("name")
},
{
'component': 'td',
'text': history.get("type")
},
{
'component': 'td',
'text': history.get("content").encode('utf-8').decode('unicode_escape') if history.get(
"content") else ''
}
]
} for history in historys
]
# 拼装页面
return [
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
},
'content': [
{
'component': 'VTable',
'props': {
'hover': True
},
'content': [
{
'component': 'thead',
'content': [
{
'component': 'th',
'props': {
'class': 'text-start ps-4'
},
'text': '执行时间'
},
{
'component': 'th',
'props': {
'class': 'text-start ps-4'
},
'text': '订阅名称'
},
{
'component': 'th',
'props': {
'class': 'text-start ps-4'
},
'text': '更新类型'
},
{
'component': 'th',
'props': {
'class': 'text-start ps-4'
},
'text': '更新内容'
},
]
},
{
'component': 'tbody',
'content': contens
}
]
}
]
}
]
}
]
def stop_service(self):
"""
退出插件
"""
pass