remove old kodiSQL function

use DatabaseCon for DB file paths
switch last few DB Cons to use the new DatabaseCon context class

resources/lib/artwork.py

@@ -13,7 +13,9 @@ import xbmcvfs
 import requests
 
 import image_cache_thread
-from utils import window, settings, dialog, language as lang, kodiSQL, JSONRPC
+from utils import window, settings, dialog, language as lang, JSONRPC
+from database import DatabaseConn
+from contextlib import closing
 
 ##################################################################################################
 
@@ -164,47 +166,49 @@ class Artwork(object):
 
     def _cache_all_video_entries(self, pdialog):
 
-        conn = kodiSQL('video')
+        with DatabaseConn('video') as conn:
-        cursor = conn.cursor()
+            with closing(conn.cursor()) as cursor_video:    
-        cursor.execute("SELECT url FROM art WHERE media_type != 'actor'") # dont include actors
-        result = cursor.fetchall()
-        total = len(result)
-        log.info("Image cache sync about to process %s images", total)
-        cursor.close()
             
-        count = 0
+                cursor_video.execute("SELECT url FROM art WHERE media_type != 'actor'") # dont include actors
-        for url in result:
+                result = cursor_video.fetchall()
+                total = len(result)
+                log.info("Image cache sync about to process %s images", total)
+                cursor_video.close()
 
-            if pdialog.iscanceled():
+                count = 0
-                break
+                for url in result:
 
-            percentage = int((float(count) / float(total))*100)
+                    if pdialog.iscanceled():
-            message = "%s of %s (%s)" % (count, total, len(self.image_cache_threads))
+                        break
-            pdialog.update(percentage, "%s %s" % (lang(33045), message))
+
-            self.cache_texture(url[0])
+                    percentage = int((float(count) / float(total))*100)
-            count += 1
+                    message = "%s of %s (%s)" % (count, total, len(self.image_cache_threads))
+                    pdialog.update(percentage, "%s %s" % (lang(33045), message))
+                    self.cache_texture(url[0])
+                    count += 1
 
     def _cache_all_music_entries(self, pdialog):
 
-        conn = kodiSQL('music')
+        with DatabaseConn('music') as conn:
-        cursor = conn.cursor()
+            with closing(conn.cursor()) as cursor_music:        
-        cursor.execute("SELECT url FROM art")
-        result = cursor.fetchall()
-        total = len(result)
-        log.info("Image cache sync about to process %s images", total)
-        cursor.close()
         
-        count = 0
+                cursor_music.execute("SELECT url FROM art")
-        for url in result:
+                result = cursor_music.fetchall()
+                total = len(result)
                 
-            if pdialog.iscanceled():
+                log.info("Image cache sync about to process %s images", total)
-                break
 
-            percentage = int((float(count) / float(total))*100)
+                count = 0
-            message = "%s of %s" % (count, total)
+                for url in result:
-            pdialog.update(percentage, "%s %s" % (lang(33045), message))
+
-            self.cache_texture(url[0])
+                    if pdialog.iscanceled():
-            count += 1
+                        break
+
+                    percentage = int((float(count) / float(total))*100)
+                    message = "%s of %s" % (count, total)
+                    pdialog.update(percentage, "%s %s" % (lang(33045), message))
+                    self.cache_texture(url[0])
+                    count += 1
 
     @classmethod
     def delete_cache(cls):
@@ -226,16 +230,14 @@ class Artwork(object):
                     log.debug("deleted: %s", filename)
 
         # remove all existing data from texture DB
-        conn = kodiSQL('texture')
+        with DatabaseConn('texture') as conn:
-        cursor = conn.cursor()
+            with closing(conn.cursor()) as cursor_texture:  
-        cursor.execute('SELECT tbl_name FROM sqlite_master WHERE type="table"')
+                cursor_texture.execute('SELECT tbl_name FROM sqlite_master WHERE type="table"')
-        rows = cursor.fetchall()
+                rows = cursor_texture.fetchall()
-        for row in rows:
+                for row in rows:
-            table_name = row[0]
+                    table_name = row[0]
-            if table_name != "version":
+                    if table_name != "version":
-                cursor.execute("DELETE FROM " + table_name)
+                        cursor_texture.execute("DELETE FROM " + table_name)
-        conn.commit()
-        cursor.close()
 
     def _add_worker_image_thread(self, url):
 
@@ -429,32 +431,28 @@ class Artwork(object):
     @classmethod
     def delete_cached_artwork(cls, url):
         # Only necessary to remove and apply a new backdrop or poster       
-        conn = kodiSQL('texture')
+        with DatabaseConn('texture') as conn:
-        cursor = conn.cursor()
+            with closing(conn.cursor()) as cursor_texture:  
+                try:
+                    cursor_texture.execute("SELECT cachedurl FROM texture WHERE url = ?", (url,))
+                    cached_url = cursor_texture.fetchone()[0]
 
-        try:
+                except TypeError:
-            cursor.execute("SELECT cachedurl FROM texture WHERE url = ?", (url,))
+                    log.info("Could not find cached url")
-            cached_url = cursor.fetchone()[0]
 
-        except TypeError:
+                except OperationalError:
-            log.info("Could not find cached url")
+                    log.info("Database is locked. Skip deletion process.")
 
-        except OperationalError:
+                else: # Delete thumbnail as well as the entry
-            log.info("Database is locked. Skip deletion process.")
+                    thumbnails = xbmc.translatePath("special://thumbnails/%s" % cached_url).decode('utf-8')
+                    log.info("Deleting cached thumbnail: %s", thumbnails)
+                    xbmcvfs.delete(thumbnails)
 
-        else: # Delete thumbnail as well as the entry
+                    try:
-            thumbnails = xbmc.translatePath("special://thumbnails/%s" % cached_url).decode('utf-8')
+                        cursor_texture.execute("DELETE FROM texture WHERE url = ?", (url,))
-            log.info("Deleting cached thumbnail: %s", thumbnails)
+                    except OperationalError:
-            xbmcvfs.delete(thumbnails)
+                        log.debug("Issue deleting url from cache. Skipping.")
 
-            try:
-                cursor.execute("DELETE FROM texture WHERE url = ?", (url,))
-                conn.commit()
-            except OperationalError:
-                log.debug("Issue deleting url from cache. Skipping.")
-
-        finally:
-            cursor.close()
 
     def get_people_artwork(self, people):
         # append imageurl if existing

resources/lib/context_entry.py

@@ -11,7 +11,7 @@ import api
 import read_embyserver as embyserver
 import embydb_functions as embydb
 import musicutils as musicutils
-from utils import settings, dialog, language as lang, kodiSQL
+from utils import settings, dialog, language as lang
 from dialogs import context
 from database import DatabaseConn
 from contextlib import closing
@@ -165,31 +165,28 @@ class ContextMenu(object):
 
     def _rate_song(self):
 
-        conn = kodiSQL('music')
+        with DatabaseConn('music') as conn:
-        cursor = conn.cursor()
+            with closing(conn.cursor()) as cursor_music:            
-        query = "SELECT rating FROM song WHERE idSong = ?"
+                query = "SELECT rating FROM song WHERE idSong = ?"
-        cursor.execute(query, (self.kodi_id,))
+                cursor_music.execute(query, (self.kodi_id,))
-        try:
+                try:
-            value = cursor.fetchone()[0]
+                    value = cursor_music.fetchone()[0]
-            current_value = int(round(float(value), 0))
+                    current_value = int(round(float(value), 0))
-        except TypeError:
+                except TypeError:
-            pass
+                    pass
-        else:
+                else:
-            new_value = dialog("numeric", 0, lang(30411), str(current_value))
+                    new_value = dialog("numeric", 0, lang(30411), str(current_value))
-            if new_value > -1:
+                    if new_value > -1:
 
-                new_value = int(new_value)
+                        new_value = int(new_value)
-                if new_value > 5:
+                        if new_value > 5:
-                    new_value = 5
+                            new_value = 5
 
-                if settings('enableUpdateSongRating') == "true":
+                        if settings('enableUpdateSongRating') == "true":
-                    musicutils.updateRatingToFile(new_value, self.api.get_file_path())
+                            musicutils.updateRatingToFile(new_value, self.api.get_file_path())
 
-                query = "UPDATE song SET rating = ? WHERE idSong = ?"
+                        query = "UPDATE song SET rating = ? WHERE idSong = ?"
-                cursor.execute(query, (new_value, self.kodi_id,))
+                        cursor_music.execute(query, (new_value, self.kodi_id,))
-                conn.commit()
-        finally:
-            cursor.close()
 
     def _delete_item(self):
 

resources/lib/entrypoint.py

@@ -180,12 +180,10 @@ def emby_backup():
     shutil.copy(src=xbmc.translatePath("special://database/emby.db").decode('utf-8'),
                 dst=database)
     # Videos database
-    shutil.copy(src=utils.getKodiVideoDBPath(),
+    shutil.copy(src=DatabaseConn()._SQL('video'), dst=database)
-                dst=database)
     # Music database
     if settings('enableMusic') == "true":
-        shutil.copy(src=utils.getKodiMusicDBPath(),
+        shutil.copy(src=DatabaseConn()._SQL('music'), dst=database)
-                    dst=database)
 
     dialog(type_="ok",
            heading="{emby}",

resources/lib/itemtypes.py

@@ -6,7 +6,9 @@ import logging
 
 import read_embyserver as embyserver
 from objects import Movies, MusicVideos, TVShows, Music
-from utils import settings, kodiSQL
+from utils import settings
+from database import DatabaseConn
+from contextlib import closing
 
 #################################################################################################
 
@@ -58,46 +60,40 @@ class Items(object):
         if pdialog:
             pdialog.update(heading="Processing %s: %s items" % (process, total))
 
-        for itemtype in items:
+        # this is going to open a music connection even if it is not needed but
+        # I feel that is better than trying to sort out the login yourself
+        with DatabaseConn('music') as conn:
+            with closing(conn.cursor()) as cursor_music:
             
-            # Safety check
+                for itemtype in items:
-            if not itemtypes.get(itemtype):
-                # We don't process this type of item
-                continue
 
-            itemlist = items[itemtype]
+                    # Safety check
-            if not itemlist:
+                    if not itemtypes.get(itemtype):
-                # The list to process is empty
+                        # We don't process this type of item
-                continue
+                        continue
 
-            musicconn = None
+                    itemlist = items[itemtype]
+                    if not itemlist:
+                        # The list to process is empty
+                        continue
 
-            if itemtype in ('MusicAlbum', 'MusicArtist', 'AlbumArtist', 'Audio'):
+                    if itemtype in ('MusicAlbum', 'MusicArtist', 'AlbumArtist', 'Audio'):
-                if self.music_enabled:
+                        if self.music_enabled:
-                    musicconn = kodiSQL('music')
+                            items_process = itemtypes[itemtype](embycursor, cursor_music, pdialog) # see note above
-                    musiccursor = musicconn.cursor()
+                        else:
-                    items_process = itemtypes[itemtype](embycursor, musiccursor, pdialog)
+                            # Music is not enabled, do not proceed with itemtype
-                else:
+                            continue
-                    # Music is not enabled, do not proceed with itemtype
+                    else:
-                    continue
+                        update_videolibrary = True
-            else:
+                        items_process = itemtypes[itemtype](embycursor, kodicursor, pdialog)
-                update_videolibrary = True
-                items_process = itemtypes[itemtype](embycursor, kodicursor, pdialog)
 
+                    if process == "added":
+                        items_process.add_all(itemtype, itemlist)
+                    elif process == "remove":
+                        items_process.remove_all(itemtype, itemlist)
+                    else:
+                        process_items = self.emby.getFullItems(itemlist)
+                        items_process.process_all(itemtype, process, process_items, total)
 
-            if process == "added":
-                items_process.add_all(itemtype, itemlist)
-            elif process == "remove":
-                items_process.remove_all(itemtype, itemlist)
-            else:
-                process_items = self.emby.getFullItems(itemlist)
-                items_process.process_all(itemtype, process, process_items, total)
-
-
-            if musicconn is not None:
-                # close connection for special types
-                log.info("updating music database")
-                musicconn.commit()
-                musiccursor.close()
 
         return (True, update_videolibrary)

resources/lib/kodimonitor.py

@@ -11,7 +11,7 @@ import xbmcgui
 import downloadutils
 import embydb_functions as embydb
 import playbackutils as pbutils
-from utils import window, settings, kodiSQL
+from utils import window, settings
 from ga_client import log_error
 from database import DatabaseConn
 from contextlib import closing

resources/lib/librarysync.py

@@ -291,7 +291,7 @@ class LibrarySync(threading.Thread):
                         continue
 
                     startTime = datetime.now()
-                    completed = process[itemtype](cursor_emby, kodicursor, pDialog)
+                    completed = process[itemtype](cursor_emby, cursor_video, pDialog)
                     if not completed:
                         xbmc.executebuiltin('InhibitIdleShutdown(false)')
                         utils.setScreensaver(value=screensaver)
@@ -913,7 +913,7 @@ class LibrarySync(threading.Thread):
 
             if not startupComplete:
                 # Verify the video database can be found
-                videoDb = utils.getKodiVideoDBPath()
+                videoDb = DatabaseConn()._SQL('video')
                 if not xbmcvfs.exists(videoDb):
                     # Database does not exists
                     log.error(

resources/lib/playlist.py

@@ -11,7 +11,7 @@ import playutils
 import playbackutils
 import embydb_functions as embydb
 import read_embyserver as embyserver
-from utils import window, kodiSQL, JSONRPC
+from utils import window, JSONRPC
 from database import DatabaseConn
 from contextlib import closing
 

resources/lib/read_embyserver.py

@@ -8,7 +8,7 @@ import hashlib
 import xbmc
 
 import downloadutils
-from utils import window, settings, kodiSQL
+from utils import window, settings
 from database import DatabaseConn
 from contextlib import closing
 

resources/lib/utils.py

@@ -120,55 +120,6 @@ def should_stop():
     else: # Keep going
         return False
 
-def kodiSQL(media_type="video"):
-
-    if media_type == "emby":
-        dbPath = xbmc.translatePath("special://database/emby.db").decode('utf-8')
-    elif media_type == "texture":
-        dbPath = xbmc.translatePath("special://database/Textures13.db").decode('utf-8')
-    elif media_type == "music":
-        dbPath = getKodiMusicDBPath()
-    else:
-        dbPath = getKodiVideoDBPath()
-
-    if settings('dblock') == "true":
-        connection = sqlite3.connect(dbPath, isolation_level=None, timeout=20)
-    else:
-        connection = sqlite3.connect(dbPath, timeout=20)
-    return connection
-
-def getKodiVideoDBPath():
-
-    dbVersion = {
-
-        "13": 78,   # Gotham
-        "14": 90,   # Helix
-        "15": 93,   # Isengard
-        "16": 99,   # Jarvis
-        "17": 107   # Krypton
-    }
-
-    dbPath = xbmc.translatePath(
-                "special://database/MyVideos%s.db"
-                % dbVersion.get(xbmc.getInfoLabel('System.BuildVersion')[:2], "")).decode('utf-8')
-    return dbPath
-
-def getKodiMusicDBPath():
-
-    dbVersion = {
-
-        "13": 46,   # Gotham
-        "14": 48,   # Helix
-        "15": 52,   # Isengard
-        "16": 56,   # Jarvis
-        "17": 60    # Krypton
-    }
-
-    dbPath = xbmc.translatePath(
-                "special://database/MyMusic%s.db"
-                % dbVersion.get(xbmc.getInfoLabel('System.BuildVersion')[:2], "")).decode('utf-8')
-    return dbPath
-
 #################################################################################################
 # Utility methods