mirror of
https://github.com/d0zingcat/MoviePilot-Plugins.git
synced 2026-05-13 15:09:12 +00:00
feat: add subscribe clear plugin
This commit is contained in:
liushaoxiong12
lsx
@@ -378,5 +378,17 @@
|
||||
"history": {
|
||||
"v1.1": "需要MoviePilot v2.2.7-1+ 版本，否则无法显示图片"
|
||||
}
|
||||
},
|
||||
"SubscribeClear": {
|
||||
"name": "订阅种子清理",
|
||||
"description": "删除指定下载信息。",
|
||||
"labels": "下载管理",
|
||||
"version": "1.0",
|
||||
"icon": "Moviepilot_A.jpg",
|
||||
"author": "k0ala",
|
||||
"level": 1,
|
||||
"history": {
|
||||
"v1.0": "支持清理QB中已下载的订阅文件"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
371
plugins.v2/subscribeclear/__init__.py
Normal file
371
plugins.v2/subscribeclear/__init__.py
Normal file
@@ -0,0 +1,371 @@
|
||||
import threading
|
||||
from typing import List, Tuple, Dict, Any, Optional
|
||||
|
||||
from app.log import logger
|
||||
from app.plugins import _PluginBase
|
||||
from app.schemas import ServiceInfo
|
||||
from app.db.downloadhistory_oper import DownloadHistoryOper, DownloadHistory
|
||||
|
||||
lock = threading.Lock()
|
||||
|
||||
|
||||
class SubscribeClear(_PluginBase):
|
||||
# 插件名称
|
||||
plugin_name = "订阅种子清理"
|
||||
# 插件描述
|
||||
plugin_desc = "删除指定下载信息。"
|
||||
# 插件图标
|
||||
plugin_icon = "Moviepilot_A.png"
|
||||
# 插件版本
|
||||
plugin_version = "1.0"
|
||||
# 插件作者
|
||||
plugin_author = "k0ala"
|
||||
# 作者主页
|
||||
author_url = "https://github.com/liushaoxiong10"
|
||||
# 插件配置项ID前缀
|
||||
plugin_config_prefix = "subscribeclear_"
|
||||
# 加载顺序
|
||||
plugin_order = 8
|
||||
# 可使用的用户级别
|
||||
auth_level = 1
|
||||
|
||||
# 私有属性
|
||||
_titles = []
|
||||
_episodes = []
|
||||
|
||||
def init_plugin(self, config: dict = None):
|
||||
if config:
|
||||
self._titles = config.get("titles") or []
|
||||
self._episodes = config.get("episodes") or []
|
||||
|
||||
self.stop_service()
|
||||
self.clear_history(self._titles, self._episodes)
|
||||
config['titles'] = []
|
||||
config['episodes'] = []
|
||||
|
||||
def clear_history(self, titles: List[str], episodes: List[str]):
|
||||
logger.info(f"清除下载历史记录：{titles} {episodes}")
|
||||
data = self.get_data()
|
||||
down_oper = DownloadHistoryOper()
|
||||
downloader_history ={}
|
||||
for d in data:
|
||||
if d.title in titles or d.id in episodes:
|
||||
tmp = downloader_history.get(d.downloader)
|
||||
if not tmp:
|
||||
tmp = []
|
||||
tmp.append(d)
|
||||
downloader_history[d.downloader] = tmp
|
||||
logger.info(f"清除下载历史记录：{d.id} {d.title} {d.seasons} {d.episodes} {d.download_hash}")
|
||||
for downloader, history in downloader_history.items():
|
||||
downloader_obj = self.__get_downloader(downloader)
|
||||
# 获取所有历史记录的hash值列表
|
||||
history_hashes = [h.download_hash for h in history]
|
||||
torrents, error = downloader_obj.get_torrents(ids=history_hashes)
|
||||
if error:
|
||||
logger.error(f"获取种子信息失败： {error}")
|
||||
continue
|
||||
history_torrents = {}
|
||||
for t in torrents:
|
||||
logger.info(f"种子信息: {t}")
|
||||
history_torrents[t.hash]=t
|
||||
for h in history:
|
||||
# 判断当前历史记录的hash是否在未找到的hash列表中
|
||||
if h.download_hash not in history_torrents.keys():
|
||||
logger.info(f"种子 {h.download_hash} 已不存在于下载器中")
|
||||
self.delete_data(history=h)
|
||||
else:
|
||||
# 从下载器删除种子
|
||||
self.delete_download_history(h, history_torrents[h.download_hash])
|
||||
|
||||
|
||||
|
||||
|
||||
def delete_data(self, history: DownloadHistory):
|
||||
"""
|
||||
从订阅记录中删除该信息
|
||||
"""
|
||||
try:
|
||||
down_oper = DownloadHistoryOper()
|
||||
down_oper.delete_history(history.id)
|
||||
logger.info(f"删除下载历史记录：{history.id} {history.title} {history.seasons} {history.episodes} {history.download_hash}")
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"删除下载历史记录失败：{str(e)}")
|
||||
return False
|
||||
|
||||
|
||||
|
||||
def delete_download_history(self,history: DownloadHistory, torrent: Any):
|
||||
downloader_name = history.downloader
|
||||
downloader_obj = self.__get_downloader(downloader_name)
|
||||
logger.info(f"删除种子信息：{history.id} {history.title} {history.seasons} {history.episodes} {history.download_hash}")
|
||||
hashs = [history.download_hash]
|
||||
# 处理辅种
|
||||
torrents, error = downloader_obj.get_torrents()
|
||||
if error :
|
||||
logger.error(f"获取辅种信息失败： {error}")
|
||||
else:
|
||||
for t in torrents:
|
||||
if t.name == torrent.name and t.size == torrent.size:
|
||||
hashs.append(t.hash)
|
||||
downloader_obj.delete_torrents(delete_file=True,ids=hashs)
|
||||
self.delete_data(history)
|
||||
|
||||
|
||||
|
||||
def get_state(self) -> bool:
|
||||
return True if self._enabled else False
|
||||
|
||||
@staticmethod
|
||||
def get_command() -> List[Dict[str, Any]]:
|
||||
pass
|
||||
|
||||
def get_api(self) -> List[Dict[str, Any]]:
|
||||
pass
|
||||
|
||||
def get_service(self) -> List[Dict[str, Any]]:
|
||||
"""
|
||||
注册插件公共服务
|
||||
[{
|
||||
"id": "服务ID",
|
||||
"name": "服务名称",
|
||||
"trigger": "触发器：cron/interval/date/CronTrigger.from_crontab()",
|
||||
"func": self.xxx,
|
||||
"kwargs": {} # 定时器参数
|
||||
}]
|
||||
"""
|
||||
return []
|
||||
|
||||
def get_form(self) -> Tuple[List[dict], Dict[str, Any]]:
|
||||
# 获取下载历史数据
|
||||
histories = self.get_data()
|
||||
|
||||
# 构造标题和剧集列表
|
||||
titles = []
|
||||
episode_options = []
|
||||
|
||||
for history in histories:
|
||||
# 标题列表
|
||||
if history.title not in titles:
|
||||
titles.append(history.title)
|
||||
|
||||
# 剧集列表
|
||||
episode_str = history.title
|
||||
if history.seasons:
|
||||
episode_str += f" {history.seasons}"
|
||||
if history.episodes:
|
||||
episode_str += f" {history.episodes}"
|
||||
episode_options.append({"title": episode_str, "value": history.id})
|
||||
|
||||
|
||||
# 将列表转换为选择框选项格式
|
||||
title_options = [{"title": t, "value": t} for t in titles]
|
||||
|
||||
# 标题和剧集选择框
|
||||
title_select = {
|
||||
'component': 'VRow',
|
||||
'content': [
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VSelect',
|
||||
'props': {
|
||||
'model': 'titles',
|
||||
'label': '标题',
|
||||
'items': title_options,
|
||||
'multiple': True,
|
||||
'chips': True,
|
||||
'clearable': True
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
episode_select = {
|
||||
'component': 'VRow',
|
||||
'content': [
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VSelect',
|
||||
'props': {
|
||||
'model': 'episodes',
|
||||
'label': '剧集',
|
||||
'items': episode_options,
|
||||
'multiple': True,
|
||||
'chips': True,
|
||||
'clearable': True
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
return [
|
||||
{
|
||||
'component': 'VForm',
|
||||
'content': [
|
||||
title_select,
|
||||
episode_select
|
||||
]
|
||||
}
|
||||
], {
|
||||
"titles": [],
|
||||
"episodes": []
|
||||
}
|
||||
|
||||
def get_data(self) -> List[DownloadHistory]:
|
||||
down_oper = DownloadHistoryOper()
|
||||
downs = []
|
||||
page = 1
|
||||
while True:
|
||||
data = down_oper.list_by_page(page=page, count=100)
|
||||
downs.extend(data)
|
||||
if len(data) < 100:
|
||||
break
|
||||
page += 1
|
||||
return downs
|
||||
|
||||
def get_page(self) -> List[dict]:
|
||||
items = []
|
||||
for down in self.get_data():
|
||||
items.append({
|
||||
'component': 'tr',
|
||||
'content': [
|
||||
{
|
||||
'component': 'td',
|
||||
'text': down.id
|
||||
},
|
||||
{
|
||||
'component': 'td',
|
||||
'text': down.title
|
||||
},
|
||||
{
|
||||
'component': 'td',
|
||||
'text':down.seasons + " " + down.episodes
|
||||
},
|
||||
{
|
||||
'component': 'td',
|
||||
'text': down.torrent_name
|
||||
}
|
||||
]
|
||||
})
|
||||
|
||||
return [
|
||||
{
|
||||
'component': 'VRow',
|
||||
'content': [
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VTable',
|
||||
'props': {
|
||||
'hover': True
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'thead',
|
||||
'content': [
|
||||
{
|
||||
'component': 'tr',
|
||||
'content': [
|
||||
{
|
||||
'component': 'th',
|
||||
'props': {
|
||||
'class': 'text-start ps-4'
|
||||
},
|
||||
'text': 'id'
|
||||
},
|
||||
{
|
||||
'component': 'th',
|
||||
'props': {
|
||||
'class': 'text-start ps-4'
|
||||
},
|
||||
'text': '名称'
|
||||
},
|
||||
{
|
||||
'component': 'th',
|
||||
'props': {
|
||||
'class': 'text-start ps-4'
|
||||
},
|
||||
'text': '剧集'
|
||||
},
|
||||
{
|
||||
'component': 'th',
|
||||
'props': {
|
||||
'class': 'text-start ps-4'
|
||||
},
|
||||
'text': '种子名称'
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'tbody',
|
||||
'content': items
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
|
||||
|
||||
@staticmethod
|
||||
def get_api(self) -> List[Dict[str, Any]]:
|
||||
"""
|
||||
注册API
|
||||
"""
|
||||
pass
|
||||
|
||||
def stop_service(self):
|
||||
"""
|
||||
退出插件
|
||||
"""
|
||||
pass
|
||||
|
||||
@property
|
||||
def service_infos(self) -> Optional[Dict[str, ServiceInfo]]:
|
||||
"""
|
||||
服务信息
|
||||
"""
|
||||
services = self.downloader_helper.get_services(type_filter="qbittorrent")
|
||||
if not services:
|
||||
logger.warning("获取下载器实例失败，请检查配置")
|
||||
return None
|
||||
|
||||
active_services = {}
|
||||
for service_name, service_info in services.items():
|
||||
if service_info.instance.is_inactive():
|
||||
logger.warning(f"下载器 {service_name} 未连接，请检查配置")
|
||||
else:
|
||||
active_services[service_name] = service_info
|
||||
|
||||
if not active_services:
|
||||
logger.warning("没有已连接的下载器，请检查配置")
|
||||
return None
|
||||
|
||||
return active_services
|
||||
|
||||
def __get_downloader(self, name: str):
|
||||
"""
|
||||
根据类型返回下载器实例
|
||||
"""
|
||||
return self.service_infos.get(name).instance
|
||||
Reference in New Issue
Block a user