From 35ba1b3345eeb1d680d5ab4c86cd6ae72ef623e1 Mon Sep 17 00:00:00 2001
From: Soulter <905617992@qq.com>
Date: Sat, 22 Feb 2025 11:37:34 +0800
Subject: [PATCH] fix: gewechat verify code

---
 astrbot/core/platform/sources/gewechat/client.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/astrbot/core/platform/sources/gewechat/client.py b/astrbot/core/platform/sources/gewechat/client.py
index 1ed0ab110..23021f1c4 100644
--- a/astrbot/core/platform/sources/gewechat/client.py
+++ b/astrbot/core/platform/sources/gewechat/client.py
@@ -5,6 +5,7 @@ import quart
 import base64
 import datetime
 import re
+import os
 from astrbot.api.platform import AstrBotMessage, MessageMember, MessageType
 from astrbot.api.message_components import Plain, Image, At, Record
 from astrbot.api import logger, sp
@@ -307,7 +308,7 @@ class SimpleGewechatClient():
             retry_cnt -= 1
 
             # 需要验证码
-            if verify_flag:
+            if verify_flag or os.path.exists("data/temp/gewe_code"):
                 with open("data/temp/gewe_code", "r") as f:
                     code = f.read().strip()
                     if not code:
@@ -316,6 +317,10 @@ class SimpleGewechatClient():
                         continue
                     payload['captchCode'] = code
                     logger.info(f"使用验证码: {code}")
+                    try:
+                        os.remove("data/temp/gewe_code")
+                    except:
+                        logger.warning("删除验证码文件 data/temp/gewe_code 失败。")
 
             async with aiohttp.ClientSession() as session:
                 async with session.post(