新增判断是否是商家管理员

This commit is contained in:
苏尹岚
2019-12-02 18:14:52 +08:00
parent f91abb83b3
commit 4ff3de2ef9
2 changed files with 33 additions and 9 deletions

View File

@@ -261,13 +261,14 @@ var (
pageExceedLimitCodes = map[string]int{
"403": 1,
}
pageCanRetryCodes = map[string]int{}
regexpTable = regexp.MustCompile(`<table class="check-container" data-container="list1">([\s\S]*?)</table>`)
regexpTd = regexp.MustCompile(`<td>([0-9].*)</td>`)
regexpJDUserPage = regexp.MustCompile(`共([\s\S].*)页/([\s\S].*)条记录`)
regexpJDTr = regexp.MustCompile(`<tr>([\s\S]*?)</tr>`)
regexpJDUserInfo = regexp.MustCompile(`<td>([\s\S]*?)</td>`)
regexpJDUserID = regexp.MustCompile(`value="(.*)"`)
pageCanRetryCodes = map[string]int{}
regexpTable = regexp.MustCompile(`<table class="check-container" data-container="list1">([\s\S]*?)</table>`)
regexpTd = regexp.MustCompile(`<td>([0-9].*)</td>`)
regexpJDUserPage = regexp.MustCompile(`共([\s\S].*)页/([\s\S].*)条记录`)
regexpJDTr = regexp.MustCompile(`<tr>([\s\S]*?)</tr>`)
regexpJDUserInfo = regexp.MustCompile(`<td>([\s\S]*?)</td>`)
regexpJDUserID = regexp.MustCompile(`value="(.*)"`)
regexpJDUserIsManager = regexp.MustCompile(`<div class="list-mask">`)
)
const (
@@ -280,8 +281,6 @@ const (
JdUserRoleJHYName = "【通用】拣货员"
JdUserRolesName = "多角色"
JdUserNoRole = "无角色"
JdUserManager = "jd_jxcs1223"
JdUserManagerGy = "jd_jxgy"
)
func (a *API) SetJdCookie(cookieValue string) {
@@ -674,3 +673,23 @@ func (a *API) UpdateJdUserRoles(id int64, roleIDs []string) (err error) {
_, err = a.AccessStorePage(url, jdParams, true)
return err
}
// 查询用户是否是商家管理员接口
// https://login-o2o.jddj.com/jpuser/store/474147?isShow=1
func (a *API) IsJdManagerUser(id int64) (isManager bool, err error) {
url := "https://login-o2o.jddj.com/jpuser/store/" + utils.Int64ToStr(id)
jdParams := map[string]interface{}{
"isShow": 1,
}
body, err := a.AccessStorePage2(url, jdParams, false, "")
if err != nil {
return isManager, err
}
bodyStr := body.(string)
listmask := regexpJDUserIsManager.FindString(bodyStr)
if listmask == "" {
return false, err
} else {
return true, err
}
}

View File

@@ -213,3 +213,8 @@ func TestPrivilegeUpdateJdUserStatus(t *testing.T) {
func TestUpdateJdUserRoles(t *testing.T) {
api.UpdateJdUserRoles(346568, []string{"28926", "28924"})
}
func TestIsJdManagerUser(t *testing.T) {
aa, _ := api.IsJdManagerUser(334683)
fmt.Println(aa)
}