60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"sort"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/aerogo/http/client"
|
||
|
|
"github.com/animenotifier/arn"
|
||
|
|
"github.com/fatih/color"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
color.Yellow("Updating anime ratings")
|
||
|
|
|
||
|
|
defer color.Green("Finished.")
|
||
|
|
defer arn.Node.Close()
|
||
|
|
|
||
|
|
characters := arn.FilterCharacters(func(character *arn.Character) bool {
|
||
|
|
return character.Image.LastModified == 0
|
||
|
|
})
|
||
|
|
|
||
|
|
sort.Slice(characters, func(i, j int) bool {
|
||
|
|
return characters[i].ID < characters[j].ID
|
||
|
|
})
|
||
|
|
|
||
|
|
for index, character := range characters {
|
||
|
|
fmt.Printf("[%d / %d] %s %s\n", index+1, len(characters), character.ID, color.CyanString(character.String()))
|
||
|
|
download(character.ID)
|
||
|
|
}
|
||
|
|
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
}
|
||
|
|
|
||
|
|
func download(characterID string) {
|
||
|
|
character, err := arn.GetCharacter(characterID)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
color.Red(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
url := fmt.Sprintf("https://media.kitsu.io/characters/images/%s/original%s", character.GetMapping("kitsu/character"), character.Image.Extension)
|
||
|
|
response, err := client.Get(url).End()
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
color.Red(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
err = character.SetImageBytes(response.Bytes())
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
color.Red(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
character.Save()
|
||
|
|
}
|