mirror of
https://github.com/thsrite/MoviePilot-Plugins.git
synced 2026-03-27 10:05:57 +00:00
671 lines
27 KiB
Python
671 lines
27 KiB
Python
import json
|
||
import os
|
||
import shutil
|
||
import urllib.parse
|
||
from datetime import datetime, timedelta
|
||
from pathlib import Path
|
||
|
||
import pytz
|
||
from typing import Any, List, Dict, Tuple, Optional
|
||
|
||
from app.schemas.types import EventType
|
||
from apscheduler.schedulers.background import BackgroundScheduler
|
||
from apscheduler.triggers.cron import CronTrigger
|
||
|
||
from app.log import logger
|
||
from app.plugins import _PluginBase
|
||
from app.core.config import settings
|
||
|
||
|
||
class CloudStrm(_PluginBase):
|
||
# 插件名称
|
||
plugin_name = "云盘Strm生成"
|
||
# 插件描述
|
||
plugin_desc = "定时扫描云盘文件，生成Strm文件。"
|
||
# 插件图标
|
||
plugin_icon = "https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/icons/create.png"
|
||
# 插件版本
|
||
plugin_version = "3.2"
|
||
# 插件作者
|
||
plugin_author = "thsrite"
|
||
# 作者主页
|
||
author_url = "https://github.com/thsrite"
|
||
# 插件配置项ID前缀
|
||
plugin_config_prefix = "cloudstrm_"
|
||
# 加载顺序
|
||
plugin_order = 26
|
||
# 可使用的用户级别
|
||
auth_level = 1
|
||
|
||
# 私有属性
|
||
_enabled = False
|
||
_cron = None
|
||
_monitor_confs = None
|
||
_onlyonce = False
|
||
_copy_files = False
|
||
_rebuild = False
|
||
_observer = []
|
||
_video_formats = ('.mp4', '.avi', '.rmvb', '.wmv', '.mov', '.mkv', '.flv', '.ts', '.webm', '.iso', '.mpg', '.m2ts')
|
||
__cloud_files_json = "cloud_files.json"
|
||
|
||
_dirconf = {}
|
||
_libraryconf = {}
|
||
_cloudtypeconf = {}
|
||
_cloudurlconf = {}
|
||
_cloudpathconf = {}
|
||
__cloud_files = []
|
||
|
||
# 定时器
|
||
_scheduler: Optional[BackgroundScheduler] = None
|
||
|
||
def init_plugin(self, config: dict = None):
|
||
# 清空配置
|
||
self._dirconf = {}
|
||
self._libraryconf = {}
|
||
self._cloudtypeconf = {}
|
||
self._cloudurlconf = {}
|
||
self._cloudpathconf = {}
|
||
self.__cloud_files_json = os.path.join(self.get_data_path(), self.__cloud_files_json)
|
||
|
||
if config:
|
||
self._enabled = config.get("enabled")
|
||
self._cron = config.get("cron")
|
||
self._onlyonce = config.get("onlyonce")
|
||
self._rebuild = config.get("rebuild")
|
||
self._copy_files = config.get("copy_files")
|
||
self._monitor_confs = config.get("monitor_confs")
|
||
|
||
# 停止现有任务
|
||
self.stop_service()
|
||
|
||
if self._enabled or self._onlyonce:
|
||
# 定时服务
|
||
self._scheduler = BackgroundScheduler(timezone=settings.TZ)
|
||
|
||
# 读取目录配置
|
||
monitor_confs = self._monitor_confs.split("\n")
|
||
if not monitor_confs:
|
||
return
|
||
for monitor_conf in monitor_confs:
|
||
# 格式 源目录:目的目录:媒体库内网盘路径:监控模式
|
||
if not monitor_conf:
|
||
continue
|
||
if str(monitor_conf).count("#") == 2:
|
||
source_dir = str(monitor_conf).split("#")[0]
|
||
target_dir = str(monitor_conf).split("#")[1]
|
||
library_dir = str(monitor_conf).split("#")[2]
|
||
self._libraryconf[source_dir] = library_dir
|
||
elif str(monitor_conf).count("#") == 4:
|
||
source_dir = str(monitor_conf).split("#")[0]
|
||
target_dir = str(monitor_conf).split("#")[1]
|
||
cloud_type = str(monitor_conf).split("#")[2]
|
||
cloud_path = str(monitor_conf).split("#")[3]
|
||
cloud_url = str(monitor_conf).split("#")[4]
|
||
self._cloudtypeconf[source_dir] = cloud_type
|
||
self._cloudpathconf[source_dir] = cloud_path
|
||
self._cloudurlconf[source_dir] = cloud_url
|
||
else:
|
||
logger.error(f"{monitor_conf} 格式错误")
|
||
continue
|
||
# 存储目录监控配置
|
||
self._dirconf[source_dir] = target_dir
|
||
|
||
# 检查媒体库目录是不是下载目录的子目录
|
||
try:
|
||
if target_dir and Path(target_dir).is_relative_to(Path(source_dir)):
|
||
logger.warn(f"{target_dir} 是下载目录 {source_dir} 的子目录，无法监控")
|
||
self.systemmessage.put(f"{target_dir} 是下载目录 {source_dir} 的子目录，无法监控")
|
||
continue
|
||
except Exception as e:
|
||
logger.debug(str(e))
|
||
pass
|
||
|
||
# 运行一次定时服务
|
||
if self._onlyonce:
|
||
logger.info("云盘监控全量执行服务启动，立即运行一次")
|
||
self._scheduler.add_job(func=self.__scan, trigger='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.__scan,
|
||
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 __scan(self):
|
||
"""
|
||
扫描
|
||
"""
|
||
if not self._enabled:
|
||
logger.error("插件未开启")
|
||
return
|
||
if not self._dirconf or not self._dirconf.keys():
|
||
logger.error("未获取到可用目录监控配置，请检查")
|
||
return
|
||
|
||
logger.info("云盘strm生成任务开始")
|
||
# 首次扫描或者重建索引
|
||
__init_flag = False
|
||
if self._rebuild or not Path(self.__cloud_files_json).exists():
|
||
logger.info("正在重建索引或初始化运行")
|
||
self.__init_cloud_files_json()
|
||
self._rebuild = False
|
||
self.__update_config()
|
||
__init_flag = True
|
||
else:
|
||
logger.info("尝试加载本地缓存")
|
||
# 尝试加载本地
|
||
with open(self.__cloud_files_json, 'r') as file:
|
||
content = file.read()
|
||
if content:
|
||
self.__cloud_files = json.loads(content)
|
||
|
||
# 本地没加载到则重建索引
|
||
if not self.__cloud_files:
|
||
logger.error("尝试加载本地缓存，开始重建索引")
|
||
self.__init_cloud_files_json()
|
||
self._rebuild = False
|
||
self.__update_config()
|
||
__init_flag = True
|
||
|
||
# 不是首次索引，则重新扫描、判断是否有新文件
|
||
if not __init_flag:
|
||
__save_flag = False
|
||
for source_dir in self._dirconf.keys():
|
||
logger.info(f"正在处理监控文件 {source_dir}")
|
||
for root, dirs, files in os.walk(source_dir):
|
||
# 如果遇到名为'extrafanart'的文件夹，则跳过处理该文件夹，继续处理其他文件夹
|
||
if "extrafanart" in dirs:
|
||
dirs.remove("extrafanart")
|
||
|
||
# 处理文件
|
||
for file in files:
|
||
source_file = os.path.join(root, file)
|
||
# 回收站及隐藏的文件不处理
|
||
if (source_file.find("/@Recycle") != -1
|
||
or source_file.find("/#recycle") != -1
|
||
or source_file.find("/.") != -1
|
||
or source_file.find("/@eaDir") != -1):
|
||
logger.info(f"{source_file} 是回收站或隐藏的文件，跳过处理")
|
||
continue
|
||
|
||
if source_file not in self.__cloud_files:
|
||
logger.info(f"扫描到新文件 {source_file}，正在开始处理")
|
||
# 云盘文件json新增
|
||
self.__cloud_files.append(source_file)
|
||
# 扫描云盘文件，判断是否有对应strm
|
||
self.__strm(source_file)
|
||
__save_flag = True
|
||
|
||
# 重新保存json文件
|
||
if __save_flag:
|
||
self.__sava_json()
|
||
|
||
logger.info("云盘strm生成任务完成")
|
||
|
||
def __init_cloud_files_json(self):
|
||
"""
|
||
初始化云盘文件json
|
||
"""
|
||
# init
|
||
for source_dir in self._dirconf.keys():
|
||
logger.info(f"正在处理监控文件 {source_dir}")
|
||
for root, dirs, files in os.walk(source_dir):
|
||
# 如果遇到名为'extrafanart'的文件夹，则跳过处理该文件夹，继续处理其他文件夹
|
||
if "extrafanart" in dirs:
|
||
dirs.remove("extrafanart")
|
||
|
||
# 处理文件
|
||
for file in files:
|
||
source_file = os.path.join(root, file)
|
||
# 回收站及隐藏的文件不处理
|
||
if (source_file.find("/@Recycle") != -1
|
||
or source_file.find("/#recycle") != -1
|
||
or source_file.find("/.") != -1
|
||
or source_file.find("/@eaDir") != -1):
|
||
logger.info(f"{source_file} 是回收站或隐藏的文件，跳过处理")
|
||
continue
|
||
|
||
logger.info(f"扫描到新文件 {source_file}，正在开始处理")
|
||
# 云盘文件json新增
|
||
self.__cloud_files.append(source_file)
|
||
# 扫描云盘文件，判断是否有对应strm
|
||
self.__strm(source_file)
|
||
|
||
# 写入本地文件
|
||
if self.__cloud_files:
|
||
self.__sava_json()
|
||
else:
|
||
logger.warning(f"未获取到文件列表")
|
||
|
||
def __sava_json(self):
|
||
"""
|
||
保存json文件
|
||
"""
|
||
logger.info(f"开始写入本地文件 {self.__cloud_files_json}")
|
||
file = open(self.__cloud_files_json, 'w')
|
||
file.write(json.dumps(self.__cloud_files))
|
||
file.close()
|
||
|
||
def __strm(self, source_file):
|
||
"""
|
||
判断文件是否有对应strm
|
||
"""
|
||
try:
|
||
# 获取文件的转移路径
|
||
for source_dir in self._dirconf.keys():
|
||
if str(source_file).startswith(source_dir):
|
||
# 转移路径
|
||
dest_dir = self._dirconf.get(source_dir)
|
||
# 媒体库容器内挂载路径
|
||
library_dir = self._libraryconf.get(source_dir)
|
||
# 云服务类型
|
||
cloud_type = self._cloudtypeconf.get(source_dir)
|
||
# 云服务挂载本地跟路径
|
||
cloud_path = self._cloudpathconf.get(source_dir)
|
||
# 云服务地址
|
||
cloud_url = self._cloudurlconf.get(source_dir)
|
||
|
||
# 转移后文件
|
||
dest_file = source_file.replace(source_dir, dest_dir)
|
||
# 如果是文件夹
|
||
if Path(dest_file).is_dir():
|
||
if not Path(dest_file).exists():
|
||
logger.info(f"创建目标文件夹 {dest_file}")
|
||
os.makedirs(dest_file)
|
||
continue
|
||
else:
|
||
if Path(dest_file).exists():
|
||
logger.info(f"目标文件 {dest_file} 已存在")
|
||
continue
|
||
|
||
# 文件
|
||
if not Path(dest_file).parent.exists():
|
||
logger.info(f"创建目标文件夹 {Path(dest_file).parent}")
|
||
os.makedirs(Path(dest_file).parent)
|
||
|
||
# 视频文件创建.strm文件
|
||
if dest_file.lower().endswith(self._video_formats):
|
||
# 创建.strm文件
|
||
self.__create_strm_file(dest_file=dest_file,
|
||
dest_dir=dest_dir,
|
||
source_file=source_file,
|
||
library_dir=library_dir,
|
||
cloud_type=cloud_type,
|
||
cloud_path=cloud_path,
|
||
cloud_url=cloud_url)
|
||
else:
|
||
if self._copy_files:
|
||
# 其他nfo、jpg等复制文件
|
||
shutil.copy2(source_file, dest_file)
|
||
logger.info(f"复制其他文件 {source_file} 到 {dest_file}")
|
||
except Exception as e:
|
||
logger.error(f"create strm file error: {e}")
|
||
print(str(e))
|
||
|
||
@staticmethod
|
||
def __create_strm_file(dest_file: str, dest_dir: str, source_file: str, library_dir: str = None,
|
||
cloud_type: str = None, cloud_path: str = None, cloud_url: str = None):
|
||
"""
|
||
生成strm文件
|
||
:param library_dir:
|
||
:param dest_dir:
|
||
:param dest_file:
|
||
"""
|
||
try:
|
||
# 获取视频文件名和目录
|
||
video_name = Path(dest_file).name
|
||
# 获取视频目录
|
||
dest_path = Path(dest_file).parent
|
||
|
||
if not dest_path.exists():
|
||
logger.info(f"创建目标文件夹 {dest_path}")
|
||
os.makedirs(str(dest_path))
|
||
|
||
# 构造.strm文件路径
|
||
strm_path = os.path.join(dest_path, f"{os.path.splitext(video_name)[0]}.strm")
|
||
logger.info(f"替换前本地路径:::{dest_file}")
|
||
|
||
# 云盘模式
|
||
if cloud_type:
|
||
# 替换路径中的\为/
|
||
dest_file = source_file.replace("\\", "/")
|
||
dest_file = dest_file.replace(cloud_path, "")
|
||
# 对盘符之后的所有内容进行url转码
|
||
dest_file = urllib.parse.quote(dest_file, safe='')
|
||
if str(cloud_type) == "cd2":
|
||
# 将路径的开头盘符"/mnt/user/downloads"替换为"http://localhost:19798/static/http/localhost:19798/False/"
|
||
dest_file = f"http://{cloud_url}/static/http/{cloud_url}/False/{dest_file}"
|
||
logger.info(f"替换后cd2路径:::{dest_file}")
|
||
elif str(cloud_type) == "alist":
|
||
dest_file = f"http://{cloud_url}/d/{dest_file}"
|
||
logger.info(f"替换后alist路径:::{dest_file}")
|
||
else:
|
||
logger.error(f"云盘类型 {cloud_type} 错误")
|
||
return
|
||
else:
|
||
# 本地挂载路径转为emby路径
|
||
dest_file = dest_file.replace(dest_dir, library_dir)
|
||
logger.info(f"替换后emby容器内路径:::{dest_file}")
|
||
|
||
# 写入.strm文件
|
||
with open(strm_path, 'w') as f:
|
||
f.write(dest_file)
|
||
|
||
logger.info(f"创建strm文件 {strm_path}")
|
||
except Exception as e:
|
||
logger.error(f"创建strm文件失败")
|
||
print(str(e))
|
||
|
||
def __update_config(self):
|
||
"""
|
||
更新配置
|
||
"""
|
||
self.update_config({
|
||
"enabled": self._enabled,
|
||
"onlyonce": self._onlyonce,
|
||
"rebuild": self._rebuild,
|
||
"copy_files": self._copy_files,
|
||
"cron": self._cron,
|
||
"monitor_confs": self._monitor_confs
|
||
})
|
||
|
||
def get_state(self) -> bool:
|
||
return self._enabled
|
||
|
||
@staticmethod
|
||
def get_command() -> List[Dict[str, Any]]:
|
||
"""
|
||
定义远程控制命令
|
||
:return: 命令关键字、事件、描述、附带数据
|
||
"""
|
||
return [{
|
||
"cmd": "/cloud_strm",
|
||
"event": EventType.PluginAction,
|
||
"desc": "云盘strm文件生成",
|
||
"category": "",
|
||
"data": {
|
||
"action": "cloud_strm"
|
||
}
|
||
}]
|
||
|
||
def get_service(self) -> List[Dict[str, Any]]:
|
||
"""
|
||
注册插件公共服务
|
||
[{
|
||
"id": "服务ID",
|
||
"name": "服务名称",
|
||
"trigger": "触发器：cron/interval/date/CronTrigger.from_crontab()",
|
||
"func": self.xxx,
|
||
"kwargs": {} # 定时器参数
|
||
}]
|
||
"""
|
||
if self._enabled:
|
||
return [{
|
||
"id": "CloudStrm",
|
||
"name": "云盘strm文件生成服务",
|
||
"trigger": CronTrigger.from_crontab(self._cron),
|
||
"func": self.__scan,
|
||
"kwargs": {}
|
||
}]
|
||
return []
|
||
|
||
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': 'onlyonce',
|
||
'label': '立即运行一次',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 4
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'rebuild',
|
||
'label': '重建索引',
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextField',
|
||
'props': {
|
||
'model': 'cron',
|
||
'label': '定时周期',
|
||
'placeholder': '0 0 * * *'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VTextarea',
|
||
'props': {
|
||
'model': 'monitor_confs',
|
||
'label': '监控目录',
|
||
'rows': 5,
|
||
'placeholder': '监控方式#监控目录#目的目录#媒体服务器内源文件路径'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
'md': 6
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VSwitch',
|
||
'props': {
|
||
'model': 'copy_files',
|
||
'label': '复制非媒体文件',
|
||
}
|
||
}
|
||
]
|
||
},
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '目录监控格式：'
|
||
'1.监控目录#目的目录#媒体服务器内源文件路径；'
|
||
'2.监控目录#目的目录#cd2#cd2挂载本地跟路径#cd2服务地址；'
|
||
'3.监控目录#目的目录#alist#alist挂载本地跟路径#alist服务地址。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '媒体服务器内源文件路径：源文件目录即云盘挂载到媒体服务器的路径。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '立即运行一次：'
|
||
'全量运行一次。'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
'component': 'VRow',
|
||
'content': [
|
||
{
|
||
'component': 'VCol',
|
||
'props': {
|
||
'cols': 12,
|
||
},
|
||
'content': [
|
||
{
|
||
'component': 'VAlert',
|
||
'props': {
|
||
'type': 'info',
|
||
'variant': 'tonal',
|
||
'text': '配置说明：'
|
||
'https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/docs/CloudStrm.md'
|
||
}
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
]
|
||
}
|
||
], {
|
||
"enabled": False,
|
||
"cron": "",
|
||
"onlyonce": False,
|
||
"rebuild": False,
|
||
"copy_files": False,
|
||
"monitor_confs": ""
|
||
}
|
||
|
||
def get_page(self) -> List[dict]:
|
||
pass
|
||
|
||
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))
|