mirror of
https://github.com/jxxghp/MoviePilot-Plugins.git
synced 2026-03-27 10:05:57 +00:00
plugins.v2: 添加多级分类插件
目前MP默认只支持二级分类，但是部分用户有多级目录的需求，比如增加按照年代 或者评分度分类。因此增加一个支持多级分类的插件, 目前仅支持电影多级分类。 Signed-off-by: Hangbin Liu <liuhangbin@gmail.com>
This commit is contained in:
Hangbin Liu
@@ -351,6 +351,18 @@
|
||||
"v2.0": "适配新的目录结构变化，短剧分类名称调整为配置目录路径，升级后需要重新调整设置后才能使用。"
|
||||
}
|
||||
},
|
||||
"MultiClass": {
|
||||
"name": "视频多级分类",
|
||||
"description": "支持视频多级分类",
|
||||
"labels": "文件整理",
|
||||
"version": "0.1",
|
||||
"icon": "Calibreweb_B.png",
|
||||
"author": "liuhangbin",
|
||||
"level": 1,
|
||||
"history": {
|
||||
"v0.1": "视频多级分类插件, 目前仅支持电影按评分，年代，系列分类。"
|
||||
}
|
||||
},
|
||||
"MoviePilotUpdateNotify": {
|
||||
"name": "MoviePilot更新推送",
|
||||
"description": "MoviePilot推送release更新通知、自动重启。",
|
||||
|
||||
326
plugins.v2/multiclass/__init__.py
Normal file
326
plugins.v2/multiclass/__init__.py
Normal file
@@ -0,0 +1,326 @@
|
||||
from pathlib import Path
|
||||
from typing import Any, List, Dict, Tuple
|
||||
|
||||
from app.core.context import MediaInfo
|
||||
from app.core.event import eventmanager, Event
|
||||
from app.log import logger
|
||||
from app.plugins import _PluginBase
|
||||
from app.schemas.types import ChainEventType, MediaType, NotificationType
|
||||
|
||||
class MultiClass(_PluginBase):
|
||||
# 插件名称
|
||||
plugin_name = "视频多级分类"
|
||||
# 插件描述
|
||||
plugin_desc = "支持电影按照评分，年代和系列分类"
|
||||
# 插件图标
|
||||
plugin_icon = "Calibreweb_B.png"
|
||||
# 插件版本
|
||||
plugin_version = "0.1"
|
||||
# 插件作者
|
||||
plugin_author = "liuhangbin"
|
||||
# 作者主页
|
||||
author_url = "https://github.com/liuhangbin"
|
||||
# 插件配置项ID前缀
|
||||
plugin_config_prefix = "multiclass_"
|
||||
# 加载顺序
|
||||
plugin_order = 1
|
||||
# 可使用的用户级别
|
||||
auth_level = 1
|
||||
|
||||
_enabled = False
|
||||
_notify = False
|
||||
_year_class = False
|
||||
_vote_class = False
|
||||
_collection_class = False
|
||||
|
||||
def init_plugin(self, config: dict = None):
|
||||
|
||||
if config:
|
||||
self._enabled = config.get("enabled", False)
|
||||
self._notify = config.get("notify", False)
|
||||
self._year_class = config.get("year_class", False)
|
||||
self._vote_class = config.get("vote_class", False)
|
||||
self._collection_class = config.get("collection_class", False)
|
||||
|
||||
def get_state(self) -> bool:
|
||||
return self._enabled
|
||||
|
||||
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': 6
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VSwitch',
|
||||
'props': {
|
||||
'model': 'enabled',
|
||||
'label': '启用插件',
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
'md': 6
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VSwitch',
|
||||
'props': {
|
||||
'model': 'year_class',
|
||||
'label': '按照年代分类',
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
'md': 6
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VSwitch',
|
||||
'props': {
|
||||
'model': 'vote_class',
|
||||
'label': '按照评分分类',
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
'md': 6
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VSwitch',
|
||||
'props': {
|
||||
'model': 'collection_class',
|
||||
'label': '按照系列分类',
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
'md': 6
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VSwitch',
|
||||
'props': {
|
||||
'model': 'notify',
|
||||
'label': '发送消息',
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VRow',
|
||||
'content': [
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VAlert',
|
||||
'props': {
|
||||
'type': 'info',
|
||||
'variant': 'tonal',
|
||||
'text': '插件目前仅支持电影(需要开启智能重命名)。如果按评分分类，7-9 高分，4-6 一般，1-3 垃圾。 系列电影不参与评分, 不按年代分类。'
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
], {
|
||||
"enabled": False,
|
||||
"notify": False,
|
||||
"year_class": False,
|
||||
"vote_class": False,
|
||||
"collection_class": False
|
||||
}
|
||||
|
||||
def get_page(self) -> List[dict]:
|
||||
pass
|
||||
|
||||
@eventmanager.register(ChainEventType.TransferRename)
|
||||
def category_handler(self, event: Event):
|
||||
"""
|
||||
根据多级分类规则重新分类组装地址
|
||||
"""
|
||||
logger.debug(f"多级分类插件触发！")
|
||||
|
||||
# 基础验证
|
||||
if not self.get_state():
|
||||
logger.debug(f"多级分类插件未启用！")
|
||||
return
|
||||
if not event:
|
||||
logger.warning(f"多级分类异常：事件对象为空")
|
||||
return
|
||||
if not hasattr(event, 'event_data'):
|
||||
logger.warning(f"多级分类异常：事件数据为空")
|
||||
return
|
||||
|
||||
try:
|
||||
data = event.event_data
|
||||
|
||||
# 验证必要的数据字段
|
||||
if not hasattr(data, 'render_str') or not data.render_str:
|
||||
logger.warning(f"多级分类异常：render_str为空")
|
||||
return
|
||||
else:
|
||||
render_str = data.render_str
|
||||
|
||||
# 暂时只支持电影分类
|
||||
if not hasattr(data, 'rename_dict') or not data.rename_dict:
|
||||
logger.warning(f"多级分类异常：rename_dict为空")
|
||||
return
|
||||
else:
|
||||
rename_dict = data.rename_dict
|
||||
video_type = rename_dict.get("type", "")
|
||||
if video_type != "电影":
|
||||
logger.debug(f"多级分类异常：不支持的媒体类型: {video_type}, 只支持电影分类")
|
||||
return
|
||||
|
||||
# 安全获取数据字段
|
||||
title = rename_dict.get("title", "")
|
||||
en_title = rename_dict.get("en_title", "")
|
||||
year = rename_dict.get("year")
|
||||
vote_average = rename_dict.get("vote_average")
|
||||
media_info = rename_dict.get("__mediainfo__")
|
||||
|
||||
# 初始化默认值
|
||||
vote_count = 0
|
||||
c_name = None
|
||||
vote_path = "未知评分"
|
||||
decade = 0
|
||||
|
||||
# 安全处理媒体信息
|
||||
if media_info and hasattr(media_info, 'vote_count'):
|
||||
try:
|
||||
vote_count = int(media_info.vote_count) if media_info.vote_count else 0
|
||||
except (ValueError, TypeError):
|
||||
vote_count = 0
|
||||
|
||||
if hasattr(media_info, 'tmdb_info') and media_info.tmdb_info:
|
||||
collection = media_info.tmdb_info.get("belongs_to_collection")
|
||||
if collection and isinstance(collection, dict):
|
||||
c_name = collection.get("name")
|
||||
|
||||
# 安全处理评分数据
|
||||
try:
|
||||
if vote_average is not None:
|
||||
vote_average = float(vote_average)
|
||||
else:
|
||||
vote_average = 0
|
||||
except (ValueError, TypeError):
|
||||
vote_average = 0
|
||||
|
||||
# 评分分类逻辑
|
||||
if vote_count < 10:
|
||||
vote_average = 0
|
||||
vote_path = "评分不足"
|
||||
elif vote_average >= 7:
|
||||
vote_path = "高分电影"
|
||||
elif vote_average >= 4:
|
||||
vote_path = "一般电影"
|
||||
else:
|
||||
vote_path = "垃圾电影"
|
||||
|
||||
# 安全处理年份数据
|
||||
try:
|
||||
if year and str(year).isdigit():
|
||||
year_int = int(year)
|
||||
if 1900 <= year_int <= 2100: # 合理的年份范围
|
||||
decade = (year_int // 10) * 10
|
||||
else:
|
||||
decade = 0
|
||||
logger.warning(f"年份超出合理范围: {year}")
|
||||
else:
|
||||
decade = 0
|
||||
except (ValueError, TypeError):
|
||||
decade = 0
|
||||
logger.warning(f"年份转换失败: {year}")
|
||||
|
||||
|
||||
# 构建分类路径
|
||||
path_parts = []
|
||||
|
||||
if self._collection_class and c_name:
|
||||
# 当collection为true时，只添加collection name
|
||||
# 清理collection名称，移除特殊字符
|
||||
clean_c_name = str(c_name).strip()
|
||||
if clean_c_name:
|
||||
path_parts.append("系列电影")
|
||||
path_parts.append(clean_c_name)
|
||||
else:
|
||||
# 当collection不为true时，根据其他配置添加路径
|
||||
if self._vote_class and vote_path:
|
||||
path_parts.append(vote_path)
|
||||
if self._year_class and decade > 0:
|
||||
path_parts.append(f"{decade}s")
|
||||
|
||||
# 构建最终的路径
|
||||
if path_parts:
|
||||
# 确保render_str不为空
|
||||
safe_render_str = str(render_str).strip() if render_str else ""
|
||||
event.event_data.updated_str = f"{'/'.join(path_parts)}/{safe_render_str}"
|
||||
# 更新事件数据
|
||||
event.event_data.updated = True
|
||||
event.event_data.source = "MultiClass"
|
||||
|
||||
# 发送消息
|
||||
if self._notify:
|
||||
self.post_message(
|
||||
mtype=NotificationType.Organize,
|
||||
title="多级分类完成",
|
||||
text=f"已重新分类: {event.event_data.updated_str}",
|
||||
)
|
||||
else:
|
||||
event.event_data.updated = False
|
||||
logger.warning(f"多级分类失败: 未找到分类路径，请检查配置是否已开启")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"多级分类异常: {str(e)}", exc_info=True)
|
||||
# 确保即使出错也不会影响原始数据
|
||||
if hasattr(event, 'event_data') and event.event_data:
|
||||
event.event_data.updated = False
|
||||
event.event_data.updated_str = getattr(data, 'render_str', '') if data else ''
|
||||
|
||||
def stop_service(self):
|
||||
"""
|
||||
停止服务
|
||||
"""
|
||||
pass
|
||||
Reference in New Issue
Block a user