stevazi/addon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

alcune migliorie a dbconverter

Browse Source
This commit is contained in:
Alhaziel01
2021-05-15 16:19:50 +02:00
parent 273fff7c48
commit 551e89e0f5
1 changed files with 61 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files
platformcode/dbconverter.py
+61 -72
View File
@@ -17,49 +17,33 @@ class addVideo(object):
self.info = self.item.infoLabels
self.videoExist, self.VideoId = self.get_id()
self.message = ''
self.sql_actions = []
if not self.videoExist:
logger.debug('Add {}: {} to Kodi Library'.format(self.item.contentType, self.item.title))
self.strPath, self.parentPath, self.strFilename, self.path = get_path(self.item)
self.date = strftime('%Y-%m-%d %H:%M:%S', localtime(float(time())))
ini = start = time()
self.set_path()
self.message += 'path:{}'.format(time() - start)
start = time()
self.set_sets()
self.message += ', sets:{}'.format(time() - start)
start = time()
self.set_files()
self.message += ', files:{}'.format(time() - start)
start = time()
self.set_rating()
self.message += ', rating:{}'.format(time() - start)
start = time()
self.set_ids()
self.message += ', ids:{}'.format(time() - start)
start = time()
self.set_actors()
self.message += ', actors:{}'.format(time() - start)
start = time()
self.set_country()
self.message += ', country:{}'.format(time() - start)
start = time()
self.set_genre()
self.message += ', genre:{}'.format(time() - start)
start = time()
self.set_studio()
self.message += ', studio:{}'.format(time() - start)
start = time()
if self.item.contentType == 'movie': self.set_movie()
elif self.item.contentType == 'tvshow': self.set_tvshow()
elif self.item.contentType == 'season': self.set_season()
else: self.set_episode()
self.message += ', video:{}'.format(time() - start)
start = time()
self.set_art()
self.message += ', art:{}'.format(time() - start)
# platformtools.dialog_ok('',self.message)
logger.debug('TEMPO TOTALE:',time() - ini, self.message)
for sql, params in self.sql_actions:
nun_records, records = execute_sql_kodi(sql, params, conn)
payload = {
"jsonrpc": "2.0",
"method": "VideoLibrary.Scan",
@@ -67,14 +51,13 @@ class addVideo(object):
"id": 1
}
get_data(payload)
conn.close()
def get_id(self):
Type = 'id' + self.item.contentType.replace('tv','').capitalize()
sql = 'select {} from {}_view where (uniqueid_value like "{}" and uniqueid_type like "kod")'.format(Type, self.item.contentType, self.item.videolibrary_id)
n, records = execute_sql_kodi(sql, conn=conn)
if n: return True, records[0][0]
sql = 'select {} from {}_view where (uniqueid_value = "{}" and uniqueid_type = "kod")'.format(Type, self.item.contentType, self.item.videolibrary_id)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if nun_records: return True, records[0][0]
sql = 'SELECT MAX({}) FROM {}_view'.format(Type, self.item.contentType)
nun_records, records = execute_sql_kodi(sql, conn=conn)
@@ -83,14 +66,14 @@ class addVideo(object):
return False, _id
def set_path(self):
sql = 'select idPath from path where (strPath like "{}")'.format(self.parentPath)
sql = 'select idPath from path where (strPath = "{}")'.format(self.parentPath)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if records:
self.idParentPath = records[0][0]
else:
return
sql = 'select idPath from path where (strPath like "{}")'.format(self.strPath)
sql = 'select idPath from path where (strPath = "{}")'.format(self.strPath)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if records:
self.idPath = records[0][0]
@@ -99,21 +82,21 @@ class addVideo(object):
sql = 'INSERT OR IGNORE INTO path (idPath, strPath, dateAdded, idParentPath, noUpdate) VALUES ( ?, ?, ?, ?, ?)'
params = (self.idPath, self.strPath, self.date, self.idParentPath, 1)
n, records = execute_sql_kodi(sql, params, conn)
self.sql_actions.append([sql, params])
def set_sets(self):
self.idSet = None
if self.info.get('set'):
sql = 'SELECT idSet from sets where (strSet like "{}")'.format(self.info.get('set'))
sql = 'SELECT idSet from sets where (strSet = "{}")'.format(self.info.get('set'))
# params = self.info.get('set')
n, records = execute_sql_kodi(sql, conn=conn)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if records:
self.idSet = records[0][0]
else:
self.idSet = get_id('idSet', 'sets')
sql = 'INSERT OR IGNORE INTO sets (idSet, strSet, strOvervieW) VALUES ( ?, ?, ?)'
params = (self.idSet, self.info.get('set'), self.info.get('setoverview'))
n, records = execute_sql_kodi(sql, params, conn)
self.sql_actions.append([sql, params])
if self.info.get('setposters'):
self.art.append({'media_id':self.idSet, 'media_type': 'set', 'type':'poster', 'url':self.info.get('setposters')[0]})
if self.info.get('setfanarts'):
@@ -127,7 +110,7 @@ class addVideo(object):
else:
sql = 'INSERT OR IGNORE INTO files (idFile, idPath, strFilename, dateAdded) VALUES ( ?, ?, ?, ?)'
params = (self.idFile, self.idPath, self.strFilename, self.date)
n, records = execute_sql_kodi(sql, params, conn)
self.sql_actions.append([sql, params])
def set_rating(self):
self.rating_id = get_id('rating_id', 'rating')
@@ -136,7 +119,7 @@ class addVideo(object):
media_type = self.item.contentType
sql = 'INSERT OR IGNORE INTO rating (rating_id, media_id, media_type, rating_type, rating, votes) VALUES ( ?, ?, ?, ?, ?, ?)'
params = (self.rating_id, self.VideoId, media_type, 'tmdb', rating, votes)
n, records = execute_sql_kodi(sql, params, conn)
self.sql_actions.append([sql, params])
def set_ids(self):
self.uniqueID = get_id('uniqueid_id', 'uniqueid')
@@ -144,14 +127,13 @@ class addVideo(object):
self.uniqueIDType = 'kod'
sql = 'INSERT OR IGNORE INTO uniqueid (uniqueid_id, media_id, media_type, value, type) VALUES ( ?, ?, ?, ?, ?)'
params = [(self.uniqueID, self.VideoId, self.item.contentType, self.uniqueIdValue, self.uniqueIDType)]
n, records = execute_sql_kodi(sql, params, conn)
nun_records, records = execute_sql_kodi(sql, params, conn)
i = self.uniqueID + 1
for _id in ['imdb', 'tmdb', 'tvdb']:
if _id +'_id' in self.info:
params.append((i, self.VideoId, self.item.contentType, self.info[_id + '_id'], _id))
if params:
n, records = execute_sql_kodi(sql, params, conn)
self.sql_actions.append([sql, params])
def set_actors(self):
actor_id = get_id('actor_id', 'actor')
@@ -177,9 +159,9 @@ class addVideo(object):
actor_link_params.append((a_id, self.VideoId, self.item.contentType, actor[1], actor[3]))
# support.dbg()
if actor_params:
n, records = execute_sql_kodi(actor_sql, actor_params)
self.sql_actions.append([actor_sql, actor_params])
if actor_link_params:
n, records = execute_sql_kodi(actor_link_sql, actor_link_params)
self.sql_actions.append([actor_link_sql, actor_link_params])
directors = self.info.get('director','').split(', ')
directors_image = self.info.get('director_image',['' for d in directors])
@@ -199,10 +181,10 @@ class addVideo(object):
d_id = current_actors[l_actors.index(director)][0]
director_link_params.append((d_id, self.VideoId, self.item.contentType))
if director_params:
n, records = execute_sql_kodi(actor_sql, director_params)
self.sql_actions.append([actor_sql, director_params])
if director_link_params:
sql = 'INSERT OR IGNORE INTO director_link (actor_id, media_id, media_type) VALUES (?, ?, ?)'
n, records = execute_sql_kodi(sql, director_link_params)
self.sql_actions.append([sql, director_link_params])
writers = self.info.get('writer','').split(', ')
writers_image = self.info.get('writer_image',['' for w in writers])
@@ -222,61 +204,69 @@ class addVideo(object):
writer_link_params.append((d_id, self.VideoId, self.item.contentType))
if writer_params:
n, records = execute_sql_kodi(actor_sql, writer_params)
self.sql_actions.append([actor_sql, writer_params])
if writer_link_params:
sql = 'INSERT OR IGNORE INTO director_link (actor_id, media_id, media_type) VALUES (?, ?, ?)'
n, records = execute_sql_kodi(sql, writer_link_params)
self.sql_actions.append([sql, writer_link_params])
def set_country(self):
countrys = self.info.get('country','').split(', ')
if countrys:
for country in countrys:
countries = self.info.get('country','').split(', ')
counrty_link_params = []
if countries:
for country in countries:
sql = 'select country_id from country where name = "{}"'.format(country)
n, records = execute_sql_kodi(sql, conn=conn)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if records:
_id = records[0][0]
else:
_id = get_id('country_id', 'country')
sql = 'INSERT OR IGNORE INTO country (country_id, name) VALUES (?, ?)'
params = (_id, country)
n, records = execute_sql_kodi(sql, params, conn)
sql = 'INSERT OR IGNORE INTO country_link (country_id, media_id, media_type) VALUES (?, ?, ?)'
params = (_id, self.VideoId, self.item.contentType)
n, records = execute_sql_kodi(sql, params, conn)
nun_records, records = execute_sql_kodi(sql, params, conn)
counrty_link_params.append((_id, self.VideoId, self.item.contentType))
if counrty_link_params:
sql = 'INSERT OR IGNORE INTO country_link (country_id, media_id, media_type) VALUES (?, ?, ?)'
self.sql_actions.append([sql, counrty_link_params])
def set_genre(self):
genres = self.info.get('genre','').split(', ')
genre_link_params = []
if genres:
for genre in genres:
sql = 'select genre_id from genre where name = "{}"'.format(genre)
n, records = execute_sql_kodi(sql, conn=conn)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if records:
_id = records[0][0]
else:
_id = get_id('genre_id', 'genre')
sql = 'INSERT OR IGNORE INTO genre (genre_id, name) VALUES (?, ?)'
params = (_id, genre)
n, records = execute_sql_kodi(sql, params, conn)
sql = 'INSERT OR IGNORE INTO genre_link (genre_id, media_id, media_type) VALUES (?, ?, ?)'
params = (_id, self.VideoId, self.item.contentType)
n, records = execute_sql_kodi(sql, params, conn)
nun_records, records = execute_sql_kodi(sql, params, conn)
genre_link_params.append((_id, self.VideoId, self.item.contentType))
if genre_link_params:
sql = 'INSERT OR IGNORE INTO genre_link (genre_id, media_id, media_type) VALUES (?, ?, ?)'
self.sql_actions.append([sql, genre_link_params])
def set_studio(self):
studios = self.info.get('studio','').split(', ')
studio_link_params = []
if studios:
for studio in studios:
sql = 'select studio_id from studio where name = "{}"'.format(studio)
n, records = execute_sql_kodi(sql, conn=conn)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if records:
_id = records[0][0]
else:
_id = get_id('studio_id', 'studio')
sql = 'INSERT OR IGNORE INTO studio (studio_id, name) VALUES (?, ?)'
params = (_id, studio)
n, records = execute_sql_kodi(sql, params, conn)
sql = 'INSERT OR IGNORE INTO studio_link (studio_id, media_id, media_type) VALUES (?, ?, ?)'
params = (_id, self.VideoId, self.item.contentType)
n, records = execute_sql_kodi(sql, params, conn)
nun_records, records = execute_sql_kodi(sql, params, conn)
studio_link_params.append((_id, self.VideoId, self.item.contentType))
if studio_link_params:
sql = 'INSERT OR IGNORE INTO studio_link (studio_id, media_id, media_type) VALUES (?, ?, ?)'
self.sql_actions.append([sql, studio_link_params])
def set_movie(self):
posters, fanarts = get_images(self.item)
@@ -304,16 +294,13 @@ class addVideo(object):
self.idPath, # c23
self.idSet, # idSet
self.info.get('premiered')) # premiered
n, records = execute_sql_kodi(sql, params, conn)
if self.item.thumbnail:
self.art.append({'media_id':self.VideoId, 'media_type': 'movie', 'type':'poster', 'url':self.item.thumbnail})
if self.item.fanart:
self.art.append({'media_id':self.VideoId, 'media_type': 'movie', 'type':'fanart', 'url':self.item.fanart})
# payload = {"jsonrpc": "2.0", "method": "VideoLibrary.SetMovieDetails", "params": {"movieid": self.VideoId, "thumbnail":self.item.thumbnail, "art":{"poster": self.item.thumbnail, "fanart":self.item.fanart}}, "id": 1}
# get_data(payload)
# if self.idSet:
# payload = {"jsonrpc": "2.0", "method": "VideoLibrary.SetMovieSetDetails", "params": {"setid": self.idSet, "art":{"poster": self.info['setposters'][0], "fanart":self.info['setfanarts'][0]}}, "id": 1}
# get_data(payload)
self.sql_actions.append([sql, params])
@@ -343,14 +330,14 @@ class addVideo(object):
self.idPath, # c23
self.idSet, # idSet
self.info.get('premiered')) # premiered
n, records = execute_sql_kodi(sql, params, conn)
nun_records, records = execute_sql_kodi(sql, params, conn)
def set_art(self):
params = []
art_urls = []
_id = get_id('art_id', 'art')
sql = 'select media_id, media_type, type from art'
n, records = execute_sql_kodi(sql, conn=conn)
nun_records, records = execute_sql_kodi(sql, conn=conn)
if records:
art_urls = [[u[0], u[1], u[2]] for u in records]
for art in self.art:
@@ -359,12 +346,14 @@ class addVideo(object):
_id += 1
if params:
sql = 'INSERT OR IGNORE INTO art (art_id, media_id, media_type, type, url) VALUES (?, ?, ?, ?, ?)'
n, records = execute_sql_kodi(sql, params, conn)
self.sql_actions.append([sql, params])
def add_video(item):
if item.contentType == 'movie':
start = time()
addVideo(item=item)
logger.debug('TOTAL TIME:', time() - start)
else:
i = 0
seasons = videolibrarydb['season'][item.videolibrary_id]