def enable_addon(self,addon):
#import json
#sCheck = {'jsonrpc': '2.0','id': 1,'method': 'Addons.GetAddonDetails','params': {'addonid':'inputstream.adaptive','properties': ['enabled']}}
#response = xbmc.executeJSONRPC(json.dumps(sCheck))
#data = json.loads(response)
#if not 'error' in data.keys():
#if data['result']['addon']['enabled'] == False:
if xbmc.getCondVisibility('System.HasAddon(inputstream.adaptive)') == 0:
do_json = '{"jsonrpc":"2.0","id":1,"method":"Addons.SetAddonEnabled","params":{"addonid":"inputstream.adaptive","enabled":true}}'
query = xbmc.executeJSONRPC(do_json)
VSlog("Activation d'inputstream.adaptive")
else:
VSlog('inputstream.adaptive déjà activé')
def enable_addon(self,addon):
#import json
#sCheck = {'jsonrpc': '2.0','id': 1,'method': 'Addons.GetAddonDetails','params': {'addonid':'inputstream.adaptive','properties': ['enabled']}}
#response = xbmc.executeJSONRPC(json.dumps(sCheck))
#data = json.loads(response)
#if not 'error' in data.keys():
#if data['result']['addon']['enabled'] == False:
if xbmc.getCondVisibility('System.HasAddon(inputstream.adaptive)') == 0:
do_json = '{"jsonrpc":"2.0","id":1,"method":"Addons.SetAddonEnabled","params":{"addonid":"inputstream.adaptive","enabled":true}}'
query = xbmc.executeJSONRPC(do_json)
VSlog("Activation d'inputstream.adaptive")
else:
VSlog('inputstream.adaptive déjà activé')
class cDb:
#On chercher le profil courant.
request = {
"jsonrpc": "2.0",
"method": "Profiles.GetCurrentProfile",
"params": {
"properties": ["thumbnail", "lockmode"]
},
"id": 1
}
req = json.dumps(request)
response = xbmc.executeJSONRPC(req)
#On recupere le nom.
name = json.loads(response)['result']['label']
#Le cas par defaut.
if name == 'Master user':
DB = 'special://home/userdata/addon_data/plugin.video.vstream/vstream.db'
else:
DB = 'special://home/userdata/profiles/' + name + '/addon_data/plugin.video.vstream/vstream.db'
try:
REALDB = VSPath(DB).decode('utf-8')
except AttributeError:
REALDB = VSPath(DB)
del request, req, name, response # delete des objets temporaires
def __init__(self):
VSlog('DB engine for db : ' + sqlite.__name__)
try:
if not xbmcvfs.exists(self.DB):
self.db = sqlite.connect(self.REALDB)
self.db.row_factory = sqlite.Row
self.dbcur = self.db.cursor()
self._create_tables()
return
except:
VSlog('Error: Unable to write to %s' % self.REALDB)
pass
try:
self.db = sqlite.connect(self.REALDB)
self.db.row_factory = sqlite.Row
self.dbcur = self.db.cursor()
except:
VSlog('Error: Unable to access to %s' % self.REALDB)
pass
def __del__(self):
''' Cleanup db when object destroyed '''
try:
self.dbcur.close()
self.db.close()
except Exception as e:
pass
def _create_tables(self, dropTable = ''):
if dropTable != '':
self.dbcur.execute("DROP TABLE IF EXISTS " + dropTable)
self.db.commit()
''' Create table '''
sql_create = "CREATE TABLE IF NOT EXISTS history ("\
"addon_id integer PRIMARY KEY AUTOINCREMENT, "\
"title TEXT, "\
"disp TEXT, "\
"icone TEXT, "\
"isfolder TEXT, "\
"level TEXT, "\
"lastwatched TIMESTAMP "", "\
"UNIQUE(title)"\
");"
self.dbcur.execute(sql_create)
sql_create = "CREATE TABLE IF NOT EXISTS resume ("\
"addon_id integer PRIMARY KEY AUTOINCREMENT, "\
"title TEXT, "\
"hoster TEXT, "\
"point TEXT, "\
"total TEXT, "\
"UNIQUE(title, hoster)"\
");"
self.dbcur.execute(sql_create)
sql_create = "CREATE TABLE IF NOT EXISTS watched ("\
"addon_id integer PRIMARY KEY AUTOINCREMENT, "\
"title TEXT, "\
"cat TEXT, "\
"UNIQUE(title)"\
");"
self.dbcur.execute(sql_create)
sql_create = "CREATE TABLE IF NOT EXISTS favorite ("\
"addon_id integer PRIMARY KEY AUTOINCREMENT, "\
"title TEXT, "\
"siteurl TEXT, "\
"site TEXT, "\
"fav TEXT, "\
"cat TEXT, "\
"icon TEXT, "\
"fanart TEXT, "\
"UNIQUE(title, site)"\
");"
self.dbcur.execute(sql_create)
sql_create = "CREATE TABLE IF NOT EXISTS viewing ("\
"addon_id integer PRIMARY KEY AUTOINCREMENT, "\
"tmdb_id TEXT, "\
"title_id TEXT, "\
"title TEXT, "\
"siteurl TEXT, "\
"site TEXT, "\
"fav TEXT, "\
"cat TEXT, "\
"season integer, "\
"UNIQUE (title_id)"\
");"
self.dbcur.execute(sql_create)
sql_create = "CREATE TABLE IF NOT EXISTS download ("\
"addon_id integer PRIMARY KEY AUTOINCREMENT, "\
"title TEXT, "\
"url TEXT, "\
"path TEXT, "\
"cat TEXT, "\
"icon TEXT, "\
"size TEXT,"\
"totalsize TEXT, "\
"status TEXT, "\
"UNIQUE(title, path)"\
");"
self.dbcur.execute(sql_create)
VSlog('Table initialized')
# Ne pas utiliser cette fonction pour les chemins
def str_conv(self, data):
if not isMatrix():
if isinstance(data, str):
# Must be encoded in UTF-8
try:
data = data.decode('utf8')
except AttributeError:
pass
import unicodedata
data = unicodedata.normalize('NFKD', data).encode('ascii', 'ignore')
try:
data = data.decode('string-escape') # ATTENTION: provoque des bugs pour les chemins a cause du caractere '/'
except:
pass
else:
data = data.encode().decode()
return data
# ***********************************
# History fonctions
# ***********************************
def insert_history(self, meta):
# title = Unquote(meta['title']).decode('ascii', 'ignore')
title = self.str_conv(Unquote(meta['title']))
disp = meta['disp']
icon = 'icon.png'
try:
ex = 'INSERT INTO history (title, disp, icone) VALUES (?, ?, ?)'
self.dbcur.execute(ex, (title, disp, icon))
self.db.commit()
VSlog('SQL INSERT history Successfully')
except Exception as e:
if 'UNIQUE constraint failed' in e.message:
ex = "UPDATE history set title = '%s', disp = '%s', icone= '%s' WHERE title = '%s'" % (title, disp, icon, title)
self.dbcur.execute(ex)
self.db.commit()
VSlog('SQL UPDATE history Successfully')
VSlog('SQL ERROR INSERT, title = %s, %s' % (title, e) )
pass
def get_history(self):
sql_select = 'SELECT * FROM history'
try:
self.dbcur.execute(sql_select)
# matchedrow = self.dbcur.fetchone()
matchedrow = self.dbcur.fetchall()
return matchedrow
except Exception as e:
VSlog('SQL ERROR EXECUTE, %s' % e)
return None
def del_history(self):
from resources.lib.gui.gui import cGui
oGui = cGui()
oInputParameterHandler = cInputParameterHandler()
if oInputParameterHandler.exist('searchtext'):
sql_delete = "DELETE FROM history WHERE title = '%s'" % (oInputParameterHandler.getValue('searchtext'))
else:
sql_delete = 'DELETE FROM history;'
try:
self.dbcur.execute(sql_delete)
self.db.commit()
dialog().VSinfo(addon().VSlang(30041))
oGui.updateDirectory()
return False, False
except Exception as e:
VSlog('SQL ERROR DELETE : %s' % sql_delete)
return False, False
# ***********************************
# Watched fonctions
# ***********************************
def insert_watched(self, meta):
title = meta['title']
if not title:
return
cat = meta['cat'] if 'cat' in meta else '1'
ex = 'INSERT or IGNORE INTO watched (title, cat) VALUES (?, ?)'
try:
self.dbcur.execute(ex, (title, cat))
self.db.commit()
VSlog('SQL INSERT watched Successfully')
except Exception as e:
if 'no such column' in str(e) or 'no column named' in str(e) or 'no such table' in str(e) :
if 'named cat' in str(e): # ajout nouvelle colonne 'cat'
self.dbcur.execute("ALTER TABLE watched add column cat TEXT")
self.db.commit()
VSlog('Table recreated : watched')
# Deuxieme tentative
self.dbcur.execute(ex, (title, cat))
self.db.commit()
else:
VSlog('SQL ERROR INSERT watched : title = %s' % e )
def get_watched(self, meta):
title = meta['title']
if not title:
return False
cat = meta['cat'] if 'cat' in meta else '1'
sql_select = "SELECT * FROM watched WHERE title = '%s'" % title
try:
self.dbcur.execute(sql_select)
matchedrow = self.dbcur.fetchall()
# Gestion des homonymes films / séries
# Si la cat est enregistrée, on vérifie si c'est la même
for data in matchedrow:
matchedcat = data['cat']
if matchedcat:
return int(matchedcat) == int(cat)
if matchedrow:
return True
return False
except Exception as e:
if 'no such column' in str(e) or 'no column named' in str(e) or 'no such table' in str(e) :
# Deuxieme tentative, sans la cat
sql_select = "SELECT * FROM watched WHERE title = '%s'" % title
self.dbcur.execute(sql_select)
return 1 if self.dbcur.fetchall() else 0
else:
VSlog('SQL ERROR %s' % sql_select)
return False
def del_watched(self, meta):
title = meta['title']
if not title:
return
sql_select = "DELETE FROM watched WHERE title = '%s'" % title
try:
self.dbcur.execute(sql_select)
self.db.commit()
return False, False
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return False, False
# ***********************************
# Resume fonctions
# ***********************************
def insert_resume(self, meta):
title = self.str_conv(meta['title'])
site = QuotePlus(meta['site'])
point = meta['point']
total = meta['total']
ex = "DELETE FROM resume WHERE title = '%s'" % title
try:
self.dbcur.execute(ex)
except Exception as e:
VSlog('SQL ERROR - ' + ex)
pass
try:
ex = 'INSERT INTO resume (title, hoster, point, total) VALUES (?, ?, ?, ?)'
self.dbcur.execute(ex, (title, site, point, total))
self.db.commit()
except Exception as e:
if 'no such column' in str(e) or 'no column named' in str(e) or 'no such table' in str(e) :
self._create_tables('resume')
VSlog('Table recreated : resume')
# Deuxieme tentative
self.dbcur.execute(ex, (title, site, point, total))
self.db.commit()
else:
VSlog('SQL ERROR INSERT : %s' % e)
def get_resume(self, meta):
title = self.str_conv(meta['title'])
# site = QuotePlus(meta['site'])
sql_select = "SELECT point, total FROM resume WHERE title = '%s'" % title
try:
self.dbcur.execute(sql_select)
matchedrow = self.dbcur.fetchone()
# matchedrow = self.dbcur.fetchall()
if not matchedrow:
return False, False
return float(matchedrow[0]), float(matchedrow[1])
except Exception as e:
if 'no such column' in str(e) or 'no column named' in str(e) :
self._create_tables('resume')
VSlog('Table recreated : resume')
else:
VSlog('SQL ERROR : %s' % e)
return False, False
def del_resume(self, meta):
title = QuotePlus(meta['title'])
sql_select = "DELETE FROM resume WHERE title = '%s'" % title
try:
self.dbcur.execute(sql_select)
self.db.commit()
return False, False
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return False, False
# ***********************************
# Bookmark fonctions
# ***********************************
def insert_bookmark(self, meta):
title = self.str_conv(meta['title'])
siteurl = QuotePlus(meta['siteurl'])
try:
sIcon = meta['icon'].decode('UTF-8')
except:
sIcon = meta['icon']
try:
ex = 'INSERT INTO favorite (title, siteurl, site, fav, cat, icon, fanart) VALUES (?, ?, ?, ?, ?, ?, ?)'
self.dbcur.execute(ex, (title, siteurl, meta['site'], meta['fav'], meta['cat'], sIcon, meta['fanart']))
self.db.commit()
dialog().VSinfo(addon().VSlang(30042), meta['title'])
VSlog('SQL INSERT favorite Successfully')
except Exception as e:
if 'UNIQUE constraint failed' in e.message:
dialog().VSinfo(addon().VSlang(30043), meta['title'])
VSlog('SQL ERROR INSERT : %s' % e)
pass
def get_bookmark(self):
sql_select = 'SELECT * FROM favorite order by addon_id desc'
try:
self.dbcur.execute(sql_select)
# matchedrow = self.dbcur.fetchone()
matchedrow = self.dbcur.fetchall()
return matchedrow
except Exception as e:
VSlog('SQL ERROR EXECUTE')
return None
def del_bookmark(self, sSiteUrl='', sMovieTitle='', sCat = '', sAll = False):
sql_delete = None
# Tous supprimer
if sAll:
sql_delete = 'DELETE FROM favorite;'
# Supprimer un bookmark selon son titre
elif sMovieTitle:
siteUrl = QuotePlus(sSiteUrl)
title = self.str_conv(sMovieTitle)
title = title.replace("'", r"''")
sql_delete = "DELETE FROM favorite WHERE siteurl = '%s' AND title = '%s'" % (siteUrl, title)
# Supprimer un bookmark selon son url
elif sSiteUrl:
siteUrl = QuotePlus(sSiteUrl)
sql_delete = "DELETE FROM favorite WHERE siteurl = '%s'" % siteUrl
# Supprimer toute une catégorie
elif sCat:
catList = ('1', '7') # films, saga
if sCat not in catList:
catList = ('2', '3', '4', '8')
if sCat not in catList:
catList = ('0', sCat)
sql_delete = "DELETE FROM favorite WHERE cat in %s" % str(catList)
if sql_delete:
from resources.lib.gui.gui import cGui
try:
self.dbcur.execute(sql_delete)
self.db.commit()
update = self.db.total_changes
if not update and sSiteUrl and sMovieTitle:
# si pas trouvé, on essaie sans le titre, seulement l'URL
return self.del_bookmark(sSiteUrl)
dialog().VSinfo(addon().VSlang(30044))
cGui().updateDirectory()
return True
except Exception as e:
VSlog('SQL ERROR %s' % sql_delete)
return False
# ***********************************
# InProgress fonctions
# ***********************************
def insert_viewing(self, meta):
if not 'title' in meta:
return
if not 'siteurl' in meta:
return
title = self.str_conv(meta['title'])
titleWatched = self.str_conv(meta['titleWatched'])
siteurl = QuotePlus(meta['siteurl'])
cat = meta['cat']
saison = meta['season'] if 'season' in meta else ''
sTmdbId = meta['sTmdbId'] if 'sTmdbId' in meta else ''
ex = "DELETE FROM viewing WHERE title_id = '%s' and cat = '%s'" % (titleWatched, cat)
try:
self.dbcur.execute(ex)
except Exception as e:
VSlog('SQL ERROR - ' + ex)
pass
try:
ex = 'INSERT INTO viewing (tmdb_id, title_id, title, siteurl, site, fav, cat, season) VALUES (?, ?, ?, ?, ?, ?, ?, ?)'
self.dbcur.execute(ex, (sTmdbId, titleWatched, title, siteurl, meta['site'], meta['fav'], cat, saison))
self.db.commit()
VSlog('SQL INSERT viewing Successfully')
except Exception as e:
if 'no such column' in str(e) or 'no column named' in str(e) or 'no such table' in str(e) :
self._create_tables('viewing')
VSlog('Table recreated : viewing')
# Deuxieme tentative
self.dbcur.execute(ex, (meta['sTmdbId'], titleWatched, title, siteurl, meta['site'], meta['fav'], meta['cat'], saison, episode))
self.db.commit()
else:
VSlog('SQL ERROR INSERT : %s' % e)
pass
def get_viewing(self):
sql_select = "SELECT * FROM viewing group by title order by addon_id DESC"
try:
self.dbcur.execute(sql_select)
matchedrow = self.dbcur.fetchall()
return matchedrow
except Exception as e:
VSlog('SQL ERROR : %s' % sql_select)
return None
def del_viewing(self, meta):
sTitleWatched = meta['titleWatched'] if 'titleWatched' in meta else None
sql_deleteCat = ""
if not sTitleWatched: # delete all
sql_delete = "DELETE FROM viewing"
else:
sql_deleteTitle = "DELETE FROM viewing WHERE title_id = '%s'" % sTitleWatched
if 'cat' in meta:
sql_deleteCat = " and cat = '%s'" %meta['cat']
sql_delete = sql_deleteTitle + sql_deleteCat
update = 0
from resources.lib.gui.gui import cGui
try:
self.dbcur.execute(sql_delete)
self.db.commit()
update = self.db.total_changes
# si pas trouvé, on essaie sans la cat, juste le titre
if not update and sql_deleteCat:
del meta['cat']
return self.del_viewing(meta)
return True
except Exception as e:
VSlog('SQL ERROR %s, error = %s' % (sql_delete, e))
return update
# ***********************************
# Download fonctions
# ***********************************
def insert_download(self, meta):
title = self.str_conv(meta['title'])
url = QuotePlus(meta['url'])
sIcon = QuotePlus(meta['icon'])
sPath = meta['path']
ex = 'INSERT INTO download (title, url, path, cat, icon, size, totalsize, status) VALUES (?, ?, ?, ?, ?, ?, ?, ?)'
try:
self.dbcur.execute(ex, (title,url, sPath,meta['cat'],sIcon, '', '', 0))
self.db.commit()
VSlog('SQL INSERT download Successfully')
dialog().VSinfo(addon().VSlang(30042), meta['title'])
except Exception as e:
VSlog('SQL ERROR INSERT into download')
pass
def get_download(self, meta=''):
if meta == '':
sql_select = 'SELECT * FROM download'
else:
url = QuotePlus(meta['url'])
sql_select = "SELECT * FROM download WHERE url = '%s' AND status = '0'" % url
try:
self.dbcur.execute(sql_select)
matchedrow = self.dbcur.fetchall()
return matchedrow
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return None
def clean_download(self):
sql_select = "DELETE FROM download WHERE status = '2'"
try:
self.dbcur.execute(sql_select)
self.db.commit()
return False, False
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return False, False
def reset_download(self, meta):
url = QuotePlus(meta['url'])
sql_select = "UPDATE download SET status = '0' WHERE status = '2' AND url = '%s'" % url
try:
self.dbcur.execute(sql_select)
self.db.commit()
return False, False
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return False, False
def del_download(self, meta):
if len(meta['url']) > 1:
url = QuotePlus(meta['url'])
sql_select = "DELETE FROM download WHERE url = '%s'" % url
elif len(meta['path']) > 1:
path = meta['path']
sql_select = "DELETE FROM download WHERE path = '%s'" % path
else:
return
try:
self.dbcur.execute(sql_select)
self.db.commit()
return False, False
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return False, False
def cancel_download(self):
sql_select = "UPDATE download SET status = '0' WHERE status = '1'"
try:
self.dbcur.execute(sql_select)
self.db.commit()
return False, False
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return False, False
def update_download(self, meta):
path = meta['path']
size = meta['size']
totalsize = meta['totalsize']
status = meta['status']
sql_select = "UPDATE download set size = '%s', totalsize = '%s', status= '%s' WHERE path = '%s'" % (size, totalsize, status, path)
try:
self.dbcur.execute(sql_select)
self.db.commit()
return False, False
except Exception as e:
VSlog('SQL ERROR %s' % sql_select)
return False, False
