from typing import List, Tuple, Dict, Any, Union, Optional from apscheduler.triggers.cron import CronTrigger from app.log import logger from app.modules.qbittorrent import Qbittorrent from qbittorrentapi.torrents import TorrentInfoList from app.modules.transmission import Transmission from transmission_rpc.torrent import Torrent from app.plugins import _PluginBase from app.schemas import NotificationType class TrackerEditor(_PluginBase): # 插件名称 plugin_name = "Tracker替换" # 插件描述 plugin_desc = "批量替换种子tracker，支持周期性巡检（如为TR，仅支持4.0以上版本）" # 插件图标 plugin_icon = "trackereditor_A.png" # 插件版本 plugin_version = "1.4" # 插件作者 plugin_author = "honue" # 作者主页 author_url = "https://github.com/honue" # 插件配置项ID前缀 plugin_config_prefix = "trackereditor_" # 加载顺序 plugin_order = 30 # 可使用的用户级别 auth_level = 1 _downloader_type: str = None _username: str = None _password: str = None _host: str = None _port: int = None _target_domain: str = None _replace_domain: str = None _onlyonce: bool = False _downloader: Union[Qbittorrent, Transmission] = None _run_con_enable: bool = False _run_con: Optional[str] = None _notify: bool = False def init_plugin(self, config: dict = None): if config: self._onlyonce = config.get("onlyonce") self._downloader_type = config.get("downloader_type") self._host = config.get("host") self._port = config.get("port") self._username = config.get("username") self._password = config.get("password") self._target_domain = config.get("target_domain") self._replace_domain = config.get("replace_domain") self._run_con_enable = config.get("run_con_enable") self._run_con = config.get("run_con") self._notify = config.get("notify") if self._onlyonce: # 执行替换 self.task() self._onlyonce = False # 更新onlyonce属性 self.__update_config() def task(self): logger.info(f"{'*' * 30}TrackerEditor: 开始执行Tracker替换{'*' * 30}") torrent_total_cnt: int = 0 torrent_update_cnt: int = 0 if self._downloader_type == "qbittorrent": self._downloader = Qbittorrent(self._host, self._port, self._username, self._password) torrent_info_list: TorrentInfoList torrent_info_list, error = self._downloader.get_torrents() torrent_total_cnt = len(torrent_info_list) if error: return for torrent in torrent_info_list: for tracker in torrent.trackers: if self._target_domain in tracker.url: original_url = tracker.url new_url = tracker.url.replace(self._target_domain, self._replace_domain) logger.info(f"{original_url} 替换为\n {new_url}") torrent.edit_tracker(orig_url=original_url, new_url=new_url) torrent_update_cnt += 1 elif self._downloader_type == "transmission": self._downloader = Transmission(self._host, self._port, self._username, self._password) torrent_list: List[Torrent] torrent_list, error = self._downloader.get_torrents() torrent_total_cnt = len(torrent_list) if error: return for torrent in torrent_list: new_tracker_list = [] for tracker in torrent.tracker_list: new_url = None if self._target_domain in tracker: new_url = tracker.replace(self._target_domain, self._replace_domain) new_tracker_list.append(new_url) else: new_tracker_list.append(tracker) logger.info(f"{tracker} 替换为\n {new_url}") torrent_update_cnt += 1 self._downloader.update_tracker(hash_string=torrent.hashString, tracker_list=new_tracker_list) logger.info(f"{'*' * 30}TrackerEditor: Tracker替换成功{'*' * 30}") if (self._run_con_enable and self._notify) or (self._onlyonce and self._notify): title = '【Tracker替换】' msg = f'''扫描下载器{self._downloader_type}\n总的种子数: {torrent_total_cnt}\n已修改种子数: {torrent_update_cnt}''' self.send_site_message(title, msg) def __update_config(self): self.update_config({ "onlyonce": self._onlyonce, "downloader_type": self._downloader_type, "username": self._username, "password": self._password, "host": self._host, "port": self._port, "target_domain": self._target_domain, "replace_domain": self._replace_domain, "run_cron_enable": self._run_con_enable, "run_cron": self._run_con, "notify": self._notify }) @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': 'run_con_enable', 'label': '启用周期性巡检 (注: 请开启时，务必填写cron表达式)', } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'onlyonce', 'label': '立即运行一次', } } ] }] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'notify', 'label': '发送通知', } } ] }] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'run_con', 'label': 'cron表达式', 'placeholder': '* * * * *' } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VSelect', 'props': { 'model': 'downloader_type', 'label': '下载器类型', 'items': [ {'title': 'Qbittorrent', 'value': 'qbittorrent'}, {'title': 'Transmission', 'value': 'transmission'} ] } } ] }] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'host', 'label': 'host主机ip', 'placeholder': '192.168.2.100' } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'port', 'label': 'qb/tr端口', 'placeholder': '8989' } } ] } ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'username', 'label': '用户名', 'placeholder': 'username' } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'password', 'label': '密码', 'placeholder': 'password' } } ] } ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'target_domain', 'label': '待替换文本', 'placeholder': 'target.com' } } ] }, { 'component': 'VCol', 'props': { 'cols': 12, 'md': 6 }, 'content': [ { 'component': 'VTextField', 'props': { 'model': 'replace_domain', 'label': '替换的文本', 'placeholder': 'replace.net' } } ] } ] }, { 'component': 'VRow', 'content': [ { 'component': 'VCol', 'props': { 'cols': 12, }, 'content': [ { 'component': 'VAlert', 'props': { 'type': 'info', 'variant': 'tonal', 'text': '对下载器中所有符合代替换文本的tacker进行字符串replace替换' + '\n' + '现有tracker: https://baidu.com/announce.php?passkey=xxxx' + '\n' + '待替换 baidu.com 或 https://baidu.com' + '\n' + '用于替换的文本 qq.com 或 https://qq.com' + '\n' + '结果为 https://qq.com/announce.php?passkey=xxxx', 'style': 'white-space: pre-line;' } }, { 'component': 'VAlert', 'props': { 'type': 'info', 'variant': 'tonal', 'text': '强烈建议自己先添加一个tracker测试替换是否符合预期，程序是否正常运行', 'style': 'white-space: pre-line;' } }, { 'component': 'VAlert', 'props': { 'type': 'info', 'variant': 'tonal', 'text': '周期性巡检时指的是允许设置间隔一段进行巡检下载器中的种子Tracker' + '\n' '当匹配到等待替换的tracker时，进行替换，其中cron表达式是5位，例如:* * * * * 指的是每过一分钟轮训一次', 'style': 'white-space: pre-line;' } } ] } ] } ] } ], { "onlyonce": False, "downloader_type": "qbittorrent", "host": "192.168.2.100", "port": 8989, "username": "username", "password": "password", "target_domain": "", "replace_domain": "", "run_con_enable": False, "run_con": "", "notify": True } def get_page(self) -> List[dict]: pass def get_state(self) -> bool: return True def stop_service(self): pass def get_service(self) -> List[Dict[str, Any]]: if self._run_con_enable and self._run_con: logger.info(f"{'*' * 30}TrackerEditor: 注册公共调度服务{'*' * 30}") return [ { "id": "TrackerChangeRun", "name": "启用周期性Tracker替换", "trigger": CronTrigger.from_crontab(self._run_con), "func": self.task, "kwargs": {} }] return [] def send_site_message(self, title, message): self.post_message( mtype=NotificationType.SiteMessage, title=title, text=message )