Evilmass commited on
Commit
858343f
·
1 Parent(s): e694d87

fix tsdm login judgement

Browse files
Files changed (1) hide show
  1. _playwright/tsdm.py +3 -2
_playwright/tsdm.py CHANGED
@@ -35,7 +35,7 @@ async def close_popup(popup):
35
  async def is_login(page):
36
  await page.goto(tsdm_url)
37
  await page.wait_for_load_state()
38
- if await page.get_by_role("link", name="签到领奖!").is_visible():
39
  tsdm_logger.info("登录成功")
40
  return True
41
  else:
@@ -88,7 +88,8 @@ async def cliwork(page):
88
  for pid in tsdm_plugin_id:
89
  await page.locator(f"#np_advid{pid}").get_by_role("link").click()
90
  page.on("popup", close_popup) # handle popup 不能 await
91
- # await page.bring_to_front() # 回到第一个标签页
 
92
  await page.locator("#workstart").click() # 领取奖励
93
  work_res = "打工成功"
94
  tsdm_logger.info(work_res)
 
35
  async def is_login(page):
36
  await page.goto(tsdm_url)
37
  await page.wait_for_load_state()
38
+ if await page.locator("#pm_ntc").is_enabled(): # 消息按钮
39
  tsdm_logger.info("登录成功")
40
  return True
41
  else:
 
88
  for pid in tsdm_plugin_id:
89
  await page.locator(f"#np_advid{pid}").get_by_role("link").click()
90
  page.on("popup", close_popup) # handle popup 不能 await
91
+
92
+ await page.bring_to_front() # 回到第一个标签页
93
  await page.locator("#workstart").click() # 领取奖励
94
  work_res = "打工成功"
95
  tsdm_logger.info(work_res)