mirror of
https://github.com/thsrite/MoviePilot-Plugins.git
synced 2026-03-27 10:05:57 +00:00
feat 订阅提醒插件
This commit is contained in:
thsrite
@@ -21,5 +21,5 @@ MoviePilot三方插件市场：https://github.com/thsrite/MoviePilot-Plugins/
|
||||
- [群辉Webhook通知 1.1](plugins_record%2FSynologyNotify.md)
|
||||
- [同步CookieCloud 1.1](plugins_record%2FSyncCookieCloud.md)
|
||||
- [日程提醒 1.0](plugins_record%2FScheduleReminder.md)
|
||||
|
||||
- [订阅提醒 1.0](plugins_record%2FSubscribeReminder.md)
|
||||
|
||||
|
||||
BIN
icons/subscribe_reminder.png
Normal file
BIN
icons/subscribe_reminder.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 4.7 KiB |
@@ -94,5 +94,13 @@
|
||||
"icon": "https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/icons/reminder.png",
|
||||
"author": "thsrite",
|
||||
"level": 1
|
||||
},
|
||||
"SubscribeReminder": {
|
||||
"name": "订阅提醒",
|
||||
"description": "推送当天订阅更新内容。",
|
||||
"version": "1.0",
|
||||
"icon": "https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/icons/subscribe_reminder.png",
|
||||
"author": "thsrite",
|
||||
"level": 1
|
||||
}
|
||||
}
|
||||
|
||||
284
plugins/subscribereminder/__init__.py
Normal file
284
plugins/subscribereminder/__init__.py
Normal file
@@ -0,0 +1,284 @@
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
import pytz
|
||||
from app.chain.media import MediaChain
|
||||
from app.chain.tmdb import TmdbChain
|
||||
from app.core.config import settings
|
||||
from app.db.subscribe_oper import SubscribeOper
|
||||
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, MediaType
|
||||
|
||||
|
||||
class SubscribeReminder(_PluginBase):
|
||||
# 插件名称
|
||||
plugin_name = "订阅提醒"
|
||||
# 插件描述
|
||||
plugin_desc = "推送当天订阅更新内容。"
|
||||
# 插件图标
|
||||
plugin_icon = "https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/icons/subscribe_reminder.png"
|
||||
# 插件版本
|
||||
plugin_version = "1.0"
|
||||
# 插件作者
|
||||
plugin_author = "thsrite"
|
||||
# 作者主页
|
||||
author_url = "https://github.com/thsrite"
|
||||
# 插件配置项ID前缀
|
||||
plugin_config_prefix = "subscribereminder_"
|
||||
# 加载顺序
|
||||
plugin_order = 33
|
||||
# 可使用的用户级别
|
||||
auth_level = 1
|
||||
|
||||
# 私有属性
|
||||
_enabled: bool = False
|
||||
_onlyonce: bool = False
|
||||
_time = None
|
||||
tmdb = None
|
||||
media = None
|
||||
subscribe_oper = None
|
||||
_scheduler: Optional[BackgroundScheduler] = None
|
||||
|
||||
def init_plugin(self, config: dict = None):
|
||||
self.subscribe_oper = SubscribeOper()
|
||||
self.tmdb = TmdbChain()
|
||||
self.media = MediaChain()
|
||||
|
||||
# 停止现有任务
|
||||
self.stop_service()
|
||||
|
||||
if config:
|
||||
self._enabled = config.get("enabled")
|
||||
self._onlyonce = config.get("onlyonce")
|
||||
self._time = config.get("time")
|
||||
|
||||
if self._enabled or self._onlyonce:
|
||||
# 周期运行
|
||||
self._scheduler = BackgroundScheduler(timezone=settings.TZ)
|
||||
|
||||
if self._time and str(self._time).isdigit():
|
||||
cron = f"0 {int(self._time)} * * *"
|
||||
try:
|
||||
self._scheduler.add_job(func=self.__send_notify,
|
||||
trigger=CronTrigger.from_crontab(cron),
|
||||
name="订阅提醒")
|
||||
except Exception as err:
|
||||
logger.error(f"定时任务配置错误：{err}")
|
||||
# 推送实时消息
|
||||
self.systemmessage.put(f"执行周期配置错误：{err}")
|
||||
|
||||
# 立即运行一次
|
||||
if self._onlyonce:
|
||||
logger.info(f"订阅提醒服务启动，立即运行一次")
|
||||
self._scheduler.add_job(self.__send_notify, 'date',
|
||||
run_date=datetime.now(
|
||||
tz=pytz.timezone(settings.TZ)) + timedelta(seconds=3),
|
||||
name="订阅提醒")
|
||||
# 关闭一次性开关
|
||||
self._onlyonce = False
|
||||
|
||||
# 保存配置
|
||||
self.__update_config()
|
||||
|
||||
# 启动任务
|
||||
if self._scheduler.get_jobs():
|
||||
self._scheduler.print_jobs()
|
||||
self._scheduler.start()
|
||||
|
||||
def __update_config(self):
|
||||
self.update_config({
|
||||
"enabled": self._enabled,
|
||||
"onlyonce": self._onlyonce,
|
||||
"time": self._time
|
||||
})
|
||||
|
||||
def __send_notify(self):
|
||||
# 查询所有订阅
|
||||
subscribes = self.subscribe_oper.list()
|
||||
if not subscribes:
|
||||
logger.error("当前没有订阅，跳过处理")
|
||||
return
|
||||
|
||||
# 当前日期
|
||||
current_date = datetime.now().date().strftime("%Y-%m-%d")
|
||||
|
||||
current_tv_subscribe = []
|
||||
current_movie_subscribe = []
|
||||
# 遍历订阅，查询tmdb
|
||||
for subscribe in subscribes:
|
||||
# 电视剧
|
||||
if subscribe.type == "电视剧":
|
||||
if not subscribe.tmdbid or not subscribe.season:
|
||||
continue
|
||||
|
||||
# 电视剧某季所有集
|
||||
episodes_info = self.tmdb.tmdb_episodes(tmdbid=subscribe.tmdbid, season=subscribe.season)
|
||||
if not episodes_info:
|
||||
continue
|
||||
|
||||
episodes = []
|
||||
# 遍历集，筛选当前日期发布的剧集
|
||||
for episode in episodes_info:
|
||||
if episode and episode.air_date and str(episode.air_date) == current_date:
|
||||
episodes.append(episode.episode_number)
|
||||
|
||||
if episodes:
|
||||
current_tv_subscribe.append({
|
||||
'name': f"{subscribe.name} ({subscribe.year})",
|
||||
'season': f"S{str(subscribe.season).rjust(2, '0')}",
|
||||
'episode': f"E{str(episodes[0]).rjust(2, '0')}-E{str(episodes[-1]).rjust(2, '0')}" if len(
|
||||
episodes) > 1 else f"E{str(episodes[0]).rjust(2, '0')}"
|
||||
})
|
||||
|
||||
# 电影
|
||||
else:
|
||||
if not subscribe.tmdbid:
|
||||
continue
|
||||
mediainfo = self.media.recognize_media(tmdbid=subscribe.tmdbid, mtype=MediaType.MOVIE)
|
||||
if not mediainfo:
|
||||
continue
|
||||
if str(mediainfo.release_date) == current_date:
|
||||
current_movie_subscribe.append({
|
||||
'name': f"{subscribe.name} ({subscribe.year})"
|
||||
})
|
||||
|
||||
# 如当前日期匹配到订阅，则发送通知
|
||||
text = ""
|
||||
for sub in current_tv_subscribe:
|
||||
text += sub.get("name") + "\n"
|
||||
text += sub.get("season") + sub.get("episode") + "\n"
|
||||
text += "\n"
|
||||
|
||||
for sub in current_movie_subscribe:
|
||||
text += sub.get("name") + "\n"
|
||||
text += "\n"
|
||||
|
||||
if text:
|
||||
self.post_message(mtype=NotificationType.Subscribe,
|
||||
title=f"{current_date}订阅提醒",
|
||||
text=text)
|
||||
|
||||
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、数据结构
|
||||
"""
|
||||
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,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VTextField',
|
||||
'props': {
|
||||
'model': 'time',
|
||||
'label': '时间',
|
||||
'placeholder': '默认9点'
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VRow',
|
||||
'content': [
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VAlert',
|
||||
'props': {
|
||||
'type': 'info',
|
||||
'variant': 'tonal',
|
||||
'text': '默认每天9点推送，需开启（订阅）通知类型。'
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
], {
|
||||
"enabled": False,
|
||||
"onlyonce": False,
|
||||
"time": 9,
|
||||
}
|
||||
|
||||
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))
|
||||
5
plugins_record/SubscribeReminder.md
Normal file
5
plugins_record/SubscribeReminder.md
Normal file
@@ -0,0 +1,5 @@
|
||||
# 订阅提醒
|
||||
|
||||
### 更新记录
|
||||
|
||||
- 1.0 推送当天订阅更新内容
|
||||
Reference in New Issue
Block a user