diff --git a/README.md b/README.md index 54ace23..e78b7ab 100644 --- a/README.md +++ b/README.md @@ -26,4 +26,5 @@ MoviePilot三方插件市场:https://github.com/thsrite/MoviePilot-Plugins/ - [豆瓣明星热映订阅 1.3](docs%2FActorSubscribe.md) - [短剧刮削 2.5](docs%2FShortPlayMonitor.md) - [云盘实时链接 1.1](docs%2FCloudLinkMonitor.md) +- [源文件恢复 1.1](docs%2FLinkToSrc.md) diff --git a/docs/LinkToSrc.md b/docs/LinkToSrc.md new file mode 100644 index 0000000..413d6df --- /dev/null +++ b/docs/LinkToSrc.md @@ -0,0 +1,5 @@ +# 源文件恢复 + +### 更新记录 + +- 1.0 根据MoviePilot的转移记录中的硬链文件恢复源文件 diff --git a/package.json b/package.json index 06ed749..0561512 100644 --- a/package.json +++ b/package.json @@ -126,5 +126,13 @@ "icon": "Linkease_A.png", "author": "thsrite", "level": 1 + }, + "LinkToSrc": { + "name": "源文件恢复", + "description": "根据MoviePilot的转移记录中的硬链文件恢复源文件。", + "version": "1.0", + "icon": "Time_machine_A.png", + "author": "thsrite", + "level": 1 } } diff --git a/plugins/linktosrc/__init__.py b/plugins/linktosrc/__init__.py new file mode 100644 index 0000000..709e8fd --- /dev/null +++ b/plugins/linktosrc/__init__.py @@ -0,0 +1,155 @@ +import sqlite3 +from pathlib import Path +from typing import List, Tuple, Dict, Any + +from app.core.config import Settings +from app.log import logger +from app.plugins import _PluginBase + + +class LinkToSrc(_PluginBase): + # 插件名称 + plugin_name = "源文件恢复" + # 插件描述 + plugin_desc = "根据MoviePilot的转移记录中的硬链文件恢复源文件" + # 插件图标 + plugin_icon = "Time_machine_A.png" + # 插件版本 + plugin_version = "1.0" + # 插件作者 + plugin_author = "thsrite" + # 作者主页 + author_url = "https://github.com/thsrite" + # 插件配置项ID前缀 + plugin_config_prefix = "linktosrc_" + # 加载顺序 + plugin_order = 32 + # 可使用的用户级别 + auth_level = 1 + + _onlyonce: bool = False + + def init_plugin(self, config: dict = None): + if config: + self._onlyonce = config.get("onlyonce") + + if self._onlyonce: + # 执行替换 + self._task() + self._onlyonce = False + self.__update_config() + + def _task(self): + db_path = Settings.CONFIG_PATH / "user.db" + try: + gradedb = sqlite3.connect(db_path) + except Exception as e: + logger.error(f"无法打开数据库文件 {db_path},请检查路径是否正确:{str(e)}") + return + + # 创建游标cursor来执行executeSQL语句 + cursor = gradedb.cursor() + sql = ''' + SELECT + src, + dest + FROM + transferhistory + WHERE + src IS NOT NULL and dest IS NOT NULL; + ''' + cursor.execute(sql) + transfer_history = cursor.fetchall() + logger.info(f"查询到历史记录{len(transfer_history)}条") + cursor.close() + for history in transfer_history: + src = history[0] + dest = history[1] + # 判断源文件是否存在 + if Path(src).exists(): + logger.warn(f"源文件{src}已存在,跳过处理") + continue + # 源文件不存在,目标文件也不存在,跳过 + if not Path(dest).exists(): + logger.warn(f"源文件{src}不存在且硬链文件{dest}不存在,跳过处理") + continue + # 目标文件硬链回源文件 + Path(src).hardlink_to(dest) + logger.info(f"硬链文件{dest}重新链接回源文件{src}") + + logger.info("全部处理完成") + + def __update_config(self): + self.update_config({ + "onlyonce": self._onlyonce + }) + + @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': 6 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'onlyonce', + 'label': '立即运行一次', + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VAlert', + 'props': { + 'type': 'info', + 'variant': 'tonal', + 'text': '根据转移记录中的硬链接恢复源文件', + 'style': 'white-space: pre-line;' + } + } + ] + } + ] + } + ] + } + ], { + "onlyonce": False + } + + def get_page(self) -> List[dict]: + pass + + def get_state(self) -> bool: + return self._onlyonce + + def stop_service(self): + pass