mirror of
https://github.com/d0zingcat/MoviePilot-Plugins.git
synced 2026-05-13 23:16:47 +00:00
Merge branch 'main' into main
This commit is contained in:
Zhilong Liu
GitHub
@@ -653,7 +653,6 @@
|
||||
"icon": "Amule_A.png",
|
||||
"author": "jxxghp",
|
||||
"level": 1,
|
||||
"v2": true,
|
||||
"history": {
|
||||
"v2.0": "适配新的目录结构变化，短剧分类名称调整为配置目录路径，升级后需要重新调整设置后才能使用。"
|
||||
}
|
||||
|
||||
@@ -195,11 +195,12 @@
|
||||
"name": "IYUU自动辅种",
|
||||
"description": "基于IYUU官方Api实现自动辅种。",
|
||||
"labels": "做种,IYUU",
|
||||
"version": "2.1",
|
||||
"version": "2.2",
|
||||
"icon": "IYUU.png",
|
||||
"author": "jxxghp",
|
||||
"level": 2,
|
||||
"history": {
|
||||
"v2.2": "修复种子校验服务未生效",
|
||||
"v2.1": "调整IYUU最新域名",
|
||||
"v2.0": "兼容MoviePilot V2 版本"
|
||||
}
|
||||
@@ -263,5 +264,18 @@
|
||||
"history": {
|
||||
"v2.0": "适配 MoviePilot V2"
|
||||
}
|
||||
},
|
||||
"PlayletCategory": {
|
||||
"name": "短剧自动分类",
|
||||
"description": "网络短剧自动整理到独立的分类目录。",
|
||||
"labels": "文件整理",
|
||||
"version": "2.1",
|
||||
"icon": "Amule_A.png",
|
||||
"author": "jxxghp,longqiuyu",
|
||||
"level": 1,
|
||||
"history": {
|
||||
"v2.1": "兼容MoviePilot V2",
|
||||
"v2.0": "适配新的目录结构变化，短剧分类名称调整为配置目录路径，升级后需要重新调整设置后才能使用。"
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -33,7 +33,7 @@ class IYUUAutoSeed(_PluginBase):
|
||||
# 插件图标
|
||||
plugin_icon = "IYUU.png"
|
||||
# 插件版本
|
||||
plugin_version = "2.1"
|
||||
plugin_version = "2.2"
|
||||
# 插件作者
|
||||
plugin_author = "jxxghp"
|
||||
# 作者主页
|
||||
@@ -140,15 +140,8 @@ class IYUUAutoSeed(_PluginBase):
|
||||
run_date=datetime.now(
|
||||
tz=pytz.timezone(settings.TZ)) + timedelta(seconds=3)
|
||||
)
|
||||
|
||||
# 关闭一次性开关
|
||||
self._onlyonce = False
|
||||
if self._scheduler.get_jobs():
|
||||
# 追加种子校验服务
|
||||
self._scheduler.add_job(self.check_recheck, 'interval', minutes=3)
|
||||
# 启动服务
|
||||
self._scheduler.print_jobs()
|
||||
self._scheduler.start()
|
||||
|
||||
if self._clearcache:
|
||||
# 关闭清除缓存开关
|
||||
@@ -156,6 +149,12 @@ class IYUUAutoSeed(_PluginBase):
|
||||
# 保存配置
|
||||
self.__update_config()
|
||||
|
||||
# 追加种子校验服务
|
||||
self._scheduler.add_job(self.check_recheck, 'interval', minutes=3)
|
||||
# 启动服务
|
||||
self._scheduler.print_jobs()
|
||||
self._scheduler.start()
|
||||
|
||||
@property
|
||||
def service_infos(self) -> Optional[Dict[str, ServiceInfo]]:
|
||||
"""
|
||||
|
||||
357
plugins.v2/playletcategory/__init__.py
Normal file
357
plugins.v2/playletcategory/__init__.py
Normal file
@@ -0,0 +1,357 @@
|
||||
import random
|
||||
import time
|
||||
import shutil
|
||||
import subprocess
|
||||
import threading
|
||||
from pathlib import Path
|
||||
from typing import Any, List, Dict, Tuple
|
||||
|
||||
from app.core.config import settings
|
||||
from app.core.context import MediaInfo
|
||||
from app.core.event import eventmanager, Event
|
||||
from app.log import logger
|
||||
from app.plugins import _PluginBase
|
||||
from app.schemas import TransferInfo
|
||||
from app.schemas.file import FileItem
|
||||
from app.schemas.types import EventType, MediaType, NotificationType
|
||||
from app.utils.system import SystemUtils
|
||||
|
||||
lock = threading.Lock()
|
||||
|
||||
|
||||
class PlayletCategory(_PluginBase):
|
||||
# 插件名称
|
||||
plugin_name = "短剧自动分类"
|
||||
# 插件描述
|
||||
plugin_desc = "网络短剧自动分类到独立目录。"
|
||||
# 插件图标
|
||||
plugin_icon = "Amule_A.png"
|
||||
# 插件版本
|
||||
plugin_version = "2.1"
|
||||
# 插件作者
|
||||
plugin_author = "jxxghp,longqiuyu"
|
||||
# 作者主页
|
||||
author_url = "https://github.com/jxxghp"
|
||||
# 插件配置项ID前缀
|
||||
plugin_config_prefix = "playletcategory_"
|
||||
# 加载顺序
|
||||
plugin_order = 29
|
||||
# 可使用的用户级别
|
||||
auth_level = 1
|
||||
|
||||
_enabled = False
|
||||
_notify = True
|
||||
_delay: int = 0
|
||||
_category_dir = ""
|
||||
_episode_duration = 8
|
||||
|
||||
def init_plugin(self, config: dict = None):
|
||||
|
||||
if config:
|
||||
self._enabled = config.get("enabled")
|
||||
self._delay = config.get("delay") or 0
|
||||
self._notify = config.get("notify")
|
||||
self._category_dir = config.get("category_dir")
|
||||
self._episode_duration = config.get("episode_duration")
|
||||
|
||||
def get_state(self) -> bool:
|
||||
return True if self._enabled and self._category_dir and self._episode_duration else False
|
||||
|
||||
@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': 'notify',
|
||||
'label': '发送消息',
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VRow',
|
||||
'content': [
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
'md': 4,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VTextField',
|
||||
'props': {
|
||||
'model': 'category_dir',
|
||||
'label': '分类目录路径',
|
||||
'placeholder': '/media/短剧'
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
'md': 4,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VTextField',
|
||||
'props': {
|
||||
'model': 'episode_duration',
|
||||
'label': '单集时长（分钟）',
|
||||
'placeholder': '8'
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
'md': 4,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VTextField',
|
||||
'props': {
|
||||
'model': 'delay',
|
||||
'label': '入库延迟时间（秒）',
|
||||
'placeholder': '使用刮削尽量设置大一些'
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
'component': 'VRow',
|
||||
'content': [
|
||||
{
|
||||
'component': 'VCol',
|
||||
'props': {
|
||||
'cols': 12,
|
||||
},
|
||||
'content': [
|
||||
{
|
||||
'component': 'VAlert',
|
||||
'props': {
|
||||
'type': 'info',
|
||||
'variant': 'tonal',
|
||||
'text': '小于单集时长的剧集视频文件将会移动到分类目录，入库延迟适用于网盘等需要延后处理的场景，需要安装FFmpeg。'
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
], {
|
||||
"enabled": False,
|
||||
"notify": True,
|
||||
"delay": '',
|
||||
"category_dir": '短剧',
|
||||
"episode_duration": '8'
|
||||
}
|
||||
|
||||
def get_page(self) -> List[dict]:
|
||||
pass
|
||||
|
||||
@eventmanager.register(EventType.TransferComplete)
|
||||
def category_handler(self, event: Event):
|
||||
"""
|
||||
根据事件实时刮削剧集组信息
|
||||
"""
|
||||
logger.debug(f"触发短剧分类！")
|
||||
if not event:
|
||||
logger.debug(f"短剧分类异常：{event}")
|
||||
return
|
||||
if not self.get_state():
|
||||
logger.debug(f"短剧分类插件配置不完整！")
|
||||
return
|
||||
try:
|
||||
event_data = event.event_data
|
||||
media_info: MediaInfo = event_data.get("mediainfo")
|
||||
transfer_info: TransferInfo = event_data.get("transferinfo")
|
||||
if not media_info or not transfer_info:
|
||||
return
|
||||
if not transfer_info.success:
|
||||
logger.debug(f"整理失败不做处理！")
|
||||
return
|
||||
if not transfer_info.target_diritem.path:
|
||||
logger.debug(f"文件路径不存在:{transfer_info.target_diritem.path}")
|
||||
return
|
||||
target_path = Path(transfer_info.target_diritem.path)
|
||||
if not target_path.exists():
|
||||
logger.debug(f"文件路径不存在:{target_path}")
|
||||
return
|
||||
if media_info.type != MediaType.TV:
|
||||
logger.info(f"{target_path} 不是电视剧，跳过分类处理")
|
||||
return
|
||||
if int(self._delay) > 0:
|
||||
# 进行延迟
|
||||
time.sleep(int(self._delay))
|
||||
# 加锁
|
||||
with lock:
|
||||
file_list = transfer_info.file_list_new or []
|
||||
# 过滤掉不存在的文件
|
||||
file_list = [file for file in file_list if Path(file).exists()]
|
||||
if not file_list:
|
||||
logger.warn(f"{target_path} 无文件，跳过分类处理")
|
||||
return
|
||||
logger.info(f"开始处理 {target_path} 短剧分类，共有 {len(file_list)} 个文件")
|
||||
# 从文件列表中随机抽取3个文件
|
||||
if len(file_list) > 3:
|
||||
check_files = random.choices(file_list, k=3)
|
||||
else:
|
||||
check_files = file_list
|
||||
# 计算文件时长，有任意文件时长大于单集时长则不处理
|
||||
need_category = True
|
||||
for file in check_files:
|
||||
duration = self.__get_duration(file)
|
||||
if duration > float(self._episode_duration):
|
||||
logger.info(f"{file} 时长 {duration} 分钟，大于单集时长 {self._episode_duration} 分钟，不需要分类处理")
|
||||
need_category = False
|
||||
break
|
||||
else:
|
||||
logger.info(f"{file} 时长：{duration} 分钟")
|
||||
if need_category:
|
||||
logger.info(f"{target_path} 需要分类处理，开始移动文件...")
|
||||
result = self.__move_files(target_path=target_path)
|
||||
if result:
|
||||
logger.info(f"{target_path} 短剧分类处理完成")
|
||||
else:
|
||||
logger.info(f"{target_path} 短剧分类移动失败！")
|
||||
else:
|
||||
logger.info(f"{target_path} 不是短剧，无需分类处理")
|
||||
except Exception as e:
|
||||
logger.info(f"短剧分类异常:{str(e)}")
|
||||
|
||||
@staticmethod
|
||||
def __get_duration(video_path: str) -> float:
|
||||
"""
|
||||
获取视频文件时长（分钟）
|
||||
"""
|
||||
|
||||
# 使用FFmpeg命令行工具获取视频时长
|
||||
cmd = ['ffprobe', '-v', 'error', '-show_entries', 'format=duration', '-of',
|
||||
'default=noprint_wrappers=1:nokey=1', str(video_path)]
|
||||
process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||||
output, error = process.communicate()
|
||||
|
||||
# 如果有错误，输出错误信息
|
||||
if error:
|
||||
logger.error(f"FFmpeg处理出错: {error.decode('utf-8')}")
|
||||
return 0
|
||||
|
||||
# 获取视频时长（秒），转换为分钟
|
||||
return round(float(output) / 60, 1)
|
||||
|
||||
def __move_files(self, target_path: Path) -> bool:
|
||||
"""
|
||||
移动文件到分类目录
|
||||
:param target_path: 电视剧时为季的目录
|
||||
"""
|
||||
logger.debug(f"target_path: {target_path}")
|
||||
if not target_path.exists():
|
||||
logger.warning(f"目标路径 {target_path} 不存在，跳过处理。")
|
||||
return False
|
||||
if target_path.is_file():
|
||||
target_path = target_path.parent
|
||||
# 剧集的根目录
|
||||
tv_path = target_path
|
||||
# 新的文件目录
|
||||
new_path = Path(self._category_dir) / target_path.name
|
||||
logger.debug(f"{new_path}")
|
||||
if not new_path.exists():
|
||||
# 移动目录
|
||||
try:
|
||||
shutil.move(target_path, new_path)
|
||||
except Exception as e:
|
||||
logger.error(f"移动文件失败：{e}")
|
||||
return False
|
||||
else:
|
||||
# 遍历目录下的所有文件，并移动到目的目录
|
||||
for file in target_path.iterdir():
|
||||
logger.debug(f"{file}")
|
||||
if file.is_file():
|
||||
try:
|
||||
# 相对路径
|
||||
relative_path = file.relative_to(target_path)
|
||||
logger.debug(f"relative_path:{to_path}")
|
||||
to_path = new_path / relative_path
|
||||
logger.debug(f"to_path:{to_path}")
|
||||
shutil.move(file, to_path)
|
||||
except Exception as e:
|
||||
logger.error(f"移动文件失败：{e}")
|
||||
return False
|
||||
else:
|
||||
# 整季移动
|
||||
try:
|
||||
shutil.move(file, new_path)
|
||||
except Exception as e:
|
||||
logger.error(f"移动文件失败：{e}")
|
||||
return False
|
||||
# 删除空目录
|
||||
if not SystemUtils.list_files(target_path, extensions=settings.RMT_MEDIAEXT + settings.DOWNLOAD_TMPEXT):
|
||||
try:
|
||||
shutil.rmtree(target_path, ignore_errors=True)
|
||||
except Exception as e:
|
||||
logger.error(f"删除空目录失败：{e}")
|
||||
|
||||
# 发送消息
|
||||
if self._notify:
|
||||
self.post_message(
|
||||
mtype=NotificationType.Organize,
|
||||
title="【短剧自动分类】",
|
||||
text=f"已将 {tv_path.name} 分类到 {self._category_dir} 目录",
|
||||
)
|
||||
return True
|
||||
|
||||
def stop_service(self):
|
||||
"""
|
||||
停止服务
|
||||
"""
|
||||
pass
|
||||
Reference in New Issue
Block a user