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 {
|
func downloadPost(post *e621.Post, directory string) error {
|
||||||
savePath := getSavePath(post, &directory)
|
savePath := getSavePath(post, &directory)
|
||||||
|
|
||||||
fmt.Print(post.File.URL)
|
if _, err := os.Stat(savePath); err == nil {
|
||||||
fmt.Print(savePath)
|
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)
|
defer resp.Body.Close()
|
||||||
if err != nil {
|
|
||||||
return err
|
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
|
return nil
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue