Fix to only use NamedTuple fields from TautulliStream (#156)

Fix to only use NamedTuple fields from TautulliStream
This commit is contained in:
samwiseg0 2019-10-07 15:15:00 -04:00 committed by GitHub
commit 1f15d07330
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,4 +1,5 @@
from logging import getLogger from logging import getLogger
from operator import itemgetter
from requests import Session, Request from requests import Session, Request
from geoip2.errors import AddressNotFoundError from geoip2.errors import AddressNotFoundError
from datetime import datetime, timezone, date, timedelta from datetime import datetime, timezone, date, timedelta
@ -40,8 +41,10 @@ class TautulliAPI(object):
if session.get('_cache_time'): if session.get('_cache_time'):
del session['_cache_time'] del session['_cache_time']
fields = itemgetter(*TautulliStream._fields)
try: try:
sessions = [TautulliStream(**session) for session in get['sessions']] sessions = [TautulliStream(*fields(session)) for session in get['sessions']]
except TypeError as e: except TypeError as e:
self.logger.error('TypeError has occurred : %s while creating TautulliStream structure', e) self.logger.error('TypeError has occurred : %s while creating TautulliStream structure', e)
return return