From 36b2392e835b0312bfd04f65f657fc31d41e0911 Mon Sep 17 00:00:00 2001
From: Marcel van der Veldt <m.vanderveldt@outlook.com>
Date: Tue, 5 May 2015 16:20:09 +0200
Subject: [PATCH] corrected the genres method

---
 resources/lib/WriteKodiDB.py | 63 +++++++++++++++++++++++++++++++++---
 1 file changed, 59 insertions(+), 4 deletions(-)

diff --git a/resources/lib/WriteKodiDB.py b/resources/lib/WriteKodiDB.py
index 2ae3787c..1e261305 100644
--- a/resources/lib/WriteKodiDB.py
+++ b/resources/lib/WriteKodiDB.py
@@ -192,12 +192,12 @@ class WriteKodiDB():
         
         #update genres
         self.AddGenresToMedia(movieid, genres, "movie", cursor)
-               
-        #update studios
-        self.AddStudiosToMedia(movieid, studios, "movie", cursor)
         
         #update countries
-        self.AddCountriesToMedia(movieid, MBitem.get("ProductionLocations"), "movie", cursor)
+        self.AddGenresToMedia(movieid, MBitem.get("ProductionLocations"), "movie", cursor)
+        
+        #update studios
+        self.AddStudiosToMedia(movieid, studios, "movie", cursor)
         
         #add streamdetails
         self.AddStreamDetailsToMedia(API().getMediaStreams(MBitem), fileid, cursor)
@@ -789,6 +789,60 @@ class WriteKodiDB():
                             peoplesql="INSERT OR REPLACE into writerlinkepisode(idWriter, idEpisode) values(?, ?)"
                             cursor.execute(peoplesql, (actorid,id))
                         
+    def AddGenresToMedia(self, id, genres, mediatype, cursor):
+
+        if genres:
+            
+            kodiVersion = 14
+            if xbmc.getInfoLabel("System.BuildVersion").startswith("15"):
+                kodiVersion = 15
+            
+            for genre in genres:
+
+                if kodiVersion == 15:
+                    genre_id = None
+                    cursor.execute("SELECT genre_id as genre_id FROM genre WHERE name = ?",(genre,))
+                    result = cursor.fetchone()
+                    if result != None:
+                        genre_id = result[0]
+                    #create genre
+                    if genre_id == None:
+                        cursor.execute("select coalesce(max(genre_id),0) as genre_id from genre")
+                        genre_id = cursor.fetchone()[0]
+                        genre_id = genre_id + 1
+                        sql="insert into genre(genre_id, name) values(?, ?)"
+                        cursor.execute(sql, (genre_id,genre))
+                        utils.logMsg("AddGenresToMedia", "Processing : " + genre)
+                    
+                    #assign genre to item    
+                    sql="INSERT OR REPLACE into genre_link(genre_id, media_id, media_type) values(?, ?, ?)"
+                    cursor.execute(sql, (genre_id, id, mediatype))
+                
+                else:
+                    idGenre = None
+                    cursor.execute("SELECT idGenre as idGenre FROM genre WHERE strGenre = ?",(genre,))
+                    result = cursor.fetchone()
+                    if result != None:
+                        idGenre = result[0]
+                    #create genre
+                    if idGenre == None:
+                        cursor.execute("select coalesce(max(idGenre),0) as idGenre from genre")
+                        idGenre = cursor.fetchone()[0]
+                        idGenre = idGenre + 1
+                        sql="insert into genre(idGenre, strGenre) values(?, ?)"
+                        cursor.execute(sql, (idGenre,genre))
+
+                    #assign genre to item    
+                    if mediatype == "movie":
+                        sql="INSERT OR REPLACE into genrelinkmovie(idGenre, idMovie) values(?, ?)"
+                    if mediatype == "tvshow":
+                        sql="INSERT OR REPLACE into genrelinktvshow(idGenre, idShow) values(?, ?)"
+                    if mediatype == "episode":
+                        return
+                    if mediatype == "musicvideo":
+                        sql="INSERT OR REPLACE into genrelinkmusicvideo(idGenre, idMVideo) values(?, ?)"
+                    cursor.execute(sql, (idGenre,id))
+    
     def AddCountriesToMedia(self, id, countries, mediatype, cursor):
         if countries:
             kodiVersion = 14
@@ -838,6 +892,7 @@ class WriteKodiDB():
                         #only movies have a country field
                         return
                         
+    
     def AddStudiosToMedia(self, id, studios, mediatype, cursor):
 
         if studios: