chickliu/jd-login
1
0
Fork 0
mirror of https://github.com/zhao-zg/jd-login.git synced 2026-01-14 15:00:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update login.py

Browse Source 
超时问题修复
This commit is contained in:
zhao-zg 2024-12-16 15:51:31 +08:00 committed by GitHub
parent cbb57ccfea
commit d50a32cc3e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
login.py
View File

@ -51,6 +51,13 @@ supported_colors = {
"红色": ([0, 50, 50], [10, 255, 255]), "红色": ([0, 50, 50], [10, 255, 255]),
} }
async def deleteSession(workList, uid):
s = workList.get(uid, "")
if s:
await asyncio.sleep(15)
del workList[uid]
async def loginPhone(chromium_path, workList, uid, headless): async def loginPhone(chromium_path, workList, uid, headless):
# 判断账号密码错误 # 判断账号密码错误
async def isWrongAccountOrPassword(page, verify=False): async def isWrongAccountOrPassword(page, verify=False):
@ -205,8 +212,7 @@ async def loginPhone(chromium_path, workList, uid, headless):
except Exception as e: except Exception as e:
logger.info("异常退出") logger.info("异常退出")
logger.error(e) logger.error(e)
workList[uid].status = "error" await deleteSession(workList, uid)
workList[uid].msg = "异常退出"
break break
logger.info("任务完成退出") logger.info("任务完成退出")
@ -495,8 +501,7 @@ async def loginPassword(chromium_path, workList, uid, headless):
content = await page.content() content = await page.content()
with open(f"error_{usernum}-html-{dateTime}.html", 'w', encoding='utf-8') as f: with open(f"error_{usernum}-html-{dateTime}.html", 'w', encoding='utf-8') as f:
f.write(content) f.write(content)
workList[uid].status = "error" await deleteSession(workList, uid)
workList[uid].msg = "异常退出"
break break
logger.info("任务完成退出") logger.info("任务完成退出")
@ -1200,5 +1205,16 @@ async def main(workList, uid, oocr, oocrDet):
await asyncio.sleep(random.uniform(2, 4)) await asyncio.sleep(random.uniform(2, 4))
logger.info(f"进行第{try_time}次重试") logger.info(f"进行第{try_time}次重试")
try_time += 1 try_time += 1
if os.path.exists("image.png"):
os.remove("image.png")
if os.path.exists("template.png"):
os.remove("template.png")
if os.path.exists("shape_image.png"):
os.remove("shape_image.png")
if os.path.exists("rgba_word_img.png"):
os.remove("rgba_word_img.png")
if os.path.exists("rgb_word_img.png"):
os.remove("rgb_word_img.png")
await deleteSession(workList, uid)
logger.info("登录完成") logger.info("登录完成")
await asyncio.sleep(10) await asyncio.sleep(10)