diff --git a/package.v2.json b/package.v2.json index 5b2d115..68190d7 100644 --- a/package.v2.json +++ b/package.v2.json @@ -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更新通知、自动重启。", diff --git a/plugins.v2/multiclass/__init__.py b/plugins.v2/multiclass/__init__.py new file mode 100644 index 0000000..e81bf00 --- /dev/null +++ b/plugins.v2/multiclass/__init__.py @@ -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