From 93748950702dfed1dc9464c92c0c16b35c4a716c Mon Sep 17 00:00:00 2001 From: Ian Richardson Date: Tue, 29 Jan 2019 17:16:27 -0600 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=BA=20fix=20for=20when=20tmdb=20data?= =?UTF-8?q?=20not=20present?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- custom_components/sensor/trakt.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/custom_components/sensor/trakt.py b/custom_components/sensor/trakt.py index bc48845..8fa7eb2 100644 --- a/custom_components/sensor/trakt.py +++ b/custom_components/sensor/trakt.py @@ -114,8 +114,8 @@ def save_token(hass, token): data_file.write(json.dumps(token)) -class TraktMyShowCalendarSensor(Entity): - """Representation of a Trakt My Show Calendar sensor.""" +class TraktUpcomingCalendarSensor(Entity): + """Representation of a Trakt Upcoming Calendar sensor.""" def __init__(self, hass, config, token): """Initialize the sensor.""" @@ -181,6 +181,7 @@ def update(self): release = '$day, $time' else: release = '$day, $date $time' + card_item = { 'airdate': show.airs_at.isoformat() + 'Z', 'release': release, @@ -188,11 +189,11 @@ def update(self): 'title': show.show, 'episode': show.title, 'number': 'S' + str(show.season) + 'E' + str(show.number), - 'rating': tmdb_json['vote_average'], - 'poster': image_url % ('500', tmdb_json['poster_path']), - 'fanart': image_url % ('780', tmdb_json['backdrop_path']), - 'runtime': tmdb_json['episode_run_time'][0] if len(tmdb_json['episode_run_time']) > 0 else '', - 'studio': tmdb_json['networks'][0]['name'] if len(tmdb_json['networks']) > 0 else '' + 'rating': tmdb_json.get('vote_average', ''), + 'poster': image_url % ('500', tmdb_json.get('poster_path', '')), + 'fanart': image_url % ('780', tmdb_json.get('backdrop_path', '')), + 'runtime': tmdb_json.get('episode_run_time')[0] if len(tmdb_json.get('episode_run_time')) > 0 else '', + 'studio': tmdb_json.get('networks')[0].get('name', '') if len(tmdb_json.get('networks')) > 0 else '' } card_json.append(card_item)