mirror of
https://github.com/thsrite/MoviePilot-Plugins.git
synced 2026-05-16 23:16:44 +00:00
493 lines
21 KiB
Python
493 lines
21 KiB
Python
import json
|
||
|
||
from app.db.downloadhistory_oper import DownloadHistoryOper
|
||
from app.db.site_oper import SiteOper
|
||
from app.plugins import _PluginBase
|
||
from app.db.subscribe_oper import SubscribeOper
|
||
from typing import Any, List, Dict, Tuple
|
||
|
||
from app.schemas.types import SystemConfigKey
|
||
|
||
|
||
class SubscribeStatistic(_PluginBase):
|
||
# 插件名称
|
||
plugin_name = "订阅下载统计"
|
||
# 插件描述
|
||
plugin_desc = "统计指定时间内各站点订阅及下载情况。"
|
||
# 插件图标
|
||
plugin_icon = "https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/icons/subscribestatistic.png"
|
||
# 插件版本
|
||
plugin_version = "1.4"
|
||
# 插件作者
|
||
plugin_author = "thsrite"
|
||
# 作者主页
|
||
author_url = "https://github.com/thsrite"
|
||
# 插件配置项ID前缀
|
||
plugin_config_prefix = "subscribestatistic_"
|
||
# 加载顺序
|
||
plugin_order = 28
|
||
# 可使用的用户级别
|
||
auth_level = 1
|
||
|
||
# 任务执行间隔
|
||
_enabled = False
|
||
_movie_subscribe_days = None
|
||
_tv_subscribe_days = None
|
||
_movie_download_days = None
|
||
_tv_download_days = None
|
||
subscribe = None
|
||
downloadhis = None
|
||
siteoper = None
|
||
|
||
def init_plugin(self, config: dict = None):
|
||
self.subscribe = SubscribeOper()
|
||
self.downloadhis = DownloadHistoryOper()
|
||
self.siteoper = SiteOper()
|
||
if config:
|
||
self._enabled = config.get("enabled")
|
||
self._movie_subscribe_days = config.get("movie_subscribe_days")
|
||
self._tv_subscribe_days = config.get("tv_subscribe_days")
|
||
self._movie_download_days = config.get("movie_download_days")
|
||
self._tv_download_days = config.get("tv_download_days")
|
||
|
||
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': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'enabled',
|
||
'label': '启用插件',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
# {
|
||
# 'component': 'VCol',
|
||
# 'props': {
|
||
# 'cols': 12,
|
||
# 'md': 4
|
||
# },
|
||
# 'content': [
|
||
# {
|
||
# 'component': 'VSwitch',
|
||
# 'props': {
|
||
# 'model': 'notify',
|
||
# 'label': '发送通知',
|
||
# }
|
||
# }
|
||
# ]
|
||
# },
|
||
# {
|
||
# 'component': 'VCol',
|
||
# 'props': {
|
||
# 'cols': 12,
|
||
# 'md': 4
|
||
# },
|
||
# 'content': [
|
||
# {
|
||
# 'component': 'VSwitch',
|
||
# 'props': {
|
||
# 'model': 'onlyonce',
|
||
# 'label': '立即运行一次',
|
||
# }
|
||
# }
|
||
# ]
|
||
# }
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'movie_subscribe_days',
|
||
'label': '电影订阅天数',
|
||
'placeholder': '30'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'tv_subscribe_days',
|
||
'label': '电视剧订阅天数',
|
||
'placeholder': '30'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'movie_download_days',
|
||
'label': '电影下载天数',
|
||
'placeholder': '7'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 3
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'tv_download_days',
|
||
'label': '电视剧下载天数',
|
||
'placeholder': '7'
|
||
}
|
||
}
|
||
]
|
||
},
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '订阅数量：MoviePilot指定天数内正在订阅的数量。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '下载数量：通过MoviePilot下载的数量，包括订阅下载、手动下载以及其他下载等场景。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
], {
|
||
"enabled": False,
|
||
"movie_subscribe_days": 30,
|
||
"tv_subscribe_days": 30,
|
||
"movie_download_days": 7,
|
||
"tv_download_days": 7
|
||
}
|
||
|
||
def get_page(self) -> List[dict]:
|
||
if not self._enabled:
|
||
return [
|
||
{
|
||
'component': 'div',
|
||
'text': '暂未开启插件',
|
||
'props': {
|
||
'class': 'text-center',
|
||
}
|
||
}
|
||
]
|
||
|
||
# 电影订阅
|
||
movie_subscribes = self.subscribe.list_by_type(mtype='电影', days=self._movie_subscribe_days)
|
||
movie_subscribe_sites = []
|
||
movie_subscribe_datas = []
|
||
if movie_subscribes:
|
||
movie_subscribe_site_ids = []
|
||
for movie_subscribe in movie_subscribes:
|
||
if movie_subscribe.sites:
|
||
movie_subscribe_site_ids += [site for site in json.loads(movie_subscribe.sites)]
|
||
else:
|
||
movie_subscribe_site_ids += self.systemconfig.get(SystemConfigKey.RssSites) or []
|
||
|
||
for movie_subscribe_site_id in movie_subscribe_site_ids:
|
||
site = self.siteoper.get(movie_subscribe_site_id)
|
||
if site:
|
||
if not movie_subscribe_sites.__contains__(site.name):
|
||
movie_subscribe_sites.append(site.name)
|
||
movie_subscribe_datas.append(movie_subscribe_site_ids.count(movie_subscribe_site_id))
|
||
|
||
# 电视剧订阅
|
||
tv_subscribes = self.subscribe.list_by_type(mtype='电视剧', days=self._tv_subscribe_days)
|
||
tv_subscribe_sites = []
|
||
tv_subscribe_datas = []
|
||
if tv_subscribes:
|
||
tv_subscribe_site_ids = []
|
||
for tv_subscribe in tv_subscribes:
|
||
if tv_subscribe.sites:
|
||
tv_subscribe_site_ids += [site for site in json.loads(tv_subscribe.sites)]
|
||
else:
|
||
tv_subscribe_site_ids += self.systemconfig.get(SystemConfigKey.RssSites) or []
|
||
|
||
for tv_subscribe_site_id in tv_subscribe_site_ids:
|
||
site = self.siteoper.get(tv_subscribe_site_id)
|
||
if site:
|
||
if not tv_subscribe_sites.__contains__(site.name):
|
||
tv_subscribe_sites.append(site.name)
|
||
tv_subscribe_datas.append(tv_subscribe_site_ids.count(tv_subscribe_site_id))
|
||
|
||
# 电影下载
|
||
movie_downloads = self.downloadhis.list_by_type(mtype="电影", days=self._movie_download_days)
|
||
movie_download_sites = []
|
||
movie_download_datas = []
|
||
if movie_downloads:
|
||
movie_download_sites2 = []
|
||
for movie_download in movie_downloads:
|
||
if movie_download.torrent_site:
|
||
movie_download_sites2.append(movie_download.torrent_site)
|
||
|
||
for movie_download_site in movie_download_sites2:
|
||
if not movie_download_sites.__contains__(movie_download_site):
|
||
movie_download_sites.append(movie_download_site)
|
||
if not movie_download_datas.__contains__(movie_download_site):
|
||
movie_download_datas.append(movie_download_sites2.count(movie_download_site))
|
||
|
||
# 电视剧下载
|
||
tv_downloads = self.downloadhis.list_by_type(mtype="电视剧", days=self._tv_download_days)
|
||
tv_download_sites = []
|
||
tv_download_datas = []
|
||
if tv_downloads:
|
||
tv_download_sites2 = []
|
||
for tv_download in tv_downloads:
|
||
if tv_download.torrent_site:
|
||
tv_download_sites2.append(tv_download.torrent_site)
|
||
|
||
for tv_download_site in tv_download_sites2:
|
||
if not tv_download_sites.__contains__(tv_download_site):
|
||
tv_download_sites.append(tv_download_site)
|
||
if not tv_download_datas.__contains__(tv_download_site):
|
||
tv_download_datas.append(tv_download_sites2.count(tv_download_site))
|
||
|
||
# 拼装页面
|
||
return [
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
# 电影订阅图表
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VApexChart',
|
||
'props': {
|
||
'height': 300,
|
||
'options': {
|
||
'chart': {
|
||
'type': 'pie',
|
||
},
|
||
'labels': movie_subscribe_sites,
|
||
'title': {
|
||
'text': f'电影近 {self._movie_subscribe_days} 天订阅 {len(movie_subscribes)} 部'
|
||
},
|
||
'legend': {
|
||
'show': True
|
||
},
|
||
'plotOptions': {
|
||
'pie': {
|
||
'expandOnClick': False
|
||
}
|
||
},
|
||
'noData': {
|
||
'text': '订阅未选择站点或站点已删除'
|
||
}
|
||
},
|
||
'series': movie_subscribe_datas
|
||
}
|
||
}
|
||
]
|
||
},
|
||
# 电视剧订阅图表
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VApexChart',
|
||
'props': {
|
||
'height': 300,
|
||
'options': {
|
||
'chart': {
|
||
'type': 'pie',
|
||
},
|
||
'labels': tv_subscribe_sites,
|
||
'title': {
|
||
'text': f'电视剧近 {self._tv_subscribe_days} 天订阅 {len(tv_subscribes)} 部'
|
||
},
|
||
'legend': {
|
||
'show': True
|
||
},
|
||
'plotOptions': {
|
||
'pie': {
|
||
'expandOnClick': False
|
||
}
|
||
},
|
||
'noData': {
|
||
'text': '订阅未选择站点或站点已删除'
|
||
}
|
||
},
|
||
'series': tv_subscribe_datas
|
||
}
|
||
}
|
||
]
|
||
},
|
||
# 电影下载图表
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VApexChart',
|
||
'props': {
|
||
'height': 300,
|
||
'options': {
|
||
'chart': {
|
||
'type': 'pie',
|
||
},
|
||
'labels': movie_download_sites,
|
||
'title': {
|
||
'text': f'电影近 {self._movie_download_days} 天下载 {len(movie_downloads)} 个种子'
|
||
},
|
||
'legend': {
|
||
'show': True
|
||
},
|
||
'plotOptions': {
|
||
'pie': {
|
||
'expandOnClick': False
|
||
}
|
||
},
|
||
'noData': {
|
||
'text': '暂无数据'
|
||
}
|
||
},
|
||
'series': movie_download_datas
|
||
}
|
||
}
|
||
]
|
||
},
|
||
# 电视剧下载图表
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VApexChart',
|
||
'props': {
|
||
'height': 300,
|
||
'options': {
|
||
'chart': {
|
||
'type': 'pie',
|
||
},
|
||
'labels': tv_download_sites,
|
||
'title': {
|
||
'text': f'电视剧近 {self._tv_download_days} 天下载 {len(tv_downloads)} 个种子'
|
||
},
|
||
'legend': {
|
||
'show': True
|
||
},
|
||
'plotOptions': {
|
||
'pie': {
|
||
'expandOnClick': False
|
||
}
|
||
},
|
||
'noData': {
|
||
'text': '暂无数据'
|
||
}
|
||
},
|
||
'series': tv_download_datas
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
]
|
||
|
||
def stop_service(self):
|
||
"""
|
||
退出插件
|
||
"""
|
||
pass
|