Fix AttributeError for songs

This commit is contained in:
croneter 2018-11-13 14:42:48 +01:00
parent 5a30b1c86a
commit 40e3a852a2

View file

@ -134,10 +134,10 @@ class Music(object):
return return
self.cursor.execute('SELECT * FROM track WHERE plex_id = ? LIMIT 1', self.cursor.execute('SELECT * FROM track WHERE plex_id = ? LIMIT 1',
(plex_id, )) (plex_id, ))
return self.entry_to_song(self.cursor.fetchone()) return self.entry_to_track(self.cursor.fetchone())
@staticmethod @staticmethod
def entry_to_song(entry): def entry_to_track(entry):
if not entry: if not entry:
return return
return { return {
@ -216,7 +216,7 @@ class Music(object):
""" """
self.cursor.execute('SELECT * FROM track WHERE album_id = ?', self.cursor.execute('SELECT * FROM track WHERE album_id = ?',
(plex_id, )) (plex_id, ))
return (self.entry_to_song(x) for x in self.cursor) return (self.entry_to_track(x) for x in self.cursor)
def song_by_artist(self, plex_id): def song_by_artist(self, plex_id):
""" """
@ -225,7 +225,7 @@ class Music(object):
""" """
self.cursor.execute('SELECT * FROM track WHERE artist_id = ?', self.cursor.execute('SELECT * FROM track WHERE artist_id = ?',
(plex_id, )) (plex_id, ))
return (self.entry_to_song(x) for x in self.cursor) return (self.entry_to_track(x) for x in self.cursor)
def album_by_artist(self, plex_id): def album_by_artist(self, plex_id):
""" """