CheckLogin

common/rod_utils.go

@@ -692,3 +692,39 @@ func InputElementXStrOrNo(p *rod.Page, xPath, inputVal string) {
 	}
 	elementX.MustSelectAllText().MustInput(inputVal)
 }
+
+func CheckLogin(page *rod.Page) error {
+	for i := 0; i < 30; i++ {
+		if MustHasXV1(page, `//div[@class='leftMain']//div[contains(@class,'title')]`) {
+			break
+		}
+		if MustHasXV1(page, `//div[@aria-label="密码修改"]`) {
+			return taxerr.NewUserV3("税局强制要求修改密码", "请修改后重新发起")
+		}
+		if MustHasXV1(page, `//div[text()="账户中心"]`) {
+			return taxerr.NewUserV3("登录后不能进入税局首页", "请检查后重新发起")
+		}
+		if MustHasXV1(page, `//div[@aria-label="用户协议"]//span[text()="确认"]`) {
+			MustClick(page, `//div[@aria-label="用户协议"]//span[text()="确认"]`)
+		}
+		if MustHasXV1(page, `//div[contains(text(),"为提升纳税人办税体验")]/../..//span[text()="确 定"]`) {
+			MustClick(page, `//div[contains(text(),"为提升纳税人办税体验")]/../..//span[text()="确 定"]`)
+		}
+		utils.Sleep(1)
+	}
+	if !MustHasXV1(page, `//div[@class='leftMain']//div[contains(@class,'title')]`) {
+		return taxerr.NewSystemV3("税局登录异常", "请稍后重试")
+	}
+	i := 0
+	for {
+		name := MustText(page, `//div[@class='leftMain']//div[contains(@class,'title')]`)
+		if name != "" {
+			return nil
+		}
+		utils.Sleep(1)
+		if i > 30 {
+			return taxerr.NewSystemV3("税局登录异常", "请稍后重试")
+		}
+		i++
+	}
+}