Fix boxsets

This commit is contained in:
angelblue05 2018-09-25 00:19:52 -05:00
parent 2148fd152e
commit 2a1a4b8a7c

View file

@ -30,6 +30,7 @@ class Movies(KodiDb):
self.emby_db = emby_db.EmbyDatabase(embydb.cursor) self.emby_db = emby_db.EmbyDatabase(embydb.cursor)
self.objects = Objects() self.objects = Objects()
self.item_ids = []
KodiDb.__init__(self, videodb.cursor) KodiDb.__init__(self, videodb.cursor)
@ -126,6 +127,8 @@ class Movies(KodiDb):
self.add_streams(*values(obj, QU.add_streams_obj)) self.add_streams(*values(obj, QU.add_streams_obj))
self.artwork.add(obj['Artwork'], obj['MovieId'], "movie") self.artwork.add(obj['Artwork'], obj['MovieId'], "movie")
self.item_ids.append(obj['Id'])
def movie_add(self, obj): def movie_add(self, obj):
''' Add object to kodi. ''' Add object to kodi.
@ -239,15 +242,13 @@ class Movies(KodiDb):
''' Add or removes movies based on the current movies found in the boxset. ''' Add or removes movies based on the current movies found in the boxset.
''' '''
obj['Current'] = []
try: try:
movies = dict(self.emby_db.get_item_id_by_parent_id(*values(obj, QUEM.get_item_id_by_parent_boxset_obj))) current = self.emby_db.get_item_id_by_parent_id(*values(obj, QUEM.get_item_id_by_parent_boxset_obj))
movies = dict(current)
except ValueError: except ValueError:
movies = {} movies = {}
for movie in movies: obj['Current'] = movies
obj['Current'].append(movie)
for all_movies in server.get_movies_by_boxset(obj['Id']): for all_movies in server.get_movies_by_boxset(obj['Id']):
for movie in all_movies['Items']: for movie in all_movies['Items']:
@ -263,13 +264,13 @@ class Movies(KodiDb):
continue continue
if temp_obj['Id'] not in movies: if temp_obj['Id'] not in obj['Current']:
self.set_boxset(*values(temp_obj, QU.update_movie_set_obj)) self.set_boxset(*values(temp_obj, QU.update_movie_set_obj))
self.emby_db.update_parent_id(*values(temp_obj, QUEM.update_parent_movie_obj)) self.emby_db.update_parent_id(*values(temp_obj, QUEM.update_parent_movie_obj))
LOG.info("ADD to boxset [%s/%s] %s: %s to boxset", temp_obj['SetId'], temp_obj['MovieId'], temp_obj['Title'], temp_obj['Id']) LOG.info("ADD to boxset [%s/%s] %s: %s to boxset", temp_obj['SetId'], temp_obj['MovieId'], temp_obj['Title'], temp_obj['Id'])
else: else:
obj['Current'].remove(temp_obj['Id']) obj['Current'].pop(temp_obj['Id'])
def boxsets_reset(self): def boxsets_reset(self):