import os import threading import time import traceback from pathlib import Path from typing import List, Tuple, Dict, Any from watchdog.events import FileSystemEventHandler from watchdog.observers import Observer from app.db.transferhistory_oper import TransferHistoryOper from app.log import logger from app.plugins import _PluginBase from app.schemas import NotificationType from app.core.event import eventmanager from app.schemas.types import EventType state_lock = threading.Lock() class FileMonitorHandler(FileSystemEventHandler): """ 目录监控处理 """ def __init__(self, monpath: str, sync: Any, **kwargs): super(FileMonitorHandler, self).__init__(**kwargs) self._watch_path = monpath self.sync = sync def on_created(self, event): if event.is_directory: return file_path = Path(event.src_path) if file_path.suffix in [".!qB", ".part", ".mp"]: return logger.info(f"监测到新增文件：{file_path}") if self.sync.exclude_keywords: for keyword in self.sync.exclude_keywords.split("\n"): if keyword and keyword in str(file_path): logger.info(f"{file_path} 命中过滤关键字 {keyword}，不处理") return # 新增文件记录 with state_lock: self.sync.state_set[str(file_path)] = file_path.stat().st_ino def on_moved(self, event): if event.is_directory: return file_path = Path(event.dest_path) if file_path.suffix in [".!qB", ".part", ".mp"]: return logger.info(f"监测到新增文件：{file_path}") if self.sync.exclude_keywords: for keyword in self.sync.exclude_keywords.split("\n"): if keyword and keyword in str(file_path): logger.info(f"{file_path} 命中过滤关键字 {keyword}，不处理") return # 新增文件记录 with state_lock: self.sync.state_set[str(file_path)] = file_path.stat().st_ino def on_deleted(self, event): file_path = Path(event.src_path) if event.is_directory: # 单独处理文件夹删除触发删除种子 if self.sync._delete_torrents: # 发送事件 logger.info(f"监测到删除文件夹：{file_path}") eventmanager.send_event( EventType.DownloadFileDeleted, {"src": str(file_path)} ) return if file_path.suffix in [".!qB", ".part", ".mp"]: return logger.info(f"监测到删除文件：{file_path}") # 命中过滤关键字不处理 if self.sync.exclude_keywords: for keyword in self.sync.exclude_keywords.split("\n"): if keyword and keyword in str(file_path): logger.info(f"{file_path} 命中过滤关键字 {keyword}，不处理") return # 删除硬链接文件 self.sync.handle_deleted(file_path) def updateState(monitor_dirs: List[str]): """ 更新监控目录的文件列表 """ # 记录开始时间 start_time = time.time() state_set = {} for mon_path in monitor_dirs: for root, dirs, files in os.walk(mon_path): for file in files: file = Path(root) / file if not file.exists(): continue # 记录文件inode state_set[str(file)] = file.stat().st_ino # 记录结束时间 end_time = time.time() # 计算耗时 elapsed_time = end_time - start_time logger.info(f"更新文件列表完成，共计{len(state_set)}个文件，耗时：{elapsed_time}秒") return state_set class RemoveLink(_PluginBase): # 插件名称 plugin_name = "清理硬链接" # 插件描述 plugin_desc = "监控目录内文件被删除时，同步删除监控目录内所有和它硬链接的文件" # 插件图标 plugin_icon = "Ombi_A.png" # 插件版本 plugin_version = "2.1" # 插件作者 plugin_author = "DzAvril" # 作者主页 author_url = "https://github.com/DzAvril" # 插件配置项ID前缀 plugin_config_prefix = "linkdeleted_" # 加载顺序 plugin_order = 0 # 可使用的用户级别 auth_level = 1 # preivate property monitor_dirs = "" exclude_dirs = "" exclude_keywords = "" _enabled = False _notify = False _delete_scrap_infos = False _delete_torrents = False _delete_history = False _transferhistory = None _observer = [] # 监控目录的文件列表 state_set: Dict[str, int] = {} def init_plugin(self, config: dict = None): logger.info(f"Hello, RemoveLink! config {config}") self._transferhistory = TransferHistoryOper() if config: self._enabled = config.get("enabled") self._notify = config.get("notify") self.monitor_dirs = config.get("monitor_dirs") self.exclude_dirs = config.get("exclude_dirs") or "" self.exclude_keywords = config.get("exclude_keywords") or "" self._delete_scrap_infos = config.get("delete_scrap_infos") self._delete_torrents = config.get("delete_torrents") self._delete_history = config.get("delete_history") # 停止现有任务 self.stop_service() if self._enabled: # 读取目录配置 monitor_dirs = self.monitor_dirs.split("\n") logger.info(f"监控目录：{monitor_dirs}") if not monitor_dirs: return for mon_path in monitor_dirs: # 格式源目录:目的目录 if not mon_path: continue try: observer = Observer(timeout=10) self._observer.append(observer) observer.schedule( FileMonitorHandler(mon_path, self), mon_path, recursive=True ) observer.daemon = True observer.start() logger.info(f"{mon_path} 的目录监控服务启动") except Exception as e: err_msg = str(e) logger.error(f"{mon_path} 启动目录监控失败：{err_msg}") self.systemmessage.put(f"{mon_path} 启动目录监控失败：{err_msg}") # 更新监控集合 with state_lock: self.state_set = updateState(monitor_dirs) def __update_config(self): """ 更新配置 """ self.update_config( { "enabled": self._enabled, "notify": self._notify, "monitor_dirs": self.monitor_dirs, "exclude_keywords": self.exclude_keywords, } ) 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]]: 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": "VRow", "content": [ { "component": "VCol", "props": {"cols": 12, "md": 4}, "content": [ { "component": "VSwitch", "props": { "model": "delete_scrap_infos", "label": "清理刮削文件(beta)", }, } ], }, { "component": "VCol", "props": {"cols": 12, "md": 4}, "content": [ { "component": "VSwitch", "props": { "model": "delete_torrents", "label": "联动删除种子", }, } ], }, { "component": "VCol", "props": {"cols": 12, "md": 4}, "content": [ { "component": "VSwitch", "props": { "model": "delete_history", "label": "删除历史记录", }, } ], }, ], }, { "component": "VRow", "content": [ { "component": "VCol", "props": {"cols": 12}, "content": [ { "component": "VTextarea", "props": { "model": "monitor_dirs", "label": "监控目录", "rows": 5, "placeholder": "源目录及硬链接目录均需加入监控，每一行一个目录", }, } ], } ], }, { "component": "VRow", "content": [ { "component": "VCol", "props": {"cols": 12}, "content": [ { "component": "VTextarea", "props": { "model": "exclude_dirs", "label": "不删除目录", "rows": 5, "placeholder": "该目录下的文件不会被动删除，一行一个目录", }, } ], } ], }, { "component": "VRow", "content": [ { "component": "VCol", "props": { "cols": 12, }, "content": [ { "component": "VTextarea", "props": { "model": "exclude_keywords", "label": "排除关键词", "rows": 2, "placeholder": "每一行一个关键词", }, } ], } ], }, { "component": "VRow", "content": [ { "component": "VCol", "props": { "cols": 12, }, "content": [ { "component": "VAlert", "props": { "type": "info", "variant": "tonal", "text": "联动删除种子需安装插件[下载器助手]并打开监听源文件事件", }, } ], }, { "component": "VCol", "props": { "cols": 12, }, "content": [ { "component": "VAlert", "props": { "type": "info", "variant": "tonal", "text": "清理刮削文件为测试功能，请谨慎开启。", }, } ], }, { "component": "VCol", "props": { "cols": 12, }, "content": [ { "component": "VAlert", "props": { "type": "info", "variant": "tonal", "text": "监控目录如有多个需换行，源目录和硬链接目录都需要添加到监控目录中；如需实现删除硬链接时不删除源文件，可把源文件目录配置到不删除目录中。", }, } ], }, ], }, ], } ], { "enabled": False, "notify": False, "monitor_dirs": "", "exclude_keywords": "", } def get_page(self) -> List[dict]: pass def stop_service(self): """ 退出插件 """ if self._observer: for observer in self._observer: try: observer.stop() observer.join() except Exception as e: print(str(e)) logger.error(f"停止目录监控失败：{str(e)}") self._observer = [] def __is_excluded(self, file_path: Path) -> bool: """ 是否排除目录 """ for exclude_dir in self.exclude_dirs.split("\n"): if exclude_dir and exclude_dir in str(file_path): return True return False @staticmethod def scrape_files_left(path): """ 检查path目录是否只包含刮削文件 """ # 检查path下是否有目录 for dir_path in os.listdir(path): if os.path.isdir(os.path.join(path, dir_path)): return False # 检查path下是否有非刮削文件 for file in path.iterdir(): if not file.suffix.lower() in [ ".jpg", ".nfo", ]: return False return True def delete_scrap_infos(self, path): """ 清理path相关的刮削文件 """ if not self._delete_scrap_infos: return # 文件所在目录已被删除则退出 if not os.path.exists(path.parent): return # logger.info(f"清理刮削文件: {path}") if not path.suffix.lower() in [ ".jpg", ".nfo", ]: # 清理与path相关的刮削文件 name_prefix = path.stem for file in path.parent.iterdir(): if file.name.startswith(name_prefix): file.unlink() logger.info(f"删除刮削文件：{file}") # 清理空目录 self.delete_empty_folders(path) def delete_history(self, path): """ 清理path相关的历史记录 """ if not self._delete_history: return # 查找历史记录 transfer_history = self._transferhistory.get_by_src(path) if transfer_history: # 删除历史记录 self._transferhistory.delete(transfer_history.id) logger.info(f"删除历史记录：{transfer_history.id}") def delete_empty_folders(self, path): """ 从指定路径开始，逐级向上层目录检测并删除空目录，直到遇到非空目录或到达指定监控目录为止 """ # logger.info(f"清理空目录: {path}") while True: parent_path = path.parent if self.__is_excluded(parent_path): break # parent_path如已被删除则退出检查 if not os.path.exists(parent_path): break # 如果当前路径等于监控目录之一，停止向上检查 if parent_path in self.monitor_dirs.split("\n"): break # 若目录下只剩刮削文件，则清空文件夹 if self.scrape_files_left(parent_path): # 清除目录下所有文件 for file in parent_path.iterdir(): file.unlink() logger.info(f"删除刮削文件：{file}") if not os.listdir(parent_path): os.rmdir(parent_path) logger.info(f"清理空目录：{parent_path}") if self._notify: self.post_message( mtype=NotificationType.SiteMessage, title=f"【清理硬链接】", text=f"清理空文件夹：[{parent_path}]\n", ) else: break # 更新路径为父目录，准备下一轮检查 path = parent_path def handle_deleted(self, file_path: Path): """ 处理删除事件 """ # 删除的文件对应的监控信息 with state_lock: # 清理刮削文件 self.delete_scrap_infos(file_path) if self._delete_torrents: # 发送事件 eventmanager.send_event( EventType.DownloadFileDeleted, {"src": str(file_path)} ) # 删除历史记录 self.delete_history(str(file_path)) # 删除的文件inode deleted_inode = self.state_set.get(str(file_path)) if not deleted_inode: logger.info(f"文件 {file_path} 未在监控列表中，不处理") return else: self.state_set.pop(str(file_path)) try: # 在current_set中查找与deleted_inode有相同inode的文件并删除 for path, inode in self.state_set.copy().items(): if inode == deleted_inode: file = Path(path) if self.__is_excluded(file): logger.info(f"文件 {file} 在不删除目录中，不处理") continue # 删除硬链接文件 logger.info(f"删除硬链接文件：{path}， inode: {inode}") file.unlink() # 清理刮削文件 self.delete_scrap_infos(file_path) if self._delete_torrents: # 发送事件 eventmanager.send_event( EventType.DownloadFileDeleted, {"src": str(file_path)} ) # 删除历史记录 self.delete_history(str(file_path)) if self._notify: self.post_message( mtype=NotificationType.SiteMessage, title=f"【清理硬链接】", text=f"监控到删除源文件：[{file_path}]\n" f"同步删除硬链接文件：[{path}]", ) except Exception as e: logger.error( "删除硬链接文件发生错误：%s - %s" % (str(e), traceback.format_exc()) )