Updated MAL sync

This commit is contained in:
2018-11-01 08:23:41 +09:00
parent 3ccf470cf0
commit 9d57aaf12a
5 changed files with 169 additions and 124 deletions

View File

@ -1,17 +1,12 @@
package main
import (
"compress/gzip"
"errors"
"os"
"path"
"path/filepath"
"strings"
"github.com/animenotifier/mal"
"github.com/animenotifier/arn"
"github.com/animenotifier/mal/parser"
"github.com/fatih/color"
)
@ -35,6 +30,7 @@ func main() {
}
}
// Read files in a given directory and apply a function on them
func readFiles(root string, onFile func(string) error) {
count := 0
@ -65,95 +61,3 @@ func readFiles(root string, onFile func(string) error) {
color.Cyan("%d files found", count)
}
func readAnimeFile(name string) error {
file, err := os.Open(name)
if err != nil {
color.Red(err.Error())
return err
}
defer file.Close()
reader, err := gzip.NewReader(file)
if err != nil {
color.Red(err.Error())
return err
}
anime, characters, err := malparser.ParseAnime(reader)
if err != nil {
color.Red(err.Error())
return err
}
if anime.ID == "" {
return errors.New("Empty ID")
}
for _, character := range characters {
obj, err := arn.MAL.Get("Character", character.ID)
if err != nil {
arn.MAL.Set("Character", character.ID, character)
continue
}
existing := obj.(*mal.Character)
modified := false
if existing.Name != character.Name {
existing.Name = character.Name
modified = true
}
if existing.Image != character.Image {
existing.Image = character.Image
modified = true
}
if modified {
arn.MAL.Set("Character", existing.ID, existing)
}
}
// fmt.Println(anime.ID, anime.Title)
arn.MAL.Set("Anime", anime.ID, anime)
return nil
}
func readCharacterFile(name string) error {
file, err := os.Open(name)
if err != nil {
color.Red(err.Error())
return err
}
defer file.Close()
reader, err := gzip.NewReader(file)
if err != nil {
color.Red(err.Error())
return err
}
character, err := malparser.ParseCharacter(reader)
if err != nil {
color.Red(err.Error())
return err
}
if character.ID == "" {
return errors.New("Empty ID")
}
// fmt.Println(character.ID, character.Name)
arn.MAL.Set("Character", character.ID, character)
return nil
}