Files
archived-MoviePilot-Plugins/plugins/actorsubscribe/__init__.py
2024-04-09 11:56:26 +08:00

809 lines
33 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
import time
from datetime import datetime, timedelta
import pytz
from app.chain.douban import DoubanChain
from app.chain.tmdb import TmdbChain
from app.chain.download import DownloadChain
from app.chain.subscribe import SubscribeChain
from app.core.config import settings
from app.core.context import MediaInfo
from app.core.metainfo import MetaInfo
from app.schemas import MediaType
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.4"
# 插件作者
plugin_author = "thsrite"
# 作者主页
author_url = "https://github.com/thsrite"
# 插件配置项ID前缀
plugin_config_prefix = "actorsubscribe_"
# 加载顺序
plugin_order = 25
# 可使用的用户级别
auth_level = 2
# 私有属性
_enabled: bool = False
_onlyonce: bool = False
_cron: str = ""
_actors = None
subscribechain = None
downloadchain = None
_scheduler: Optional[BackgroundScheduler] = None
_quality = None
_resolution = None
_effect = None
_clear = False
_clear_already_handle = False
_source = "showing"
# 质量选择框数据
_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.downloadchain = DownloadChain()
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")
self._clear_already_handle = config.get("clear_already_handle")
self._source = config.get("source")
# 清理插件订阅历史
if self._clear:
self.del_data(key="history")
self._clear = False
self.__update_config()
logger.info("订阅历史清理完成")
# 清理已处理历史
if self._clear_already_handle:
self.del_data(key="already_handle")
self._clear_already_handle = False
self.__update_config()
logger.info("已处理历史清理完成")
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 []
already_handle: List[dict] = self.get_data('already_handle') or []
medias = []
for source in self._source.split("\n"):
if source.strip() == "douban_showing":
medias += self.__douban_movie_showing()
elif source.strip() == "douban_movies":
medias += self.__douban_movies()
elif source.strip() == "douban_tvs":
medias += self.__douban_tvs()
elif source.strip() == "douban_movie_top250":
medias += self.__douban_movie_top250()
elif source.strip() == "douban_tv_weekly_chinese":
medias += self.__douban_tv_weekly_chinese()
elif source.strip() == "douban_tv_weekly_global":
medias += self.__douban_tv_weekly_global()
elif source.strip() == "douban_tv_animation":
medias += self.__douban_tv_animation()
elif source.strip() == "douban_movie_hot":
medias += self.__douban_movie_hot()
elif source.strip() == "douban_tv_hot":
medias += self.__douban_tv_hot()
elif source.strip() == "tmdb_movies":
medias += self.__tmdb_movies()
elif source.strip() == "tmdb_tvs":
medias += self.__tmdb_tvs()
elif source.strip() == "tmdb_trending":
medias += self.__tmdb_trending()
else:
logger.warn(f"未知的订阅源:{source}")
# 检查订阅
actors = str(self._actors).split(",")
for mediainfo in medias:
if mediainfo.title_year in already_handle:
logger.info(f"{mediainfo.type.name} {mediainfo.title_year} 已被处理,跳过")
continue
already_handle.append(mediainfo.title_year)
logger.info(f"开始处理电影 {mediainfo.title_year}")
# 元数据
meta = MetaInfo(mediainfo.title)
# 演员中文名
if mediainfo.actors or mediainfo.directors:
mediainfo_actiors = mediainfo.actors + mediainfo.directors
else:
# 查询豆瓣中文演员名
mediainfo_actiors = self.__get_douban_actors(mediainfo)
if not mediainfo_actiors:
logger.warn(f'未识别到演员信息,标题:{mediainfo.title}tmdbid{mediainfo.tmdb_id}')
continue
# 判断有无tmdbid
if not mediainfo.tmdb_id:
oldmediainfo = mediainfo
# 主要获取tmdbid
mediainfo = self.chain.recognize_media(meta=meta, doubanid=mediainfo.douban_id)
if not mediainfo:
logger.warn(f'未识别到媒体信息,标题:{oldmediainfo.title}豆瓣ID{oldmediainfo.douban_id}')
continue
# 查询缺失的媒体信息
exist_flag, _ = self.downloadchain.get_no_exists_info(meta=meta, mediainfo=mediainfo)
if exist_flag:
logger.info(f'{mediainfo.title_year} 媒体库中已存在')
continue
# 判断用户是否已经添加订阅
if self.subscribechain.exists(mediainfo=mediainfo):
logger.info(f'{mediainfo.title_year} 订阅已存在')
continue
if mediainfo_actiors:
is_subscribe = False
for actor in mediainfo_actiors:
# logger.info(f'正在处理 {mediainfo.title_year} 演员 {actor}')
if actor and actor in actors:
# 开始订阅
logger.info(
f"{mediainfo.type.name} {mediainfo.title_year} {mediainfo.tmdb_id} 命中订阅演员 {actor},开始订阅")
is_subscribe = True
# 添加订阅
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")
})
if not is_subscribe:
logger.info(f"{mediainfo.type.name} {mediainfo.title_year} {mediainfo.tmdb_id} 未命中订阅演员,跳过")
# 保存历史记录
self.save_data('history', history)
self.save_data('already_handle', already_handle)
def __get_douban_actors(self, mediainfo: MediaInfo, season: int = None) -> List[dict]:
"""
获取豆瓣演员信息
"""
# 随机休眠 3-10 秒
sleep_time = 3 + int(time.time()) % 7
logger.debug(f"随机休眠 {sleep_time}秒 ...")
time.sleep(sleep_time)
# 匹配豆瓣信息
doubaninfo = DoubanChain().match_doubaninfo(name=mediainfo.title,
imdbid=mediainfo.imdb_id,
mtype=mediainfo.type,
year=mediainfo.year,
season=season)
# 豆瓣演员
if doubaninfo:
doubanitem = DoubanChain().douban_info(doubaninfo.get("id")) or {}
actors = (doubanitem.get("actors") or []) + (doubanitem.get("directors") or [])
return [actor.get("name") for actor in actors]
else:
logger.debug(f"未找到豆瓣信息:{mediainfo.title_year}")
return []
def __douban_movie_showing(self):
"""
豆瓣正在热映
"""
movies = DoubanChain().movie_showing(page=1, count=30)
if not movies:
return []
medias = [MediaInfo(douban_info=movie) for movie in movies]
logger.info(f"获取到豆瓣正在热映 {len(medias)}")
return medias
def __douban_movies(self):
"""
豆瓣电影
"""
movies = DoubanChain().douban_discover(mtype=MediaType.MOVIE,
sort="R", tags="", page=1, count=30)
if not movies:
return []
medias = [MediaInfo(douban_info=movie) for movie in movies]
logger.info(f"获取到豆瓣电影 {len(medias)}")
return medias
def __douban_tvs(self):
"""
豆瓣剧集
"""
tvs = DoubanChain().douban_discover(mtype=MediaType.TV,
sort="R", tags="", page=1, count=30)
if not tvs:
return []
medias = [MediaInfo(douban_info=tv) for tv in tvs]
logger.info(f"获取到豆瓣剧集 {len(medias)}")
return medias
def __douban_movie_top250(self):
"""
豆瓣电影TOP250
"""
movies = DoubanChain().movie_top250(mtype=MediaType.MOVIE, page=1, count=30)
if not movies:
return []
medias = [MediaInfo(douban_info=movie) for movie in movies]
logger.info(f"获取到豆瓣电影TOP250 {len(medias)}")
return medias
def __douban_tv_weekly_chinese(self):
"""
豆瓣国产剧集周榜
"""
tvs = DoubanChain().tv_weekly_chinese(page=1, count=30)
if not tvs:
return []
medias = [MediaInfo(douban_info=tv) for tv in tvs]
logger.info(f"获取到豆瓣国产剧集周榜 {len(medias)}")
return medias
def __douban_tv_weekly_global(self):
"""
全球每周剧集口碑榜
"""
tvs = DoubanChain().tv_weekly_global(page=1, count=30)
if not tvs:
return []
medias = [MediaInfo(douban_info=tv) for tv in tvs]
logger.info(f"获取到全球每周剧集口碑榜 {len(medias)}")
return medias
def __douban_tv_animation(self):
"""
豆瓣动画剧集
"""
tvs = DoubanChain().tv_animation(page=1, count=30)
if not tvs:
return []
medias = [MediaInfo(douban_info=tv) for tv in tvs]
logger.info(f"获取到豆瓣动画剧集 {len(medias)}")
return medias
def __douban_movie_hot(self):
"""
豆瓣热门电影
"""
movies = DoubanChain().movie_hot(page=1, count=30)
if not movies:
return []
medias = [MediaInfo(douban_info=movie) for movie in movies]
logger.info(f"获取到豆瓣热门电影 {len(medias)}")
return medias
def __douban_tv_hot(self):
"""
豆瓣热门电视剧
"""
tvs = DoubanChain().tv_hot(page=1, count=30)
if not tvs:
return []
medias = [MediaInfo(douban_info=tv) for tv in tvs]
logger.info(f"获取到豆瓣热门电视剧 {len(medias)}")
return medias
def __tmdb_movies(self):
"""
TMDB电影
"""
movies = TmdbChain().tmdb_discover(mtype=MediaType.MOVIE,
sort_by="popularity.desc",
with_genres="",
with_original_language="",
page=1)
if not movies:
return []
medias = [MediaInfo(tmdb_info=movie) for movie in movies]
logger.info(f"获取到TMDB电影 {len(medias)}")
return medias
def __tmdb_tvs(self):
"""
TMDB剧集
"""
tvs = TmdbChain().tmdb_discover(mtype=MediaType.TV,
sort_by="popularity.desc",
with_genres="",
with_original_language="",
page=1)
if not tvs:
return []
medias = [MediaInfo(tmdb_info=tv) for tv in tvs]
logger.info(f"获取到TMDB剧集 {len(medias)}")
return medias
def __tmdb_trending(self):
"""
TMDB流行趋势
"""
tvs = TmdbChain().tmdb_trending(page=1)
if not tvs:
return []
medias = [MediaInfo(tmdb_info=tv) for tv in tvs]
logger.info(f"获取到TMDB流行趋势 {len(medias)}")
return medias
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,
"clear_already_handle": self._clear_already_handle,
"source": self._source,
})
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': 3
},
'content': [
{
'component': 'VSwitch',
'props': {
'model': 'enabled',
'label': '启用插件',
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSwitch',
'props': {
'model': 'onlyonce',
'label': '立即运行一次',
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSwitch',
'props': {
'model': 'clear',
'label': '清理订阅记录',
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSwitch',
'props': {
'model': 'clear_already_handle',
'label': '清理已处理记录',
}
}
]
}
]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VTextField',
'props': {
'model': 'cron',
'label': '执行周期',
'placeholder': '5位cron表达式留空自动'
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 9
},
'content': [
{
'component': 'VTextField',
'props': {
'model': 'actors',
'label': '明星',
'placeholder': '多个英文逗号分割'
}
}
]
},
]
},
{
'component': 'VRow',
'content': [
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSelect',
'props': {
'multiple': True,
'chips': True,
'model': 'source',
'label': '订阅来源',
'items': [
{'title': '豆瓣正在热映', 'value': 'douban_showing'},
{'title': '豆瓣电影', 'value': 'douban_movies'},
{'title': '豆瓣剧集', 'value': 'douban_tvs'},
{'title': '豆瓣电影TOP250', 'value': 'douban_movie_top250'},
{'title': '豆瓣国产剧集周榜', 'value': 'douban_tv_weekly_chinese'},
{'title': '豆瓣全球剧集周榜', 'value': 'douban_tv_weekly_global'},
{'title': '豆瓣动画剧集', 'value': 'douban_tv_animation'},
{'title': '豆瓣热门电影', 'value': 'douban_movie_hot'},
{'title': '豆瓣热门电视剧', 'value': 'douban_tv_hot'},
{'title': 'TMDB电影', 'value': 'tmdb_movies'},
{'title': 'TMDB剧集', 'value': 'tmdb_tvs'},
{'title': 'TMDB流行趋势', 'value': 'tmdb_trending'},
]
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSelect',
'props': {
'multiple': False,
'chips': True,
'model': 'quality',
'label': '质量',
'items': qualityOptions
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSelect',
'props': {
'multiple': False,
'chips': True,
'model': 'resolution',
'label': '分辨率',
'items': resolutionOptions
}
}
]
},
{
'component': 'VCol',
'props': {
'cols': 12,
'md': 3
},
'content': [
{
'component': 'VSelect',
'props': {
'multiple': False,
'chips': True,
'model': 'effect',
'label': '特效',
'items': effectOptions
}
}
]
},
]
},
]
}
], {
"enabled": False,
"onlyonce": False,
"cron": "5 1 * * *",
"actors": "",
"quality": "",
"resolution": "",
"effect": "",
"clear": False,
"clear_already_handle": False,
"source": "douban_showing"
}
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))