From 5827bb2c976688d79d00000078809a768804c793 Mon Sep 17 00:00:00 2001 From: "Nicholas St. Germain" Date: Wed, 24 Apr 2019 17:50:09 -0500 Subject: [PATCH] fix dict check for lidarr --- varken/lidarr.py | 10 ++++++---- varken/structures.py | 2 +- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/varken/lidarr.py b/varken/lidarr.py index 8bcd070..e631372 100644 --- a/varken/lidarr.py +++ b/varken/lidarr.py @@ -48,10 +48,12 @@ class LidarrAPI(object): # Add Album to missing list if album is not complete for album in albums: - if album.statistics['percentOfTracks'] != 100: - influx_albums.append((album.title, album.releaseDate, album.artist['artistName'], album.id, - album.statistics['percentOfTracks'], - f"{album.statistics['trackFileCount']}/{album.statistics['TrackCount']}")) + percent_of_tracks = album.statistics.get('percentOfTracks', 0) + if percent_of_tracks != 100: + influx_albums.append( + (album.title, album.releaseDate, album.artist['artistName'], album.id,percent_of_tracks, + f"{album.statistics.get('trackFileCount', 0)}/{album.statistics.get('trackCount', 0)}") + ) for title, release_date, artist_name, album_id, percent_complete, complete_count in influx_albums: hash_id = hashit(f'{self.server.id}{title}{album_id}') diff --git a/varken/structures.py b/varken/structures.py index 9225adb..94e2478 100644 --- a/varken/structures.py +++ b/varken/structures.py @@ -501,5 +501,5 @@ class LidarrAlbum(NamedTuple): artist: dict = None images: list = None links: list = None - statistics: dict = None + statistics: dict = {} id: int = None