From cd7aa936534ed161d74d8ffbd8b7b42ffe9e41bc Mon Sep 17 00:00:00 2001
From: zhao-zg <61549925+zhao-zg@users.noreply.github.com>
Date: Tue, 30 Jul 2024 11:45:51 +0800
Subject: [PATCH] Update login.py

---
 login.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/login.py b/login.py
index bd02a42..e94ca10 100644
--- a/login.py
+++ b/login.py
@@ -164,9 +164,9 @@ async def loginPhone(chromium_path, workList, uid, headless):
                 workList[uid].msg = "账号异常，自行检查"
                 break
             if False == sms_sent:
-                button = await page.querySelector('.getMsg-btn.text-btn')
+                button = await page.querySelector('.getMsg-btn.text-btn.timer.active')
                 print(button.evaluate("textContent"))
-                if "获取验证码" not in button.evaluate("textContent"):
+                if button and "获取验证码" not in button.evaluate("textContent"):
                     print("进入直接发短信分支")
                     if not workList[uid].isAuto:
                         workList[uid].status = "SMS"
@@ -197,7 +197,6 @@ async def loginPhone(chromium_path, workList, uid, headless):
 
             await asyncio.sleep(1)
         except Exception as e:
-            continue
             print("异常退出")
             print(e)
             await browser.close()