Update download.go
This commit is contained in:
parent
0707246c77
commit
4088e2baf1
|
@ -121,25 +121,28 @@ func getSavePath(post *e621.Post, directory *string) string {
|
|||
func downloadPost(post *e621.Post, directory string) error {
|
||||
savePath := getSavePath(post, &directory)
|
||||
|
||||
fmt.Print(post.File.URL)
|
||||
fmt.Print(savePath)
|
||||
if _, err := os.Stat(savePath); err == nil {
|
||||
fmt.Print("File exists, skip...\n")
|
||||
}
|
||||
else {
|
||||
resp, err := e621.HTTPGet(post.File.URL)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
resp, err := e621.HTTPGet(post.File.URL)
|
||||
if err != nil {
|
||||
return err
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = ioutil.WriteFile(savePath, body, 0755)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = ioutil.WriteFile(savePath, body, 0755)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue