diff --git a/README.md b/README.md index 477af1e..814183d 100644 --- a/README.md +++ b/README.md @@ -22,4 +22,5 @@ MoviePilot三方插件市场：https://github.com/thsrite/MoviePilot-Plugins/ - [同步CookieCloud 1.1](docs%2FSyncCookieCloud.md) - [日程提醒 1.0](docs%2FScheduleReminder.md) - [订阅提醒 1.1](docs%2FSubscribeReminder.md) +- [Emby观影报告 1.0](docs%2FEmbyReporter.md) diff --git a/docs/EmbyReporter.md b/docs/EmbyReporter.md new file mode 100644 index 0000000..a970701 --- /dev/null +++ b/docs/EmbyReporter.md @@ -0,0 +1,11 @@ +# Emby观影报告 + +### 更新记录 + +- 1.0 推送Emby观影报告 + +### 使用说明 + +需Emby安装Playback Report 插件 + +插件依赖安装失败，请手动执行 `pip install python_telegram_bot` \ No newline at end of file diff --git a/package.json b/package.json index 8e50bc3..44e0d0b 100644 --- a/package.json +++ b/package.json @@ -102,5 +102,13 @@ "icon": "https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/icons/subscribe_reminder.png", "author": "thsrite", "level": 1 + }, + "EmbyReporter": { + "name": "Emby观影报告", + "description": "推送Emby观影报告，需Emby安装Playback Report 插件。", + "version": "1.0", + "icon": "Pydiocells_A.png", + "author": "thsrite", + "level": 1 } } diff --git a/plugins/embyreporter/__init__.py b/plugins/embyreporter/__init__.py new file mode 100644 index 0000000..821ec76 --- /dev/null +++ b/plugins/embyreporter/__init__.py @@ -0,0 +1,356 @@ +from datetime import datetime, timedelta + +import pytz +from telegram.bot import Bot, Request +from telegram import ParseMode +from app.core.config import settings +from app.modules.wechat import WeChat +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.plugins.embyreporter.emby import EmbyService +from app.plugins.embyreporter.ranks_draw import RanksDraw + + +class EmbyReporter(_PluginBase): + # 插件名称 + plugin_name = "Emby观影报告" + # 插件描述 + plugin_desc = "推送Emby观影报告，需Emby安装Playback Report 插件。" + # 插件图标 + plugin_icon = "Pydiocells_A.png" + # 插件版本 + plugin_version = "1.0" + # 插件作者 + 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 + _mp_host = None + _scheduler: Optional[BackgroundScheduler] = None + + def init_plugin(self, config: dict = None): + # 停止现有任务 + self.stop_service() + + 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._type = config.get("type") or "tg" + self._mp_host = config.get("mp_host") + + 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观影报告 + """ + # 初始化对象 + emby = EmbyService(settings.EMBY_HOST, settings.EMBY_API_KEY) + draw = RanksDraw(emby, self._res_dir) + + # 获取数据 + success, movies = emby.get_report(types=emby.PLAYBACK_REPORTING_TYPE_MOVIE, days=self._days, limit=5) + if not success: + exit(movies) + success, tvshows = emby.get_report(types=emby.PLAYBACK_REPORTING_TYPE_TVSHOWS, days=self._days, limit=5) + if not success: + exit(tvshows) + + # 绘制海报 + draw.draw(movies, tvshows) + report_path = draw.save() + + # 发送海报 + if not self._type: + return + + report_text = f"🌟*过去{self._days}日观影排行*\r\n\r\n" + if str(self._type) == "tg": + proxy = Request(proxy_url=settings.PROXY_HOST) + bot = Bot(token=settings.TELEGRAM_TOKEN, request=proxy) + bot.send_photo( + chat_id=settings.TELEGRAM_CHAT_ID, + photo=open(report_path, "rb"), + caption=report_text, + parse_mode=ParseMode.MARKDOWN_V2 + ) + logger.info("Emby观影记录推送Telegram成功") + else: + # 本地路径转为url + if not self._mp_host: + return + + report_url = self._mp_host + report_path.replace("/public", "") + WeChat().send_msg(title=report_text, + image=report_url) + logger.info("Emby观影记录推送微信应用成功") + + def __update_config(self): + self.update_config({ + "enabled": self._enabled, + "onlyonce": self._onlyonce, + "cron": self._cron, + "days": self._days, + "mp_host": self._mp_host, + "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、数据结构 + """ + # 编历 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': 'VSelect', + 'props': { + 'multiple': False, + 'chips': True, + 'model': 'type', + 'label': '推送方式', + 'items': [ + {'title': 'Telegram', 'value': "tg"}, + {'title': '微信', 'value': "wx"} + ] + } + } + ] + }, + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12 + }, + 'content': [ + { + 'component': 'VTextField', + 'props': { + 'model': 'mp_host', + 'label': 'MoviePilot域名', + 'placeholder': '推送方式非tg可用' + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VAlert', + 'props': { + 'type': 'info', + 'variant': 'tonal', + 'text': 'MoviePilot域名仅在微信推送方式时需要填写。末尾不带/' + } + } + ] + } + ] + } + ] + } + ], { + "enabled": False, + "onlyonce": False, + "cron": "5 1 * * *", + "res_dir": "", + "days": 7, + "mp_host": "", + "type": "tg" + } + + 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)) diff --git a/plugins/embyreporter/emby.py b/plugins/embyreporter/emby.py new file mode 100644 index 0000000..f1442e1 --- /dev/null +++ b/plugins/embyreporter/emby.py @@ -0,0 +1,109 @@ +import pytz +import requests +from cacheout import Cache +from datetime import datetime, timedelta + +cache = Cache() + + +class EmbyService(): + PLAYBACK_REPORTING_TYPE_MOVIE = "ItemName" + PLAYBACK_REPORTING_TYPE_TVSHOWS = "substr(ItemName,0, instr(ItemName, ' - '))" + + def __init__(self, host, api_key): + self.host = host + self.base_url = host + "/emby/{0}?api_key=" + api_key + self.api_key = api_key + + @cache.memoize(ttl=600) + def primary(self, item_id, width=200, height=300, quality=90, ret_url=False): + url = self.host + f"/emby/Items/{item_id}/Images/Primary?maxHeight={height}&maxWidth={width}&quality={quality}" + if ret_url: + return url + resp = requests.get(url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.content + + @cache.memoize(ttl=600) + def backdrop(self, item_id, width=1920, quality=70, ret_url=False): + url = self.host + f"/emby/Items/{item_id}/Images/Backdrop/0?&maxWidth={width}&quality={quality}" + if ret_url: + return url + resp = requests.get(url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.content + + @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 = requests.get(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): + url = self.base_url.format(f"Users/{user_id}/Items/{item_id}") + resp = requests.get(url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.json() + + def get_report(self, types=None, user_id=None, days=7, 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=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_duarion " + 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 play_count DESC " + sql += "LIMIT " + str(limit) + + url = self.base_url.format(f"user_usage_stats/submit_custom_query") + data = { + "CustomQueryString": sql, + "ReplaceUserId": False + } + resp = requests.post(url, 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"] + + +class LibraryService: + def __init__(self, host, api_key): + self.base_url = host + "/Library/{0}?api_key=" + api_key + + """ + Gets all user media folders + """ + + @cache.memoize(ttl=600) + def folders(self): + url = self.base_url.format("SelectableMediaFolders") + resp = requests.get(url) + + if resp.status_code != 204 and resp.status_code != 200: + return False, "🤕Emby 服务器连接失败!" + return True, resp.json() \ No newline at end of file diff --git a/plugins/embyreporter/ranks_draw.py b/plugins/embyreporter/ranks_draw.py new file mode 100644 index 0000000..f0bc630 --- /dev/null +++ b/plugins/embyreporter/ranks_draw.py @@ -0,0 +1,91 @@ +import os +from pathlib import Path + +import pytz +import random +from io import BytesIO +from PIL import Image +from PIL import ImageFont +from PIL import ImageDraw +from datetime import datetime +from utils import draw_text_psd_style + +""" +Misty 周榜海报样式 +你可以根据你的需求自行封装或更改为你自己的周榜海报样式！ +""" + + +class RanksDraw: + def __init__(self, emby, res_path): + # 默认路径 默认图 + 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)]) + # 初始绘图对象 + self.bg = Image.open(bg_path) + mask = Image.open(mask_path) + self.bg.paste(mask, (0, 0), mask) + self.font = ImageFont.truetype(font_path, 18) + self.font_small = ImageFont.truetype(font_path, 14) + self.font_count = ImageFont.truetype(font_path, 12) + # 初始化封面对象 + self.emby = emby + + def draw(self, movies, tvshows, show_count=True): + # 合并绘制 + all_ranks = movies + tvshows + index, offset_y = (0, 0) + for i in all_ranks: + # 榜单项数据 + user_id, item_id, item_type, name, count, duarion = tuple(i) + print(item_type, item_id, name, count) + # 图片获取，剧集主封面获取 + if item_type != "Movie": + # 获取剧ID + success, data = self.emby.items(user_id, item_id) + if not success: + exit(data) + item_id = data["SeriesId"] + # 封面图像获取 + success, data = self.emby.primary(item_id) + if not success: + exit(data) + # 剧集Y偏移 + if index >= 5: + index = 0 + offset_y = 331 + # 名称显示偏移 + font_offset_y = 0 + temp_font = self.font + # 名称超出长度缩小省略 + if self.font.getlength(name) > 110: + temp_font = self.font_small + font_offset_y = 4 + for i in range(len(name)): + name = name[:len(name) - 1] + if self.font.getlength(name) <= 110: + break + name += ".." + # 绘制封面 + cover = Image.open(BytesIO(data)) + cover = cover.resize((108, 159)) + self.bg.paste(cover, (73 + 145 * index, 379 + offset_y)) + # 绘制 播放次数、影片名称 + text = ImageDraw.Draw(self.bg) + if show_count: + draw_text_psd_style(text, (177 + 145 * index - self.font_count.getlength(str(count)), 353 + offset_y), + str(count), self.font_count, 126) + draw_text_psd_style(text, (74 + 145 * index, 542 + font_offset_y + offset_y), name, temp_font, 126) + index += 1 + + def save(self, save_path=os.path.join("/public", "ranks", + datetime.now(pytz.timezone("Asia/Shanghai")).strftime("%Y-%m-%d.jpg"))): + self.bg.save(save_path) + return save_path diff --git a/plugins/embyreporter/requirements.txt b/plugins/embyreporter/requirements.txt new file mode 100644 index 0000000..19c4f90 --- /dev/null +++ b/plugins/embyreporter/requirements.txt @@ -0,0 +1 @@ +python_telegram_bot==13.7 \ No newline at end of file diff --git a/plugins/embyreporter/res/PingFang Bold.ttf b/plugins/embyreporter/res/PingFang Bold.ttf new file mode 100644 index 0000000..accaf1f Binary files /dev/null and b/plugins/embyreporter/res/PingFang Bold.ttf differ diff --git a/plugins/embyreporter/res/bg/0.jpg b/plugins/embyreporter/res/bg/0.jpg new file mode 100644 index 0000000..91038ee Binary files /dev/null and b/plugins/embyreporter/res/bg/0.jpg differ diff --git a/plugins/embyreporter/res/bg/1.jpg b/plugins/embyreporter/res/bg/1.jpg new file mode 100644 index 0000000..8db3393 Binary files /dev/null and b/plugins/embyreporter/res/bg/1.jpg differ diff --git a/plugins/embyreporter/res/bg/2.jpg b/plugins/embyreporter/res/bg/2.jpg new file mode 100644 index 0000000..5d114ee Binary files /dev/null and b/plugins/embyreporter/res/bg/2.jpg differ diff --git a/plugins/embyreporter/res/bg/3.jpg b/plugins/embyreporter/res/bg/3.jpg new file mode 100644 index 0000000..f964c26 Binary files /dev/null and b/plugins/embyreporter/res/bg/3.jpg differ diff --git a/plugins/embyreporter/res/bg/4.jpg b/plugins/embyreporter/res/bg/4.jpg new file mode 100644 index 0000000..bc665b6 Binary files /dev/null and b/plugins/embyreporter/res/bg/4.jpg differ diff --git a/plugins/embyreporter/res/bg/5.jpg b/plugins/embyreporter/res/bg/5.jpg new file mode 100644 index 0000000..95ef6a5 Binary files /dev/null and b/plugins/embyreporter/res/bg/5.jpg differ diff --git a/plugins/embyreporter/res/bg/6.jpg b/plugins/embyreporter/res/bg/6.jpg new file mode 100644 index 0000000..1343af2 Binary files /dev/null and b/plugins/embyreporter/res/bg/6.jpg differ diff --git a/plugins/embyreporter/res/cover-ranks-1.psd b/plugins/embyreporter/res/cover-ranks-1.psd new file mode 100644 index 0000000..00e1397 Binary files /dev/null and b/plugins/embyreporter/res/cover-ranks-1.psd differ diff --git a/plugins/embyreporter/res/cover-ranks-2.psd b/plugins/embyreporter/res/cover-ranks-2.psd new file mode 100644 index 0000000..7fa49b3 Binary files /dev/null and b/plugins/embyreporter/res/cover-ranks-2.psd differ diff --git a/plugins/embyreporter/res/cover-ranks-mask-1.png b/plugins/embyreporter/res/cover-ranks-mask-1.png new file mode 100644 index 0000000..8d1003a Binary files /dev/null and b/plugins/embyreporter/res/cover-ranks-mask-1.png differ diff --git a/plugins/embyreporter/res/cover-ranks-mask-2.png b/plugins/embyreporter/res/cover-ranks-mask-2.png new file mode 100644 index 0000000..335c61b Binary files /dev/null and b/plugins/embyreporter/res/cover-ranks-mask-2.png differ diff --git a/plugins/embyreporter/res/cover-ranks-mask-test.png b/plugins/embyreporter/res/cover-ranks-mask-test.png new file mode 100644 index 0000000..3a3f016 Binary files /dev/null and b/plugins/embyreporter/res/cover-ranks-mask-test.png differ diff --git a/plugins/embyreporter/utils.py b/plugins/embyreporter/utils.py new file mode 100644 index 0000000..c193ddd --- /dev/null +++ b/plugins/embyreporter/utils.py @@ -0,0 +1,34 @@ +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] \ No newline at end of file