2018-03-08 23:07:48 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"strings"
|
|
|
|
|
2018-03-09 03:32:31 +00:00
|
|
|
"github.com/animenotifier/arn"
|
2018-03-08 23:07:48 +00:00
|
|
|
"github.com/animenotifier/mal/parser"
|
2018-03-16 23:28:36 +00:00
|
|
|
"github.com/fatih/color"
|
2018-03-08 23:07:48 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
2018-03-16 23:28:36 +00:00
|
|
|
color.Yellow("Importing MAL anime")
|
|
|
|
|
|
|
|
defer color.Green("Finished.")
|
2018-03-21 04:15:03 +00:00
|
|
|
defer arn.Node.Close()
|
2018-03-08 23:07:48 +00:00
|
|
|
|
2018-03-16 23:28:36 +00:00
|
|
|
// readFile("../mal-download/files/anime-31240.html")
|
2018-03-08 23:07:48 +00:00
|
|
|
|
2018-03-16 23:28:36 +00:00
|
|
|
filepath.Walk("../mal-download/files", func(name string, info os.FileInfo, err error) error {
|
2018-03-08 23:07:48 +00:00
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if info.IsDir() {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
if !strings.HasSuffix(name, ".html") {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
return readFile(name)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func readFile(name string) error {
|
|
|
|
file, err := os.Open(name)
|
|
|
|
defer file.Close()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
anime, err := malparser.ParseAnime(file)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
fmt.Println(err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if anime.ID == "" {
|
|
|
|
return errors.New("Empty ID")
|
|
|
|
}
|
|
|
|
|
2018-03-09 23:25:37 +00:00
|
|
|
// fmt.Println(anime.ID, anime.Title)
|
2018-03-09 03:32:31 +00:00
|
|
|
arn.MAL.Set("Anime", anime.ID, anime)
|
2018-03-08 23:07:48 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// prettyPrint prints the object as indented JSON data on the console.
|
|
|
|
func prettyPrint(obj interface{}) {
|
|
|
|
pretty, _ := json.MarshalIndent(obj, "", "\t")
|
|
|
|
fmt.Println(string(pretty))
|
|
|
|
}
|