Update database.py

This commit is contained in:
angelblue05 2016-11-04 12:13:07 -05:00
parent 012b83f48b
commit 9783b5aec2

View file

@ -7,6 +7,8 @@ import sqlite3
import xbmc import xbmc
from utils import window, should_stop
################################################################################################# #################################################################################################
log = logging.getLogger("EMBY."+__name__) log = logging.getLogger("EMBY."+__name__)
@ -46,6 +48,33 @@ def music_database():
return path return path
def texture_database():
return xbmc.translatePath("special://database/Textures13.db").decode('utf-8')
def emby_database():
return xbmc.translatePath("special://database/emby.db").decode('utf-8')
def kodi_commit():
# verification for the Kodi video scan
kodi_scan = window('emby_kodiScan') == "true"
count = 0
while kodi_scan:
log.info("kodi scan is running, waiting...")
if count == 10:
log.info("flag still active, but will try to commit")
window('emby_kodiScan', clear=True)
elif should_stop() or xbmc.Monitor().waitForAbort(1):
log.info("commit unsuccessful. sync terminating")
return False
kodi_scan = window('emby_kodiScan') == "true"
count += 1
return True
class DatabaseConn(object): class DatabaseConn(object):
# To be called as context manager - i.e. with DatabaseConn() as conn: #dostuff # To be called as context manager - i.e. with DatabaseConn() as conn: #dostuff
@ -70,16 +99,13 @@ class DatabaseConn(object):
def _SQL(self, media_type): def _SQL(self, media_type):
if media_type == "emby": databases = {
return xbmc.translatePath("special://database/emby.db").decode('utf-8') 'emby': emby_database,
elif media_type == "texture": 'texture': texture_database,
return xbmc.translatePath("special://database/Textures13.db").decode('utf-8') 'music': music_database,
elif media_type == "music": 'video': video_database
return music_database() }
elif media_type == "video": return databases[media_type]() if media_type in databases else self.db_file
return video_database()
else: # custom path
return self.db_file
def __exit__(self, exc_type, exc_val, exc_tb): def __exit__(self, exc_type, exc_val, exc_tb):
# Close the connection # Close the connection
@ -87,7 +113,7 @@ class DatabaseConn(object):
if exc_type is not None: if exc_type is not None:
# Errors were raised in the with statement # Errors were raised in the with statement
log.error("rollback: Type: %s Value: %s", exc_type, exc_val) log.error("Type: %s Value: %s", exc_type, exc_val)
if "database is locked" in exc_val: if "database is locked" in exc_val:
self.conn.rollback() self.conn.rollback()
else: else:
@ -95,7 +121,10 @@ class DatabaseConn(object):
elif self.commit_mode is not None and changes: elif self.commit_mode is not None and changes:
log.info("number of rows updated: %s", changes) log.info("number of rows updated: %s", changes)
self.conn.commit() if self.db_file == "video" and kodi_commit():
self.conn.commit()
else:
self.conn.commit()
log.info("close: %s", self.path) log.info("close: %s", self.path)
self.conn.close() self.conn.close()