mirror of
https://github.com/Tautulli/Tautulli.git
synced 2025-07-08 06:00:51 -07:00
Update grandparent/parent/child rating keys
* Updates all rating keys for show/season/episode or artist/album/track
This commit is contained in:
Jonathan Wong
parent
339361d15c
commit
c7d51b90c6
3 changed files with 297 additions and 33 deletions
|
|
@ -842,35 +842,127 @@ class DataFactory(object):
|
|||
|
||||
return query
|
||||
|
||||
def update_rating_key(self, old_rating_key='', new_rating_key='', media_type=''):
|
||||
def get_rating_keys_list(self, rating_key='', media_type=''):
|
||||
monitor_db = database.MonitorDatabase()
|
||||
|
||||
if new_rating_key.isdigit():
|
||||
logger.info(u"PlexPy DataFactory :: Updating rating key %s to %s in the session history database." % (old_rating_key, new_rating_key))
|
||||
if media_type == 'movie' or media_type == 'episode' or media_type == 'track':
|
||||
monitor_db.action('UPDATE session_history SET rating_key = ? WHERE rating_key = ?', [new_rating_key, old_rating_key])
|
||||
monitor_db.action('UPDATE session_history_media_info SET rating_key = ? WHERE rating_key = ?', [new_rating_key, old_rating_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET rating_key = ? WHERE rating_key = ?', [new_rating_key, old_rating_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET thumb = replace(thumb, ?, ?) WHERE thumb LIKE "/library/metadata/%s/thumb/%%"' % old_rating_key,
|
||||
[old_rating_key, new_rating_key])
|
||||
if media_type == 'movie':
|
||||
key_list = {0: {'rating_key': int(rating_key)}}
|
||||
return key_list
|
||||
|
||||
if media_type == 'season' or media_type == 'album':
|
||||
monitor_db.action('UPDATE session_history SET parent_rating_key = ? WHERE parent_rating_key = ?', [new_rating_key, old_rating_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET parent_rating_key = ? WHERE parent_rating_key = ?', [new_rating_key, old_rating_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET parent_thumb = replace(parent_thumb, ?, ?) WHERE parent_thumb LIKE "/library/metadata/%s/thumb/%%"' % old_rating_key,
|
||||
[old_rating_key, new_rating_key])
|
||||
if media_type == 'show' or media_type == 'season' or media_type == 'episode':
|
||||
match_type = 'index'
|
||||
elif media_type == 'artist' or media_type == 'album' or media_type == 'track':
|
||||
match_type = 'title'
|
||||
|
||||
if media_type == 'show' or media_type == 'artist':
|
||||
monitor_db.action('UPDATE session_history SET grandparent_rating_key = ? WHERE grandparent_rating_key = ?', [new_rating_key, old_rating_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET grandparent_rating_key = ? WHERE grandparent_rating_key = ?', [new_rating_key, old_rating_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET grandparent_thumb = replace(grandparent_thumb, ?, ?) WHERE grandparent_thumb LIKE "/library/metadata/%s/thumb/%%"' % old_rating_key,
|
||||
[old_rating_key, new_rating_key])
|
||||
# Get the grandparent rating key
|
||||
try:
|
||||
query = 'SELECT rating_key, parent_rating_key, grandparent_rating_key ' \
|
||||
'FROM session_history_metadata ' \
|
||||
'WHERE rating_key = ? ' \
|
||||
'OR parent_rating_key = ? ' \
|
||||
'OR grandparent_rating_key = ? ' \
|
||||
'LIMIT 1'
|
||||
result = monitor_db.select(query=query, args=[rating_key, rating_key, rating_key])
|
||||
|
||||
if media_type == 'movie' or media_type == 'show':
|
||||
monitor_db.action('UPDATE session_history_metadata SET art = replace(art, ?, ?) WHERE art LIKE "/library/metadata/%s/art/%%"' % old_rating_key,
|
||||
[old_rating_key, new_rating_key])
|
||||
grandparent_rating_key = result[0]['grandparent_rating_key']
|
||||
|
||||
return 'Updated rating key %s to %s.' % (old_rating_key, new_rating_key)
|
||||
else:
|
||||
return 'Unable to update rating key. Input new_rating_key not valid.'
|
||||
except:
|
||||
logger.warn("Unable to execute database query.")
|
||||
return {}
|
||||
|
||||
query = 'SELECT rating_key, parent_rating_key, grandparent_rating_key, title, parent_title, grandparent_title, ' \
|
||||
'media_index, parent_media_index ' \
|
||||
'FROM session_history_metadata ' \
|
||||
'WHERE {0} = ? ' \
|
||||
'GROUP BY {1} '
|
||||
|
||||
# get grandparent_rating_keys
|
||||
grandparents = {}
|
||||
result = monitor_db.select(query=query.format('grandparent_rating_key', 'grandparent_rating_key'),
|
||||
args=[grandparent_rating_key])
|
||||
for item in result:
|
||||
# get parent_rating_keys
|
||||
parents = {}
|
||||
result = monitor_db.select(query=query.format('grandparent_rating_key', 'parent_rating_key'),
|
||||
args=[item['grandparent_rating_key']])
|
||||
for item in result:
|
||||
# get rating_keys
|
||||
children = {}
|
||||
result = monitor_db.select(query=query.format('parent_rating_key', 'rating_key'),
|
||||
args=[item['parent_rating_key']])
|
||||
for item in result:
|
||||
key = item['media_index'] if match_type == 'index' else item['title']
|
||||
children.update({key: {'rating_key': item['rating_key']}})
|
||||
|
||||
key = item['parent_media_index'] if match_type == 'index' else item['parent_title']
|
||||
parents.update({key:
|
||||
{'rating_key': item['parent_rating_key'],
|
||||
'children': children}
|
||||
})
|
||||
|
||||
key = 0 if match_type == 'index' else item['grandparent_title']
|
||||
grandparents.update({key:
|
||||
{'rating_key': item['grandparent_rating_key'],
|
||||
'children': parents}
|
||||
})
|
||||
|
||||
key_list = grandparents
|
||||
|
||||
return key_list
|
||||
|
||||
def update_rating_key(self, old_key_list='', new_key_list='', media_type=''):
|
||||
monitor_db = database.MonitorDatabase()
|
||||
|
||||
# function to map rating keys pairs
|
||||
def get_pairs(old, new):
|
||||
pairs = {}
|
||||
for k, v in old.iteritems():
|
||||
if k in new:
|
||||
if v['rating_key'] != new[k]['rating_key']:
|
||||
pairs.update({v['rating_key']: new[k]['rating_key']})
|
||||
if 'children' in old[k]:
|
||||
pairs.update(get_pairs(old[k]['children'], new[k]['children']))
|
||||
|
||||
return pairs
|
||||
|
||||
# map rating keys pairs
|
||||
mapping = {}
|
||||
if old_key_list and new_key_list:
|
||||
mapping = get_pairs(old_key_list, new_key_list)
|
||||
|
||||
if mapping:
|
||||
logger.info(u"PlexPy DataFactory :: Updating rating keys in the database.")
|
||||
for old_key, new_key in mapping.iteritems():
|
||||
# check rating_key (3 tables)
|
||||
monitor_db.action('UPDATE session_history SET rating_key = ? WHERE rating_key = ?', [new_key, old_key])
|
||||
monitor_db.action('UPDATE session_history_media_info SET rating_key = ? WHERE rating_key = ?', [new_key, old_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET rating_key = ? WHERE rating_key = ?', [new_key, old_key])
|
||||
|
||||
# check parent_rating_key (2 tables)
|
||||
monitor_db.action('UPDATE session_history SET parent_rating_key = ? WHERE parent_rating_key = ?', [new_key, old_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET parent_rating_key = ? WHERE parent_rating_key = ?', [new_key, old_key])
|
||||
|
||||
# check grandparent_rating_key (2 tables)
|
||||
monitor_db.action('UPDATE session_history SET grandparent_rating_key = ? WHERE grandparent_rating_key = ?', [new_key, old_key])
|
||||
monitor_db.action('UPDATE session_history_metadata SET grandparent_rating_key = ? WHERE grandparent_rating_key = ?', [new_key, old_key])
|
||||
|
||||
# check thumb (1 table)
|
||||
monitor_db.action('UPDATE session_history_metadata SET thumb = replace(thumb, ?, ?) WHERE thumb LIKE "/library/metadata/%s/thumb/%%"' % old_key,
|
||||
[new_key, old_key])
|
||||
|
||||
# check parent_thumb (1 table)
|
||||
monitor_db.action('UPDATE session_history_metadata SET parent_thumb = replace(parent_thumb, ?, ?) WHERE parent_thumb LIKE "/library/metadata/%s/thumb/%%"' % old_key,
|
||||
[new_key, old_key])
|
||||
|
||||
# check grandparent_thumb (1 table)
|
||||
monitor_db.action('UPDATE session_history_metadata SET grandparent_thumb = replace(grandparent_thumb, ?, ?) WHERE grandparent_thumb LIKE "/library/metadata/%s/thumb/%%"' % old_key,
|
||||
[new_key, old_key])
|
||||
|
||||
# check art (1 table)
|
||||
monitor_db.action('UPDATE session_history_metadata SET art = replace(art, ?, ?) WHERE art LIKE "/library/metadata/%s/art/%%"' % old_key,
|
||||
[new_key, old_key])
|
||||
|
||||
#return 'Updated rating key in database.'
|
||||
#else:
|
||||
#return 'No updated rating key needed in database. No changes were made.'
|
||||
return mapping
|
||||
Loading…
Add table
Add a link
Reference in a new issue