diff --git a/plugins/jd_sign_token/main.py b/plugins/jd_sign_token/main.py index 33f12e0..53e9519 100644 --- a/plugins/jd_sign_token/main.py +++ b/plugins/jd_sign_token/main.py @@ -56,7 +56,8 @@ class QL: url = f"{self.address}/open/envs?searchValue=" headers = {"Authorization": self.auth} try: - async with aiohttp.ClientSession().get(url, headers=headers, timeout=aiohttp.ClientTimeout(total=15)) as response: + async with aiohttp.ClientSession().get(url, headers=headers, + timeout=aiohttp.ClientTimeout(total=15)) as response: if response.status == 200: rjson = await response.json() if rjson['code'] == 200: @@ -81,7 +82,7 @@ class QL: headers = {"Authorization": self.auth, "content-type": "application/json"} try: async with aiohttp.ClientSession().delete(url, headers=headers, json=ids, - timeout=aiohttp.ClientTimeout(total=15)) as response: + timeout=aiohttp.ClientTimeout(total=15)) as response: if response.status == 200: rjson = await response.json() if rjson['code'] == 200: @@ -106,7 +107,7 @@ class QL: headers = {"Authorization": self.auth, "content-type": "application/json"} try: async with aiohttp.ClientSession().post(url, headers=headers, json=envs, - timeout=aiohttp.ClientTimeout(total=15)) as response: + timeout=aiohttp.ClientTimeout(total=15)) as response: if response.status == 200: rjson = await response.json() if rjson['code'] == 200: @@ -130,7 +131,8 @@ class QL: url = f"{self.address}/open/envs" headers = {"Authorization": self.auth, "content-type": "application/json"} try: - async with aiohttp.ClientSession().put(url, headers=headers, json=env, timeout=aiohttp.ClientTimeout(total=15)) as response: + async with aiohttp.ClientSession().put(url, headers=headers, json=env, + timeout=aiohttp.ClientTimeout(total=15)) as response: if response.status == 200: rjson = await response.json() if rjson['code'] == 200: @@ -279,7 +281,7 @@ class JDTokenPlugin(MessagePluginInterface): self.LOG.info(f"处理后的token格式: {token}") try: # 设置京东Token - result = self.set_jd_token(token, remark) + result = await self.set_jd_token(token, remark) await bot.send_text_message((roomid if roomid else sender), result, sender) return True, "处理成功" @@ -288,13 +290,13 @@ class JDTokenPlugin(MessagePluginInterface): await bot.send_text_message((roomid if roomid else sender), f"❌处理出错: {str(e)}", sender) return False, f"处理出错: {e}" - def set_jd_token(self, token: str, remark: str) -> str: + async def set_jd_token(self, token: str, remark: str) -> str: """设置京东Token""" if not self.ql.valid: return f"❌ 青龙面板连接失败,请检查配置" # 检查是否已存在相同备注的环境变量 - envs = self.ql.getEnvs() + envs = await self.ql.getEnvs() if not envs: return f"❌ 获取环境变量失败" @@ -324,10 +326,10 @@ class JDTokenPlugin(MessagePluginInterface): # 更新已存在的环境变量,保留原有的备注或使用新的备注 existing_remarks = existing_env.get('remarks', '') final_remarks = remark if remark else existing_remarks - + existing_env['value'] = token env_update: dict = {"id": env_id, "value": token, "remarks": final_remarks, "name": "JD_COOKIE"} - result = self.ql.updateEnv(env_update) + result = await self.ql.updateEnv(env_update) if result: return f"✅ 已成功更新京东账号 [{final_remarks}] 的Token (pt_pin: {current_pt_pin})" else: @@ -335,7 +337,7 @@ class JDTokenPlugin(MessagePluginInterface): else: # 添加新的环境变量 new_env = [{"name": "JD_COOKIE", "value": token, "remarks": remark}] - result = self.ql.addEnvs(new_env) + result = await self.ql.addEnvs(new_env) if result: return f"✅ 已成功添加京东账号 [{remark}] 的Token (pt_pin: {current_pt_pin})" else: