1 files changed, 23 insertions, 0 deletions

diff --git a/tools/importer/main.go b/tools/importer/main.go
index fd7af37..4eba9a0 100644
--- a/tools/importer/main.go
+++ b/tools/importer/main.go
@@ -179,6 +179,28 @@ func InsertTraits(dbID int, g WrappedGril, traitlist map[string]int) {
                 LogErr(err)
         }
 }
+
+func InsertSeries(dbID int, g WrappedGril) {
+        rows, err := db.Query("SELECT appearance_id FROM grils_appearances WHERE gril_id = $1;", dbID)
+        LogErr(err)
+        existingSeries := make(map[int]bool, 0)
+        for rows.Next() {
+                var seriesID int
+                rows.Scan(&seriesID)
+                existingSeries[seriesID] = true
+        }
+        for _, v := range g.Gril.Series {
+                if _, ok := existingSeries[v.ID]; ok {
+                        continue
+                }
+                _, err := db.Exec("INSERT INTO grilist.grils_appearances (gril_id, appearance_id) VALUES ($1, $2);",
+                        dbID, v.ID)
+                existingSeries[v.ID] = true
+                fmt.Printf("Assigned series %d to %s\n", v.ID, g.Gril.RomajiName)
+                LogErr(err)
+        }
+}
+
 func main() {
         flag.StringVar(&path, "path", "", "path of the source files")
         flag.StringVar(&sourceType, "type", "", "type of the files (ACD, AniDB)")
@@ -252,6 +274,7 @@ func main() {
                 }
                 InsertTags(dbID, g, taglist)
                 InsertTraits(dbID, g, traitlist)
+                InsertSeries(dbID, g)
 
                 return nil
         })