Commit cd6f1f5

author

Openset

committed

Add: checkErr

1 parent db96ead commit cd6f1f5Copy full SHA for cd6f1f5

File tree

-7

lines changed

-7

lines changed

Lines changed: 5 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -30,13 +30,15 @@ func getPath(f string) string {`
`30`	`30`	`if err == nil && u.HomeDir != "" {`
`31`	`31`	`dir = path.Join(u.HomeDir, ".leetcode")`
`32`	`32`	`}`
`33`		`- os.MkdirAll(dir, 0755)`
	`33`	`+ err = os.MkdirAll(dir, 0755)`
	`34`	`+ checkErr(err)`
`34`	`35`	`return path.Join(dir, f)`
`35`	`36`	`}`
`36`	`37`
`37`	`38`	`func saveCookies(cookies []*http.Cookie) {`
`38`		`- data, _ := json.Marshal(cookies)`
`39`		`- err := ioutil.WriteFile(getPath(cookiesFile), data, 0755)`
	`39`	`+ data, err := json.Marshal(cookies)`
	`40`	`+ checkErr(err)`
	`41`	`+ err = ioutil.WriteFile(getPath(cookiesFile), data, 0755)`
`40`	`42`	`checkErr(err)`
`41`	`43`	`}`
`42`	`44`

Lines changed: 3 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,8 @@ import (`
`7`	`7`	`)`
`8`	`8`
`9`	`9`	`func AccountsLogin(user string, pwd string) (*http.Response, error) {`
`10`		`- resp, _ := http.Head(AccountsLoginUrl)`
	`10`	`+ resp, err := http.Head(AccountsLoginUrl)`
	`11`	`+ checkErr(err)`
`11`	`12`	`cookies := resp.Cookies()`
`12`	`13`	`saveCookies(cookies)`
`13`	`14`	`csrftoken := getCsrfToken(cookies)`
`@@ -17,9 +18,7 @@ func AccountsLogin(user string, pwd string) (*http.Response, error) {`
`17`	`18`	`"csrfmiddlewaretoken": {csrftoken},`
`18`	`19`	`}`
`19`	`20`	`req, err := http.NewRequest("POST", AccountsLoginUrl, strings.NewReader(data.Encode()))`
`20`		`- if err != nil {`
`21`		`- return nil, err`
`22`		`- }`
	`21`	`+ checkErr(err)`
`23`	`22`	`req.Header.Set("Content-Type", "application/x-www-form-urlencoded")`
`24`	`23`	`req.Header.Set("Referer", AccountsLoginUrl)`
`25`	`24`	`for _, cookie := range cookies {`

Comments

(0)