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

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{ pageExceedLimitCodes = map[string]int{
"403": 1, "403": 1,
} }
pageCanRetryCodes = map[string]int{} pageCanRetryCodes = map[string]int{}
regexpTable = regexp.MustCompile(`<table class="check-container" data-container="list1">([\s\S]*?)</table>`) regexpTable = regexp.MustCompile(`<table class="check-container" data-container="list1">([\s\S]*?)</table>`)
regexpTd = regexp.MustCompile(`<td>([0-9].*)</td>`) regexpTd = regexp.MustCompile(`<td>([0-9].*)</td>`)
regexpJDUserPage = regexp.MustCompile(`共([\s\S].*)页/([\s\S].*)条记录`) regexpJDUserPage = regexp.MustCompile(`共([\s\S].*)页/([\s\S].*)条记录`)
regexpJDTr = regexp.MustCompile(`<tr>([\s\S]*?)</tr>`) regexpJDTr = regexp.MustCompile(`<tr>([\s\S]*?)</tr>`)
regexpJDUserInfo = regexp.MustCompile(`<td>([\s\S]*?)</td>`) regexpJDUserInfo = regexp.MustCompile(`<td>([\s\S]*?)</td>`)
regexpJDUserID = regexp.MustCompile(`value="(.*)"`) regexpJDUserID = regexp.MustCompile(`value="(.*)"`)
regexpJDUserIsManager = regexp.MustCompile(`<div class="list-mask">`)
) )
const ( const (
@@ -280,8 +281,6 @@ const (
JdUserRoleJHYName = "【通用】拣货员" JdUserRoleJHYName = "【通用】拣货员"
JdUserRolesName = "多角色" JdUserRolesName = "多角色"
JdUserNoRole = "无角色" JdUserNoRole = "无角色"
JdUserManager = "jd_jxcs1223"
JdUserManagerGy = "jd_jxgy"
) )
func (a *API) SetJdCookie(cookieValue string) { 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) _, err = a.AccessStorePage(url, jdParams, true)
return err 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) { func TestUpdateJdUserRoles(t *testing.T) {
api.UpdateJdUserRoles(346568, []string{"28926", "28924"}) api.UpdateJdUserRoles(346568, []string{"28926", "28924"})
} }
func TestIsJdManagerUser(t *testing.T) {
aa, _ := api.IsJdManagerUser(334683)
fmt.Println(aa)
}