Fix KeyError

This commit is contained in:
croneter 2018-11-03 16:56:51 +01:00
parent 4cbad1f1f7
commit b9a8d66e3d

View file

@ -46,17 +46,13 @@ class PlexDBBase(object):
""" """
answ = None answ = None
if plex_type == v.PLEX_TYPE_MOVIE: if plex_type == v.PLEX_TYPE_MOVIE:
entry = self.movie(plex_id) answ = self.movie(plex_id)
answ = self.entry_to_movie(entry)
elif plex_type == v.PLEX_TYPE_EPISODE: elif plex_type == v.PLEX_TYPE_EPISODE:
entry = self.episode(plex_id) answ = self.episode(plex_id)
answ = self.entry_to_episode(entry)
elif plex_type == v.PLEX_TYPE_SHOW: elif plex_type == v.PLEX_TYPE_SHOW:
entry = self.show(plex_id) answ = self.show(plex_id)
answ = self.entry_to_show(entry)
elif plex_type == v.PLEX_TYPE_SEASON: elif plex_type == v.PLEX_TYPE_SEASON:
entry = self.season(plex_id) answ = self.season(plex_id)
answ = self.entry_to_season(entry)
else: else:
# SLOW - lookup plex_id in all our tables # SLOW - lookup plex_id in all our tables
for kind in (v.PLEX_TYPE_MOVIE, for kind in (v.PLEX_TYPE_MOVIE,
@ -64,10 +60,8 @@ class PlexDBBase(object):
v.PLEX_TYPE_EPISODE, v.PLEX_TYPE_EPISODE,
v.PLEX_TYPE_SEASON): v.PLEX_TYPE_SEASON):
method = getattr(self, kind) method = getattr(self, kind)
entry = method(plex_id) answ = method(plex_id)
if entry: if answ:
method = getattr(self, 'entry_to_%s' % kind)
answ = method(entry)
break break
return answ return answ