1 files changed, 53 insertions, 1 deletions

diff --git a/tools/importer/main.go b/tools/importer/main.go
index ba98fb5..13ed0cb 100644
--- a/tools/importer/main.go
+++ b/tools/importer/main.go
@@ -108,7 +108,8 @@ func InsertGril(g WrappedGril) int {
 
 func InsertTags(dbID int, g WrappedGril, taglist map[string]int) {
         for _, v := range g.Gril.Tags {
-                if _, ok := taglist[v]; ok {
+                v := filterTags(v)
+                if _, ok := taglist[v]; ok || v == "" {
                         continue
                 }
                 var id int
@@ -126,16 +127,58 @@ func InsertTags(dbID int, g WrappedGril, taglist map[string]int) {
                 existingTags[tagID] = true
         }
         for _, v := range g.Gril.Tags {
+                v = filterTags(v)
+                if v == "" {
+                        continue
+                }
                 tagID := taglist[v]
                 if _, ok := existingTags[tagID]; ok {
                         continue
                 }
                 _, err := db.Exec(`INSERT INTO grilist.grils_tags (gril_id, tag_id) VALUES ($1, $2);`, dbID, tagID)
+                existingTags[tagID] = true
                 fmt.Printf("Assigned tag %s to %s\n", v, g.Gril.RomajiName)
                 LogErr(err)
         }
 }
 
+func InsertTraits(dbID int, g WrappedGril, traitlist map[string]int) {
+        for _, v := range g.Gril.Traits {
+                v.Name = filterTraits(v.Name)
+                if _, ok := traitlist[v.Name]; ok || v.Name == "" {
+                        continue
+                }
+                var id int
+                row := db.QueryRow("INSERT INTO grilist.traits (name) VALUES ($1) RETURNING id;", v.Name)
+                row.Scan(&id)
+                traitlist[v.Name] = id
+                fmt.Printf("Inserted trait %s as %d\n", v.Name, id)
+        }
+        rows, err := db.Query("SELECT trait_id FROM grils_traits WHERE gril_id = $1;", dbID)
+        LogErr(err)
+        existingTraits := make(map[int]bool, 0)
+        for rows.Next() {
+                var traitID int
+                rows.Scan(&traitID)
+                existingTraits[traitID] = true
+        }
+        for _, v := range g.Gril.Traits {
+                v.Name = filterTraits(v.Name)
+                if v.Name == "" {
+                        continue
+                }
+                traitID := traitlist[v.Name]
+                if _, ok := existingTraits[traitID]; ok {
+                        continue
+                }
+                _, err := db.Exec(`INSERT INTO grilist.grils_traits (gril_id,
+                        trait_id, value, official) VALUES ($1, $2, $3, $4);`, dbID, traitID,
+                        v.Value, v.OfficialValue)
+                existingTraits[traitID] = true
+                fmt.Printf("Assigned trait %s to %s\n", v, 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)")
@@ -180,6 +223,14 @@ func main() {
                 rows.Scan(&id, &name)
                 taglist[name] = id
         }
+        traitlist := make(map[string]int)
+        rows, _ = db.Query("SELECT id, name FROM grilist.traits;")
+        for rows.Next() {
+                var id int
+                var name string
+                rows.Scan(&id, &name)
+                traitlist[name] = id
+        }
         idmap := make(map[int]int)
         rows, _ = db.Query("SELECT source_id, gril_id FROM grilist.grils_id_mappings WHERE source = $1;", r.ID())
         for rows.Next() {
@@ -200,6 +251,7 @@ func main() {
                         dbID = InsertGril(g)
                 }
                 InsertTags(dbID, g, taglist)
+                InsertTraits(dbID, g, traitlist)
 
                 return nil
         })