feat 豆瓣明星热映订阅插件

This commit is contained in:
thsrite
2024-01-02 10:17:23 +08:00
parent a6acad3077
commit c4edc5d344
4 changed files with 378 additions and 0 deletions

View File

@@ -23,4 +23,5 @@ MoviePilot三方插件市场https://github.com/thsrite/MoviePilot-Plugins/
- [日程提醒 1.0](docs%2FScheduleReminder.md)
- [订阅提醒 1.1](docs%2FSubscribeReminder.md)
- [Emby观影报告 1.4](docs%2FEmbyReporter.md)
- [豆瓣明星热映订阅 1.0](docs%2FActorSubscribe.md)

5
docs/ActorSubscribe.md Normal file
View File

@@ -0,0 +1,5 @@
# 豆瓣明星热映订阅
### 更新记录
- 1.0 自动订阅豆瓣明星最新电影

View File

@@ -110,5 +110,13 @@
"icon": "Pydiocells_A.png",
"author": "thsrite",
"level": 1
},
"ActorSubscribe": {
"name": "豆瓣明星热映订阅",
"description": "自动订阅豆瓣明星最新电影。",
"version": "1.0",
"icon": "Mdcng_A.png",
"author": "thsrite",
"level": 1
}
}

View File

@@ -0,0 +1,364 @@
from datetime import datetime, timedelta
import pytz
from app.chain.douban import DoubanChain
from app.chain.subscribe import SubscribeChain
from app.core.config import settings
from app.core.context import MediaInfo
from app.plugins import _PluginBase
from typing import Any, List, Dict, Tuple
from app.log import logger
from apscheduler.schedulers.background import BackgroundScheduler
from apscheduler.triggers.cron import CronTrigger
class ActorSubscribe(_PluginBase):
# 插件名称
plugin_name = "豆瓣明星热映订阅"
# 插件描述
plugin_desc = "自动订阅豆瓣明星最新电影。"
# 插件图标
plugin_icon = "Mdcng_A.png"
# 插件版本
plugin_version = "1.0"
# 插件作者
plugin_author = "thsrite"
# 作者主页
author_url = "https://github.com/thsrite"
# 插件配置项ID前缀
plugin_config_prefix = "actorsubscribe_"
# 加载顺序
plugin_order = 25
# 可使用的用户级别
auth_level = 1
# 私有属性
_enabled: bool = False
_onlyonce: bool = False
_cron: str = ""
_actors = None
subscribechain = None
def init_plugin(self, config: dict = None):
self.subscribechain = SubscribeChain()
# 停止现有任务
self.stop_service()
if config:
self._enabled = config.get("enabled")
self._onlyonce = config.get("onlyonce")
self._cron = config.get("cron")
self._actors = config.get("actors")
if self._enabled or self._onlyonce:
# 定时服务
self._scheduler = BackgroundScheduler(timezone=settings.TZ)
# 立即运行一次
if self._onlyonce:
logger.info(f"明星热映订阅服务启动,立即运行一次")
self._scheduler.add_job(self.__actor_subscribe, 'date',
run_date=datetime.now(
tz=pytz.timezone(settings.TZ)) + timedelta(seconds=3),
name="明星热映订阅")
# 关闭一次性开关
self._onlyonce = False
# 保存配置
self.__update_config()
# 周期运行
if self._cron:
try:
self._scheduler.add_job(func=self.__actor_subscribe,
trigger=CronTrigger.from_crontab(self._cron),
name="明星热映订阅")
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 __actor_subscribe(self):
"""
明星热映订阅
"""
if not self._actors:
logger.warn("暂无订阅明星,停止运行")
return
history: List[dict] = self.get_data('history') or []
movies = DoubanChain().movie_showing(page=1, count=100)
if not movies:
return []
medias = [MediaInfo(douban_info=movie) for movie in movies]
logger.info(f"获取到豆瓣正在热映 {len(medias)}")
# 检查订阅
actors = str(self._actors).split(",")
for mediainfo in medias:
if mediainfo.actors:
for actor in mediainfo.actors:
if actors.__contains__(actor):
# 开始订阅
logger.info(f"电影 {mediainfo.title_year} 命中订阅演员 {actor},开始订阅")
# 判断用户是否已经添加订阅
if self.subscribechain.exists(mediainfo=mediainfo):
logger.info(f'{mediainfo.title_year} 订阅已存在')
continue
# 添加订阅
self.subscribechain.add(title=mediainfo.title,
year=mediainfo.year,
mtype=mediainfo.type,
tmdbid=mediainfo.tmdb_id,
exist_ok=True,
username="豆瓣榜单")
# 存储历史记录
history.append({
"title": mediainfo.title,
"type": mediainfo.type.value,
"year": mediainfo.year,
"poster": mediainfo.get_poster_image(),
"overview": mediainfo.overview,
"tmdbid": mediainfo.tmdb_id,
"doubanid": mediainfo.douban_id,
"time": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
})
def __update_config(self):
self.update_config({
"enabled": self._enabled,
"onlyonce": self._onlyonce,
"cron": self._cron,
"actors": self._actors
})
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,
'md': 6
},
'content': [
{
'component': 'VTextField',
'props': {
'model': 'cron',
'label': '执行周期',
'placeholder': '5位cron表达式留空自动'
}
}
]
},
]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 6
},
'content': [
{
'component': 'VTextField',
'props': {
'model': 'actors',
'label': '明星',
'placeholder': '多个英文逗号分割'
}
}
]
},
]
},
]
}
], {
"enabled": False,
"onlyonce": False,
"cron": "5 1 * * *",
"actors": "",
}
def get_page(self) -> List[dict]:
"""
拼装插件详情页面,需要返回页面配置,同时附带数据
"""
# 查询历史记录
historys = self.get_data('history')
if not historys:
return [
{
'component': 'div',
'text': '暂无数据',
'props': {
'class': 'text-center',
}
}
]
# 数据按时间降序排序
historys = sorted(historys, key=lambda x: x.get('time'), reverse=True)
# 拼装页面
contents = []
for history in historys:
title = history.get("title")
poster = history.get("poster")
mtype = history.get("type")
time_str = history.get("time")
doubanid = history.get("doubanid")
contents.append(
{
'component': 'VCard',
'content': [
{
'component': 'div',
'props': {
'class': 'd-flex justify-space-start flex-nowrap flex-row',
},
'content': [
{
'component': 'div',
'content': [
{
'component': 'VImg',
'props': {
'src': poster,
'height': 120,
'width': 80,
'aspect-ratio': '2/3',
'class': 'object-cover shadow ring-gray-500',
'cover': True
}
}
]
},
{
'component': 'div',
'content': [
{
'component': 'VCardSubtitle',
'props': {
'class': 'pa-2 font-bold break-words whitespace-break-spaces'
},
'content': [
{
'component': 'a',
'props': {
'href': f"https://movie.douban.com/subject/{doubanid}",
'target': '_blank'
},
'text': title
}
]
},
{
'component': 'VCardText',
'props': {
'class': 'pa-0 px-2'
},
'text': f'类型:{mtype}'
},
{
'component': 'VCardText',
'props': {
'class': 'pa-0 px-2'
},
'text': f'时间:{time_str}'
}
]
}
]
}
]
}
)
return [
{
'component': 'div',
'props': {
'class': 'grid gap-3 grid-info-card',
},
'content': contents
}
]
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))