From 05534c408074d670a3bbf090274a4c6f71ba8038 Mon Sep 17 00:00:00 2001
From: zhao-zg <61549925+zhao-zg@users.noreply.github.com>
Date: Mon, 9 Dec 2024 09:39:17 +0800
Subject: [PATCH] Update login.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

修复短信
---
 login.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/login.py b/login.py
index 145ec3e..e7ec6c1 100644
--- a/login.py
+++ b/login.py
@@ -318,7 +318,7 @@ async def loginPassword(chromium_path, workList, uid, headless):
                 workList[uid].status = "error"
                 workList[uid].msg = "登录超时"
                 logger.info("超时了，正在保存当前页面信息......")
-                dateTime = datetime.datetime.now()
+                dateTime = datetime.datetime.now().strftime('%Y%m%d %H_%M_%S.%f')
                 logger.info(f"页面截图保存到： {usernum}-screenshot-{dateTime}.png")
                 await page.screenshot({'path': f"{usernum}-screenshot-{dateTime}.png"})
                 logger.info(f"页面HTML保存到： {usernum}-html-{dateTime}.html")
@@ -363,7 +363,7 @@ async def loginPassword(chromium_path, workList, uid, headless):
                 elif await page.J('.drag-content'):
                     logger.info("进入旋转图片分支")
                     logger.info("正在保存当前页面信息......")
-                    dateTime = datetime.datetime.now()
+                    dateTime = datetime.datetime.now().strftime('%Y%m%d %H_%M_%S.%f')
                     logger.info(f"页面截图保存到： drag_{usernum}-screenshot-{dateTime}.png")
                     await page.screenshot({'path': f"drag_{usernum}-screenshot-{dateTime}.png"})
                     logger.info("即将重启浏览器重试")
@@ -486,6 +486,14 @@ async def loginPassword(chromium_path, workList, uid, headless):
         except Exception as e:
             logger.info("异常退出")
             logger.error(e)
+            logger.info("异常退出，正在保存当前页面信息......")
+            dateTime = datetime.datetime.now().strftime('%Y%m%d %H_%M_%S.%f')
+            logger.info(f"页面截图保存到： error_{usernum}-screenshot-{dateTime}.png")
+            await page.screenshot({'path': f"error_{usernum}-screenshot-{dateTime}.png"})
+            logger.info(f"页面HTML保存到： error_{usernum}-html-{dateTime}.html")
+            content = await page.content()
+            with open(f"error_{usernum}-html-{dateTime}.html", 'w', encoding='utf-8') as f:
+                f.write(content)
             workList[uid].status = "error"
             workList[uid].msg = "异常退出"
             break
@@ -588,7 +596,7 @@ async def sendSMSDirectly(page):
 
     try:
         while True:
-            if await page.xpath('//*[@id="captcha_modal"]/div/div[3]/div'):
+            if await page.xpath('//*[@id="small_img"]'):
                 await verification(page)
 
             elif await page.xpath('//*[@id="captcha_modal"]/div/div[3]/button'):
@@ -630,7 +638,7 @@ async def sendSMS(page):
 
     try:
         while True:
-            if await page.xpath('//*[@id="captcha_modal"]/div/div[3]/div'):
+            if await page.xpath('//*[@id="small_img"]'):
                 await verification(page)
 
             elif await page.xpath('//*[@id="captcha_modal"]/div/div[3]/button'):