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, Optional 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.1" # 插件作者 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 _scheduler: Optional[BackgroundScheduler] = None _quality = None _resolution = None _effect = None _clear = False # 质量选择框数据 _qualityOptions = { '全部': '', '蓝光原盘': 'Blu-?Ray.+VC-?1|Blu-?Ray.+AVC|UHD.+blu-?ray.+HEVC|MiniBD', 'Remux': 'Remux', 'BluRay': 'Blu-?Ray', 'UHD': 'UHD|UltraHD', 'WEB-DL': 'WEB-?DL|WEB-?RIP', 'HDTV': 'HDTV', 'H265': '[Hx].?265|HEVC', 'H264': '[Hx].?264|AVC' } # 分辨率选择框数据 _resolutionOptions = { '全部': '', '4k': '4K|2160p|x2160', '1080p': '1080[pi]|x1080', '720p': '720[pi]|x720' } # 特效选择框数据 _effectOptions = { '全部': '', '杜比视界': 'Dolby[\\s.]+Vision|DOVI|[\\s.]+DV[\\s.]+', '杜比全景声': 'Dolby[\\s.]*\\+?Atmos|Atmos', 'HDR': '[\\s.]+HDR[\\s.]+|HDR10|HDR10\\+', 'SDR': '[\\s.]+SDR[\\s.]+', } 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") self._quality = config.get("quality") self._resolution = config.get("resolution") self._effect = config.get("effect") self._clear = config.get("_clear") 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 if self._clear: history: List[dict] = [] already_handle: List[dict] = [] else: history: List[dict] = self.get_data('history') or [] already_handle: List[dict] = self.get_data('already_handle') 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.title_year in already_handle: logger.info(f"电影 {mediainfo.title_year} 已被处理，跳过") already_handle.append(mediainfo.title_year) logger.info(f"开始处理电影 {mediainfo.title_year}") if mediainfo.actors: for actor in mediainfo.actors: if actor in actors: # 开始订阅 logger.info(f"电影 {mediainfo.title_year} {mediainfo.tmdb_id} 命中订阅演员 {actor}，开始订阅") # 判断用户是否已经添加订阅 if self.subscribechain.exists(mediainfo=mediainfo): logger.info(f'{mediainfo.title_year} 订阅已存在') continue else: # 添加订阅 self.subscribechain.add(title=mediainfo.title, year=mediainfo.year, mtype=mediainfo.type, tmdbid=mediainfo.tmdb_id, doubanid=mediainfo.douban_id, exist_ok=True, quality=self._quality, resolution=self._resolution, effect=self._effect, username=settings.SUPERUSER) # 存储历史记录 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") }) # 保存历史记录 self.save_data('history', history) self.save_data('already_handle', already_handle) self._clear = False self.__update_config() def __update_config(self): self.update_config({ "enabled": self._enabled, "onlyonce": self._onlyonce, "cron": self._cron, "actors": self._actors, "quality": self._quality, "resolution": self._resolution, "effect": self._effect, "clear": self._clear, }) 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、数据结构 """ qualityOptions = [{"title": i, "value": self._qualityOptions.get(i)} for i in self._qualityOptions.keys()] resolutionOptions = [{"title": i, "value": self._resolutionOptions.get(i)} for i in self._resolutionOptions.keys()] effectOptions = [{"title": i, "value": self._effectOptions.get(i)} for i in self._effectOptions.keys()] 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': 'actors', 'label': '明星', 'placeholder': '多个英文逗号分割' } } ] }, ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 4 }, 'content': [ { 'component': 'VSelect', 'props': { 'multiple': False, 'chips': True, 'model': 'quality', 'label': '质量', 'items': qualityOptions } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 4 }, 'content': [ { 'component': 'VSelect', 'props': { 'multiple': False, 'chips': True, 'model': 'resolution', 'label': '分辨率', 'items': resolutionOptions } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 4 }, 'content': [ { 'component': 'VSelect', 'props': { 'multiple': False, 'chips': True, 'model': 'effect', 'label': '特效', 'items': effectOptions } } ] }, ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'clear', 'label': '清理历史记录', } } ] } ] } ] } ], { "enabled": False, "onlyonce": False, "cron": "5 1 * * *", "actors": "", "quality": "", "resolution": "", "effect": "", "clear": False } 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))