mirror of
https://github.com/jxxghp/MoviePilot-Plugins.git
synced 2026-03-27 10:05:57 +00:00
6357dc8e4a
目前MP默认只支持二级分类，但是部分用户有多级目录的需求，比如增加按照年代 或者评分度分类。因此增加一个支持多级分类的插件, 目前仅支持电影多级分类。 Signed-off-by: Hangbin Liu <liuhangbin@gmail.com>
327 lines
12 KiB
Python
327 lines
12 KiB
Python
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
|