import json,re from datetime import datetime, timedelta from app.modules.emby import Emby from app.core.config import settings from app.plugins import _PluginBase from app.log import logger from typing import List, Tuple, Dict, Any, Optional import pytz from app.schemas import WebhookEventInfo from app.schemas.types import EventType from app.core.event import eventmanager, Event from apscheduler.triggers.cron import CronTrigger from apscheduler.schedulers.background import BackgroundScheduler class DiagParamAdjust(_PluginBase): # 插件名称 plugin_name = "诊断参数调整" # 插件描述 plugin_desc = "Emby专用插件|暂时性解决emby字幕偏移问题，需要emby安装Diagnostics插件。" # 插件图标 plugin_icon = "Gatus_A.png" # 插件版本 plugin_version = "1.3" # 插件作者 plugin_author = "jeblove" # 作者主页 author_url = "https://github.com/jeblove" # 插件配置项ID前缀 plugin_config_prefix = "dpa_" # 加载顺序 plugin_order = 14 # 可使用的用户级别 auth_level = 1 # 私有属性 _enabled: bool = False # 修正字幕偏移用途（播放时执行） _offset_play = True _onlyonce = False _base_url = None _endpoint = None _api_key = None _search_text = None _replace_text = None _cron = None _cron_switch = False # 请求接口 _url = "[HOST]emby/EncodingDiagnostics/DiagnosticOptions?api_key=[APIKEY]" # 定时器 _scheduler: Optional[BackgroundScheduler] = None # 目标消息 _webhook_actions = { "playback.start": "开始播放", } # 分辨率标识 _resolution = None # 分辨率改动 _last_resolution = None # 目标参数 _target_search_text = None _target_replace_text = None def init_plugin(self, config: dict = None): # 停止现有任务 self.stop_service() if config: self._enabled = config.get("enabled") self._offset_play = config.get("offset_play") self._onlyonce = config.get("onlyonce") self._search_text = config.get("search") self._replace_text = config.get("replace") self._cron = config.get("cron") self._cron_switch = config.get("cron_switch") if self._onlyonce: self._scheduler = BackgroundScheduler(timezone=settings.TZ) logger.info(f"诊断参数调整服务启动，立刻运行一次") self._scheduler.add_job(func=self.run, trigger='date', run_date=datetime.now(tz=pytz.timezone(settings.TZ)) + timedelta(seconds=3), name="诊断参数调整") # 关闭一次性开关 self._onlyonce = False self.update_config({ "enabled": self._enabled, "offset_play": self._offset_play, "onlyonce": False, "search": self._search_text, "replace": self._replace_text, "cron": self._cron, "cron_switch": self._cron_switch, }) # 启动任务 if self._scheduler.get_jobs(): self._scheduler.print_jobs() self._scheduler.start() 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_service(self) -> List[Dict[str, Any]]: """ 注册插件公共服务 [{ "id": "服务ID", "name": "服务名称", "trigger": "触发器：cron/interval/date/CronTrigger.from_crontab()", "func": self.xxx, "kwargs": {} # 定时器参数 }] """ if self._enabled and self._cron and self._cron_switch: return [{ "id": "DiagParamAdjust", "name": "诊断参数调整定时服务", "trigger": CronTrigger.from_crontab(self._cron), "func": self.run, "kwargs": {} }] 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': 4 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'enabled', 'label': '启用插件', } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 4 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'offset_play', 'label': '修正字幕偏移(播放时执行)', } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 4 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'onlyonce', 'label': '立即运行一次', } } ] } ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'search', 'label': '搜索文本' } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'replace', 'label': '替换文本' } } ] } ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'cron', 'label': '检测执行周期', 'placeholder': '*/5 * * * *' } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 3 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'cron_switch', 'label': '周期模式', } } ] } ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, }, 'content': [ { 'component': 'VAlert', 'props': { 'type': 'info', 'variant': 'tonal', 'text': '- 暂时性解决emby字幕偏移问题，如默认参数不合适请在基础上修改【替换文本】x、y至适合(4K视频情况下！)，如[x=W/4:y=h/5]。\n - 【修正字幕偏移(播放时执行)】需要emby配置webhooks消息通知：勾选[播放-开始](具体可参考【媒体库服务器通知】插件)', 'style': 'white-space: pre-line;' } } ] } ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, }, 'content': [ { 'component': 'VAlert', 'props': { 'type': 'info', 'variant': 'tonal', 'text': '- 播放视频分辨率与上次视频分辨率不一致时，在通知延迟和已加载旧位置字幕影响下，需要片刻后才会加载到新位置字幕，或关闭视频再次打开(建议)。\n - 此替换文本参数应用于emby-Diagnostics-Parameter Adjustment。\n - 默认参数用于修改ffmpeg中字幕覆盖在视频上的位置。\n - 方案来源于https://opve.cn/archives/983.html', 'style': 'white-space: pre-line;' } } ] } ] } ] } ], { "enabled": False, "offset_play": True, "onlyonce": False, "search": "x=(W-w)/2:y=(H-h):repeatlast=0", "replace": "x=W/4:y=h/4:repeatlast=0", "cron": "*/5 * * * *", "cron_switch": False, } def detect(self): """ 检测是否存在目标参数（修正字幕偏移用途） :return True: 存在; False: 不存在 """ logger.info('字幕偏移修正，检测目标参数') try: res = Emby().get_data(self._url) result = res.json() data = result['Object']['CommandLineOptions'] searchText = data['SearchText'] replaceText = data['ReplaceText'] except json.JSONDecodeError: logger.error('服务停止，Emby请安装【Diagnostics】插件') return None except KeyError: # 已装插件，未设置过该参数 # logger.info('目标参数为空') return False # 符合所有情况 if (('repeatlast' in replaceText \ and 'x=(W-w)/2:y=(H-h):repeatlast=0' in searchText \ and result['Object']['TranscodingOptions']['DisableHardwareSubtitleOverlay'] is True) \ or (searchText == "" and replaceText == "")) \ and self._resolution == self._last_resolution: # (A or B) and C return True return False def set_options(self): """ 向Emby发送请求设置参数 """ # 根据分辨率情况而选择是否替换 if self._resolution == 0 and self._offset_play is True: # 1080p，不替换（清空文本） self._target_search_text = "" self._target_replace_text = "" logger.info('清空替换参数') else: # >1080p or 非字幕偏移用途 self._target_search_text = self._search_text self._target_replace_text = self._replace_text logger.info("替换值为：{}".format(self._target_replace_text)) data = { "CommandLineOptions": { "SearchText": self._target_search_text, "ReplaceText": self._target_replace_text }, "TranscodingOptions": { "DisableHardwareSubtitleOverlay": True } } data = json.dumps(data) headers = { 'Content-Type': 'application/octet-stream' } res = Emby().post_data(self._url, data, headers) if res.status_code // 100 == 2: logger.info('参数设置成功') return True else: logger.error('参数设置失败 {}'.format(res.status_code)) return False @eventmanager.register(EventType.WebhookMessage) def get_msg(self, event: Event): # 消息方式开关 if not self._enabled or not self._offset_play: return # 消息获取 event_info: WebhookEventInfo = event.event_data if not event_info: return # 非目标消息 if not self._webhook_actions.get(event_info.event): return # 根据视频名获得分辨率信息 item_path = event_info.item_path video_resolution = re.findall(r"\d{3,4}p", item_path) video_width = int(video_resolution[0][:-1]) logger.info('视频分辨率:{}'.format(video_width)) self._last_resolution = self._resolution # 分辨率变化情况 if video_width > 1080: # 2160p/4k self._resolution = 1 else: self._resolution = 0 self.run() def run(self): # 字幕偏移修正，则带检测 if self._offset_play: state = self.detect() if state: logger.info('参数正常，无需修正') return True elif state is None: logger.info('插件退出') return None self.set_options() def get_page(self) -> List[dict]: pass def stop_service(self): """ 退出插件 """ try: if self._scheduler: self._scheduler.remove_all_jobs() if self._scheduler.running: self._scheduler.shutdown() self._scheduler = None except Exception as e: logger.error("退出插件失败：%s" % str(e))