diff --git a/package.json b/package.json index 64c0bf3..a10e3ec 100644 --- a/package.json +++ b/package.json @@ -222,7 +222,6 @@ "icon": "Pydiocells_A.png", "author": "thsrite", "level": 1, - "v2": true, "history": { "v1.8": "推送微信增加时间戳，防止使用缓存", "v1.7": "分块发送，兼容微信推送", diff --git a/package.v2.json b/package.v2.json index d2f3551..095ddfe 100644 --- a/package.v2.json +++ b/package.v2.json @@ -66,5 +66,25 @@ "v1.1": "添加远程交互命令", "v1.0": "自动给媒体库媒体添加标签" } + }, + "EmbyReporter": { + "name": "Emby观影报告", + "description": "推送Emby观影报告，需Emby安装Playback Report 插件。", + "labels": "Emby", + "version": "1.8", + "icon": "Pydiocells_A.png", + "author": "thsrite", + "level": 1, + "history": { + "v1.8": "推送微信增加时间戳，防止使用缓存", + "v1.7": "分块发送，兼容微信推送", + "v1.6": "fix #67", + "v1.5": "按观影市场排序", + "v1.4": "支持自定义emby && 支持每日一言", + "v1.3": "修复bug", + "v1.2": "过滤已删除媒体", + "v1.1": "修复推送", + "v1.0": "推送Emby观影报告" + } } } diff --git a/plugins.v2/embymetarefresh/__init__.py b/plugins.v2/embymetarefresh/__init__.py index 333c9a5..8de717e 100644 --- a/plugins.v2/embymetarefresh/__init__.py +++ b/plugins.v2/embymetarefresh/__init__.py @@ -92,7 +92,7 @@ class EmbyMetaRefresh(_PluginBase): self._EMBY_USER = emby_server.instance.get_user() self._EMBY_HOST = emby_server.config.get("host") - self._EMBY_HOST = emby_server.config.get("apikey") + self._EMBY_APIKEY = emby_server.config.get("apikey") if not self._EMBY_HOST.endswith("/"): self._EMBY_HOST += "/" if not self._EMBY_HOST.startswith("http"): diff --git a/plugins.v2/embymetatag/__init__.py b/plugins.v2/embymetatag/__init__.py index 1e7e413..b6d7055 100644 --- a/plugins.v2/embymetatag/__init__.py +++ b/plugins.v2/embymetatag/__init__.py @@ -71,7 +71,7 @@ class EmbyMetaTag(_PluginBase): self._EMBY_USER = emby_server.instance.get_user() self._EMBY_HOST = emby_server.config.get("host") - self._EMBY_HOST = emby_server.config.get("apikey") + self._EMBY_APIKEY = emby_server.config.get("apikey") if not self._EMBY_HOST.endswith("/"): self._EMBY_HOST += "/" if not self._EMBY_HOST.startswith("http"): diff --git a/plugins.v2/embyreporter/__init__.py b/plugins.v2/embyreporter/__init__.py new file mode 100644 index 0000000..b6bf199 --- /dev/null +++ b/plugins.v2/embyreporter/__init__.py @@ -0,0 +1,800 @@ +import os + +from app.core.config import settings +from app.helper.mediaserver import MediaServerHelper +from app.plugins import _PluginBase +from typing import Any, List, Dict, Tuple, Optional +from app.log import logger +from apscheduler.schedulers.background import BackgroundScheduler +from apscheduler.triggers.cron import CronTrigger + +from app.schemas import NotificationType +from pathlib import Path + +import random +from io import BytesIO +from PIL import Image +from PIL import ImageFont +from PIL import ImageDraw +import pytz +from cacheout import Cache +from datetime import datetime, timedelta + +from app.utils.http import RequestUtils +from app.utils.string import StringUtils + +cache = Cache() + + +class EmbyReporter(_PluginBase): + # 插件名称 + plugin_name = "Emby观影报告" + # 插件描述 + plugin_desc = "推送Emby观影报告，需Emby安装Playback Report 插件。" + # 插件图标 + plugin_icon = "Pydiocells_A.png" + # 插件版本 + plugin_version = "1.8" + # 插件作者 + plugin_author = "thsrite" + # 作者主页 + author_url = "https://github.com/thsrite" + # 插件配置项ID前缀 + plugin_config_prefix = "embyreporter_" + # 加载顺序 + plugin_order = 30 + # 可使用的用户级别 + auth_level = 1 + + # 私有属性 + _enabled: bool = False + _onlyonce: bool = False + _res_dir = None + _cron = None + _days = None + _type = None + _cnt = None + _mp_host = None + _emby_host = None + _emby_api_key = None + show_time = True + _scheduler: Optional[BackgroundScheduler] = None + mediaserver_helper = None + PLAYBACK_REPORTING_TYPE_MOVIE = "ItemName" + PLAYBACK_REPORTING_TYPE_TVSHOWS = "substr(ItemName,0, instr(ItemName, ' - '))" + host = None + api_key = None + + def init_plugin(self, config: dict = None): + # 停止现有任务 + self.stop_service() + self.mediaserver_helper = MediaServerHelper() + emby_server = self.mediaserver_helper.get_service(name="emby") + if not emby_server: + logger.error("未配置Emby媒体服务器") + return + self.host = emby_server.config.get("host") + self.api_key = emby_server.config.get("apikey") + + if config: + self._enabled = config.get("enabled") + self._onlyonce = config.get("onlyonce") + self._cron = config.get("cron") + self._res_dir = config.get("res_dir") + self._days = config.get("days") or 7 + self._cnt = config.get("cnt") or 10 + self._type = config.get("type") or "tg" + self._mp_host = config.get("mp_host") + self.show_time = config.get("show_time") + self._emby_host = config.get("emby_host") + self._emby_api_key = config.get("emby_api_key") + if self._emby_host and self._emby_api_key: + self.host = f"http://{self._emby_host}" if not str(self._emby_host).startswith( + "http") else self._emby_host + self.api_key = self._emby_api_key + + if self._enabled or self._onlyonce: + # 定时服务 + self._scheduler = BackgroundScheduler(timezone=settings.TZ) + + # 立即运行一次 + if self._onlyonce: + logger.info(f"Emby观影报告服务启动，立即运行一次") + self._scheduler.add_job(self.__report, 'date', + run_date=datetime.now( + tz=pytz.timezone(settings.TZ)) + timedelta(seconds=3), + name="Emby观影报告") + # 关闭一次性开关 + self._onlyonce = False + + # 保存配置 + self.__update_config() + + # 周期运行 + if self._cron: + try: + self._scheduler.add_job(func=self.__report, + trigger=CronTrigger.from_crontab(self._cron), + name="Emby观影报告") + except Exception as err: + logger.error(f"定时任务配置错误：{err}") + # 推送实时消息 + self.systemmessage.put(f"执行周期配置错误：{err}") + + # 启动任务 + if self._scheduler.get_jobs(): + self._scheduler.print_jobs() + self._scheduler.start() + + def __report(self): + """ + 发送Emby观影报告 + """ + # 本地路径转为url + if not self._mp_host: + return + + if not self._type: + return + + # 获取当前时间并格式化 + current_time = datetime.now().strftime("%Y%m%d%H%M%S") + + # 获取数据 + success, movies = self.get_report(types=self.PLAYBACK_REPORTING_TYPE_MOVIE, days=int(self._days), + limit=int(self._cnt)) + if not success: + exit(movies) + logger.info(f"获取到电影 {movies}") + success, tvshows = self.get_report(types=self.PLAYBACK_REPORTING_TYPE_TVSHOWS, days=int(self._days), + limit=int(self._cnt)) + if not success: + exit(tvshows) + logger.info(f"获取到电视剧 {tvshows}") + + # 绘制海报 + report_path = self.draw(res_path=self._res_dir, + movies=movies, + tvshows=tvshows, + show_time=self.show_time) + + if not report_path: + logger.error("生成海报失败") + return + + # 示例调用 + self.__split_image_by_height(report_path, "/public/report", [250, 330, 335]) + + # 分块推送 + for i in range(2, 4): + report_path_part = f"/public/report_part_{i}.jpg" + report_url = self._mp_host + report_path_part.replace("/public", "") + f"?_timestamp={current_time}" + mtype = NotificationType.MediaServer + if self._type: + mtype = NotificationType.__getitem__(str(self._type)) or NotificationType.MediaServer + + self.post_message( + title=f'Movies 近{self._days}日观影排行' if i == 2 else f'TV Shows 近{self._days}日观影排行', + mtype=mtype, + image=report_url) + logger.info(f"Emby观影记录推送成功 {report_url}") + + @staticmethod + def __split_image_by_height(image_path, output_path_prefix, heights): + # 打开原始图像 + img = Image.open(image_path) + img_width, img_height = img.size + + # 如果图像是 RGBA 模式，转换为 RGB 模式 + if img.mode == 'RGBA': + img = img.convert('RGB') + + # 分割图像的起始位置 + top = 0 + + # 按指定高度分割图像 + for i, height in enumerate(heights): + # 确保不会超出图像边界 + if top + height > img_height: + height = img_height - top + + bottom = top + height + + # 裁剪图像 + box = (0, top, img_width, bottom) + part = img.crop(box) + + # 保存图像部分 + part.save(f"{output_path_prefix}_part_{i + 1}.jpg") + + # 更新下一个部分的上边界 + top = bottom + + # 如果已经到达图像底部，停止 + if top >= img_height: + break + + print("图片按照指定高度分割完成！") + + def __update_config(self): + self.update_config({ + "enabled": self._enabled, + "onlyonce": self._onlyonce, + "cron": self._cron, + "days": self._days, + "cnt": self._cnt, + "type": self._type, + "mp_host": self._mp_host, + "show_time": self.show_time, + "emby_host": self._emby_host, + "emby_api_key": self._emby_api_key, + "res_dir": self._res_dir + }) + + 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_form(self) -> Tuple[List[dict], Dict[str, Any]]: + """ + 拼装插件配置页面，需要返回两块数据：1、页面配置；2、数据结构 + """ + MsgTypeOptions = [] + for item in NotificationType: + MsgTypeOptions.append({ + "title": item.value, + "value": item.name + }) + # 编历 NotificationType 枚举，生成消息类型选项 + 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': 'onlyonce', + 'label': '立即运行一次', + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'cron', + 'label': '执行周期', + 'placeholder': '5位cron表达式，留空自动' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'res_dir', + 'label': '素材路径', + 'placeholder': '本地素材路径' + } + } + ] + }, + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'days', + 'label': '报告天数', + 'placeholder': '向前获取数据的天数' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'cnt', + 'label': '观影记录数量', + 'placeholder': '获取观影数据数量，默认10' + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'mp_host', + 'label': 'MoviePilot域名', + 'placeholder': '必填，末尾不带/' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VSelect', + 'props': { + 'multiple': False, + 'chips': True, + 'model': 'type', + 'label': '推送方式', + 'items': MsgTypeOptions + } + } + ] + }, + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VSelect', + 'props': { + 'model': 'show_time', + 'label': '是否显示观看时长', + 'items': [ + {'title': '是', 'value': True}, + {'title': '否', 'value': False} + ] + } + } + ] + }, + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'emby_host', + 'label': '自定义emby host', + 'placeholder': 'IP:PORT' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'emby_api_key', + 'label': '自定义emby apiKey' + } + } + ] + } + ] + }, + { + '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': 'info', + 'variant': 'tonal', + 'text': '如未设置自定义emby配置，则读取环境变量emby配置。' + } + } + ] + } + ] + } + ] + } + ], { + "enabled": False, + "onlyonce": False, + "cron": "5 1 * * *", + "res_dir": "", + "days": 7, + "cnt": 10, + "emby_host": "", + "emby_api_key": "", + "mp_host": "", + "show_time": True, + "type": "" + } + + 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)) + + def draw(self, res_path, movies, tvshows, show_time=True): + # 默认路径 默认图 + if not res_path: + res_path = os.path.join(Path(__file__).parent, "res") + # 绘图文件路径初始化 + bg_path = os.path.join(res_path, "bg") + mask_path = os.path.join(res_path, "cover-ranks-mask-2.png") + font_path = os.path.join(res_path, "PingFang Bold.ttf") + # 随机调取背景, 路径: res/ranks/bg/... + bg_list = os.listdir(bg_path) + bg_path = os.path.join(bg_path, bg_list[random.randint(0, len(bg_list) - 1)]) + # 初始绘图对象 + bg = Image.open(bg_path) + mask = Image.open(mask_path) + bg.paste(mask, (0, 0), mask) + font = ImageFont.truetype(font_path, 18) + font_small = ImageFont.truetype(font_path, 14) + font_count = ImageFont.truetype(font_path, 8) + + exists_movies = [] + for i in movies: + try: + # 榜单项数据 + user_id, item_id, item_type, name, count, duration = tuple(i) + print(item_type, item_id, name, count, StringUtils.str_secends(int(duration))) + # 封面图像获取 + success, data = self.primary(item_id) + if not success: + continue + exists_movies.append(i) + except Exception: + continue + + logger.info(f"过滤后未删除电影 {len(exists_movies)} 部") + # 合并绘制 + if len(exists_movies) < 5: + for i in range(5 - len(exists_movies) + 1): + exists_movies.append({"item_id": i}) + if len(exists_movies) > 5: + exists_movies = exists_movies[:5] + + exists_tvs = [] + for i in tvshows: + try: + # 榜单项数据 + user_id, item_id, item_type, name, count, duration = tuple(i) + print(item_type, item_id, name, count, StringUtils.str_secends(int(duration))) + # 图片获取，剧集主封面获取 + # 获取剧ID + success, data = self.items(user_id, item_id) + if not success: + continue + item_id = data["SeriesId"] + # 封面图像获取 + success, data = self.primary(item_id) + if not success: + continue + exists_tvs.append(i) + except Exception as e: + print(str(e)) + continue + logger.info(f"过滤后未删除电视剧 {len(exists_tvs)} 部") + if len(exists_tvs) > 5: + exists_tvs = exists_tvs[:5] + + all_ranks = exists_movies + exists_tvs + index, offset_y = (-1, 0) + for i in all_ranks: + index += 1 + try: + # 榜单项数据 + user_id, item_id, item_type, name, count, duration = tuple(i) + # 图片获取，剧集主封面获取 + if item_type != "Movie": + # 获取剧ID + success, data = self.items(user_id, item_id) + if not success: + index -= 1 + continue + item_id = data["SeriesId"] + # 封面图像获取 + success, data = self.primary(item_id) + if not success: + if item_type != "Movie": + index -= 1 + continue + # 剧集Y偏移 + if index >= 5: + index = 0 + offset_y = 331 + # 名称显示偏移 + font_offset_y = 0 + temp_font = font + # 名称超出长度缩小省略 + if font.getlength(name) > 110: + temp_font = font_small + font_offset_y = 4 + for i in range(len(name)): + name = name[:len(name) - 1] + if font.getlength(name) <= 110: + break + name += ".." + # 绘制封面 + cover = Image.open(BytesIO(data)) + cover = cover.resize((108, 159)) + bg.paste(cover, (73 + 145 * index, 379 + offset_y)) + # 绘制 播放次数、影片名称 + text = ImageDraw.Draw(bg) + if show_time: + self.draw_text_psd_style(text, + (177 + 145 * index - font_count.getlength( + StringUtils.str_secends(int(duration))), + 355 + offset_y), + StringUtils.str_secends(int(duration)), font_count, 126) + self.draw_text_psd_style(text, (74 + 145 * index, 542 + font_offset_y + offset_y), name, temp_font, 126) + except Exception: + continue + + if index >= 0: + save_path = "/public/report.jpg" + if Path(save_path).exists(): + Path.unlink(Path(save_path)) + bg.save(save_path) + return save_path + return None + + @staticmethod + def draw_text_psd_style(draw, xy, text, font, tracking=0, leading=None, **kwargs): + """ + usage: draw_text_psd_style(draw, (0, 0), "Test", + tracking=-0.1, leading=32, fill="Blue") + + Leading is measured from the baseline of one line of text to the + baseline of the line above it. Baseline is the invisible line on which most + letters—that is, those without descenders—sit. The default auto-leading + option sets the leading at 120% of the type size (for example, 12‑point + leading for 10‑point type). + + Tracking is measured in 1/1000 em, a unit of measure that is relative to + the current type size. In a 6 point font, 1 em equals 6 points; + in a 10 point font, 1 em equals 10 points. Tracking + is strictly proportional to the current type size. + """ + + def stutter_chunk(lst, size, overlap=0, default=None): + for i in range(0, len(lst), size - overlap): + r = list(lst[i:i + size]) + while len(r) < size: + r.append(default) + yield r + + x, y = xy + font_size = font.size + lines = text.splitlines() + if leading is None: + leading = font.size * 1.2 + for line in lines: + for a, b in stutter_chunk(line, 2, 1, ' '): + w = font.getlength(a + b) - font.getlength(b) + draw.text((x, y), a, font=font, **kwargs) + x += w + (tracking / 1000) * font_size + y += leading + x = xy[0] + + @cache.memoize(ttl=600) + def primary(self, item_id, width=720, height=1440, quality=90, ret_url=False): + try: + url = self.host + f"/emby/Items/{item_id}/Images/Primary?maxHeight={height}&maxWidth={width}&quality={quality}" + if ret_url: + return url + resp = RequestUtils().get_res(url=url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.content + except Exception: + return False, "🤕Emby 服务器连接失败!" + + @cache.memoize(ttl=600) + def backdrop(self, item_id, width=1920, quality=70, ret_url=False): + try: + url = self.host + f"/emby/Items/{item_id}/Images/Backdrop/0?&maxWidth={width}&quality={quality}" + if ret_url: + return url + resp = RequestUtils().get_res(url=url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.content + except Exception: + return False, "🤕Emby 服务器连接失败!" + + @cache.memoize(ttl=600) + def logo(self, item_id, quality=70, ret_url=False): + url = self.host + f"/emby/Items/{item_id}/Images/Logo?quality={quality}" + if ret_url: + return url + resp = RequestUtils().get_res(url=url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.content + + @cache.memoize(ttl=300) + def items(self, user_id, item_id): + try: + url = f"{self.host}/emby/Users/{user_id}/Items/{item_id}?api_key={self.api_key}" + resp = RequestUtils().get_res(url=url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.json() + except Exception: + return False, "🤕Emby 服务器连接失败!" + + def get_report(self, days, types=None, user_id=None, end_date=datetime.now(pytz.timezone("Asia/Shanghai")), + limit=10): + if not types: + types = self.PLAYBACK_REPORTING_TYPE_MOVIE + sub_date = end_date - timedelta(days=int(days)) + start_time = sub_date.strftime("%Y-%m-%d 00:00:00") + end_time = end_date.strftime("%Y-%m-%d 23:59:59") + sql = "SELECT UserId, ItemId, ItemType, " + sql += types + " AS name, " + sql += "COUNT(1) AS play_count, " + sql += "SUM(PlayDuration - PauseDuration) AS total_duration " + sql += "FROM PlaybackActivity " + sql += f"WHERE ItemType = '{'Movie' if types == self.PLAYBACK_REPORTING_TYPE_MOVIE else 'Episode'}' " + sql += f"AND DateCreated >= '{start_time}' AND DateCreated <= '{end_time}' " + sql += "AND UserId not IN (select UserId from UserList) " + if user_id: + sql += f"AND UserId = '{user_id}' " + sql += "GROUP BY name " + sql += "ORDER BY total_duration DESC " + sql += "LIMIT " + str(limit) + + url = f"{self.host}/emby/user_usage_stats/submit_custom_query?api_key={self.api_key}" + + data = { + "CustomQueryString": sql, + "ReplaceUserId": False + } + resp = RequestUtils().post_res(url=url, data=data) + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + ret = resp.json() + if len(ret["colums"]) == 0: + return False, ret["message"] + return True, ret["results"]