Improved low resolution search

This commit is contained in:
2018-03-20 01:43:23 +01:00
parent bf118c5f10
commit 947752333d
3 changed files with 102 additions and 56 deletions

View File

@ -1,46 +1,60 @@
package main
import (
"fmt"
"io/ioutil"
"os"
"path"
"github.com/animenotifier/arn"
"github.com/fatih/color"
)
func main() {
color.Yellow("Updating anime thumbnails")
defer arn.Node.Close()
defer color.Green("Finished.")
for anime := range arn.StreamAnime() {
base := path.Join(arn.Root, "/images/anime/original/", anime.ID)
if _, err := os.Stat(base + ".png"); err == nil {
anime.Image.Extension = ".png"
anime.Save()
update(anime, base+".png")
continue
}
if _, err := os.Stat(base + ".jpg"); err == nil {
anime.Image.Extension = ".jpg"
anime.Save()
update(anime, base+".jpg")
continue
}
if _, err := os.Stat(base + ".jpeg"); err == nil {
anime.Image.Extension = ".jpg"
anime.Save()
update(anime, base+".jpg")
continue
}
if _, err := os.Stat(base + ".gif"); err == nil {
anime.Image.Extension = ".gif"
anime.Save()
update(anime, base+".gif")
continue
}
if _, err := os.Stat(base + ".webp"); err == nil {
anime.Image.Extension = ".webp"
anime.Save()
update(anime, base+".webp")
continue
}
}
}
func update(anime *arn.Anime, filePath string) {
fmt.Println(anime.ID, anime)
data, err := ioutil.ReadFile(filePath)
if err != nil {
return
}
anime.SetImageBytes(data)
anime.Save()
}