From ea563f7a8e7d084fc3e825f097ee058d0506f0c0 Mon Sep 17 00:00:00 2001 From: JonnyWong16 Date: Mon, 7 Oct 2019 11:35:27 -0700 Subject: [PATCH] Fix to only use NamedTuple fields from TautulliStream * Allow new fields to be added to Tautulli API without breaking Varken --- varken/tautulli.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/varken/tautulli.py b/varken/tautulli.py index c0d11a2..af0d979 100644 --- a/varken/tautulli.py +++ b/varken/tautulli.py @@ -1,4 +1,5 @@ from logging import getLogger +from operator import itemgetter from requests import Session, Request from geoip2.errors import AddressNotFoundError from datetime import datetime, timezone, date, timedelta @@ -40,8 +41,10 @@ class TautulliAPI(object): if session.get('_cache_time'): del session['_cache_time'] + fields = itemgetter(*TautulliStream._fields) + try: - sessions = [TautulliStream(**session) for session in get['sessions']] + sessions = [TautulliStream(*fields(session)) for session in get['sessions']] except TypeError as e: self.logger.error('TypeError has occurred : %s while creating TautulliStream structure', e) return