🐛 fix pikpak and shandian account status

This commit is contained in:
微凉 2021-12-30 17:59:35 +08:00
parent 3331462229
commit 5f2621eca9
2 changed files with 15 additions and 8 deletions

View File

@ -35,17 +35,19 @@ func (driver PikPak) Login(account *model.Account) error {
}).Post(url)
if err != nil {
account.Status = err.Error()
_ = model.SaveAccount(account)
return err
}
log.Debug(res.String())
if e.ErrorCode != 0 {
account.Status = e.Error
return errors.New(e.Error)
err = errors.New(e.Error)
} else {
data := res.Body()
account.Status = "work"
account.RefreshToken = jsoniter.Get(data, "refresh_token").ToString()
account.AccessToken = jsoniter.Get(data, "access_token").ToString()
}
data := res.Body()
account.Status = "work"
account.RefreshToken = jsoniter.Get(data, "refresh_token").ToString()
account.AccessToken = jsoniter.Get(data, "access_token").ToString()
_ = model.SaveAccount(account)
return nil
}
@ -72,6 +74,8 @@ func (driver PikPak) RefreshToken(account *model.Account) error {
// refresh_token 失效,重新登陆
return driver.Login(account)
}
account.Status = e.Error
_ = model.SaveAccount(account)
return errors.New(e.Error)
}
data := res.Body()

View File

@ -34,11 +34,14 @@ func (driver Shandian) Login(account *model.Account) error {
return err
}
if resp.Code != 0 {
return errors.New(resp.Msg)
account.Status = resp.Msg
err = errors.New(resp.Msg)
} else {
account.Status = "work"
account.AccessToken = resp.Data.Token
}
account.AccessToken = resp.Data.Token
_ = model.SaveAccount(account)
return nil
return err
}
type File struct {