diff --git a/varken/sonarr.py b/varken/sonarr.py index 335ea84..1ff0263 100644 --- a/varken/sonarr.py +++ b/varken/sonarr.py @@ -4,6 +4,7 @@ from datetime import datetime, timezone, date, timedelta from varken.helpers import hashit, connection_handler from varken.structures import Queue, TVShow + class SonarrAPI(object): def __init__(self, server, dbmanager): # Set Time of initialization @@ -34,7 +35,11 @@ class SonarrAPI(object): return # Iteratively create a list of TVShow Objects from response json - tv_shows = [TVShow(**show) for show in get] + try: + tv_shows = [TVShow(**show) for show in get] + except TypeError as e: + logger.error('TypeError has occurred : %s', e) + return # Add show to missing list if file does not exist for show in tv_shows: diff --git a/varken/tautulli.py b/varken/tautulli.py index 5931179..ebd2573 100644 --- a/varken/tautulli.py +++ b/varken/tautulli.py @@ -7,8 +7,6 @@ from requests import Session, Request from varken.helpers import geo_lookup, hashit, connection_handler from varken.structures import TautulliStream -logger = logging.getLogger() - class TautulliAPI(object): def __init__(self, server, dbmanager): @@ -19,6 +17,7 @@ class TautulliAPI(object): self.session = Session() self.session.params = {'apikey': self.server.api_key, 'cmd': 'get_activity'} self.endpoint = '/api/v2' + self.logger = logging.getLogger() def __repr__(self): return "".format(self.server.id) @@ -37,7 +36,7 @@ class TautulliAPI(object): try: sessions = [TautulliStream(**session) for session in get['sessions']] except TypeError as e: - logger.error('TypeError has occured : %s', e) + self.logger.error('TypeError has occurred : %s', e) return for session in sessions: