Improved import

This commit is contained in:
2017-07-02 01:44:10 +02:00
parent 3309e5fb63
commit f9bbf7e6db
3 changed files with 52 additions and 73 deletions

View File

@ -0,0 +1,42 @@
package main
import (
"fmt"
"github.com/animenotifier/arn"
"github.com/fatih/color"
)
var userName = "Akyoto"
var allAnime []*arn.Anime
func init() {
allAnime, _ = arn.AllAnime()
}
func main() {
arn.PanicOnError(arn.AniList.Authorize())
println(arn.AniList.AccessToken)
user, _ := arn.GetUserByNick(userName)
animeList, err := arn.AniList.GetAnimeList(user)
arn.PanicOnError(err)
importList(animeList.Lists.Watching)
importList(animeList.Lists.Completed)
}
func importList(animeListItems []*arn.AniListAnimeListItem) {
imported := []*arn.Anime{}
for _, item := range animeListItems {
anime := arn.FindAniListAnime(item.Anime, allAnime)
if anime != nil {
fmt.Println(item.Anime.TitleRomaji, "=>", anime.Title.Romaji)
imported = append(imported, anime)
}
}
color.Green("%d / %d", len(imported), len(animeListItems))
}

View File

@ -0,0 +1,73 @@
package main
import (
"encoding/json"
"fmt"
"io/ioutil"
"strconv"
"github.com/animenotifier/arn"
)
// OldMatch ...
type OldMatch struct {
ID int `json:"id"`
ServiceID int `json:"providerId"`
Title string `json:"title"`
ServiceTitle string `json:"providerTitle"`
Similarity float64 `json:"similarity"`
Edited string `json:"edited"`
EditedBy string `json:"editedBy"`
}
func main() {
matches := []OldMatch{}
data, _ := ioutil.ReadFile("MatchKitsu.json")
json.Unmarshal(data, &matches)
for _, match := range matches {
// Custom anime in 3.0
if match.ID >= 1000000 {
continue
}
// New match type
newMatch := &arn.AniListToAnime{
AnimeID: strconv.Itoa(match.ServiceID),
ServiceID: strconv.Itoa(match.ID),
Similarity: match.Similarity,
Edited: match.Edited,
EditedBy: match.EditedBy,
}
// Get anime
anime, err := arn.GetAnime(newMatch.AnimeID)
if err != nil {
continue
}
if anime.GetMapping("anilist/anime") != "" {
continue
}
anime.Mappings = append(anime.Mappings, &arn.Mapping{
Service: "anilist/anime",
ServiceID: newMatch.ServiceID,
Created: newMatch.Edited,
CreatedBy: newMatch.EditedBy,
})
// Save
fmt.Println(anime.Title.Canonical)
arn.PanicOnError(anime.Save())
arn.PanicOnError(arn.DB.Set("AniListToAnime", newMatch.ServiceID, newMatch))
}
}
// AnilistToAnime
/*
AnimeID
ServiceID
*/