diff --git a/plugins/jd_sign_token/main.py b/plugins/jd_sign_token/main.py index 298a126..9fdf17c 100644 --- a/plugins/jd_sign_token/main.py +++ b/plugins/jd_sign_token/main.py @@ -158,12 +158,12 @@ class JDTokenPlugin(MessagePluginInterface): self._commands = self._config.get("JD_Token", {}).get("command", ["设置京东"]) self.command_format = self._config.get("JD_Token", {}).get("command-format", "设置京东 token内容 备注名称") self.enable = self._config.get("JD_Token", {}).get("enable", True) - + # 从TOML配置文件加载青龙面板配置 ql_host = self._config.get("JD_Token", {}).get("QL_HOST", "http://localhost:5700") client_id = self._config.get("JD_Token", {}).get("CLIENT_ID", "") client_secret = self._config.get("JD_Token", {}).get("CLIENT_SECRET", "") - + # 初始化青龙面板连接 self.ql = QL(ql_host, client_id, client_secret) @@ -206,7 +206,7 @@ class JDTokenPlugin(MessagePluginInterface): # 检查命令格式 pattern = r'^设置京东\s+([^\s]+)\s+(.+)$' match = re.match(pattern, content) - + if not match: await bot.send_text_message((roomid if roomid else sender), f"❌命令格式错误!\n{self.command_format}" , sender) @@ -219,12 +219,12 @@ class JDTokenPlugin(MessagePluginInterface): # 提取token和备注 token = match.group(1) remark = match.group(2) - + # 简单预检查token格式 if not token.startswith("pt_key=") or "pt_pin=" not in token: - await bot.send_text_message((roomid if roomid else sender), - f"❌ Token格式错误!正确格式应为:pt_key=xxx;pt_pin=xxx;", - sender) + await bot.send_text_message((roomid if roomid else sender), + f"❌ Token格式错误!正确格式应为:pt_key=xxx;pt_pin=xxx;", + sender) return False, "Token格式错误" try: @@ -242,19 +242,19 @@ class JDTokenPlugin(MessagePluginInterface): """设置京东Token""" if not self.ql.valid: return f"❌ 青龙面板连接失败,请检查配置" - + # 检查是否已存在相同备注的环境变量 envs = self.ql.getEnvs() if not envs: return f"❌ 获取环境变量失败" - + # 查找是否有相同备注的JD_COOKIE existing_env = None for env in envs: if env.get('name') == 'JD_COOKIE' and env.get('remarks') == remark: existing_env = env break - + result = False if existing_env: # 更新已存在的环境变量 @@ -271,4 +271,4 @@ class JDTokenPlugin(MessagePluginInterface): if result: return f"✅ 已成功添加京东账号 [{remark}] 的Token" else: - return f"❌ 添加京东账号 [{remark}] 的Token失败" \ No newline at end of file + return f"❌ 添加京东账号 [{remark}] 的Token失败"