d_mcknight/Varken
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use itemgetter_with_default helper function for Tautulli sessions

Browse source
This commit is contained in:
JonnyWong16 2019-10-07 13:02:22 -07:00
parent 16e716e270
commit fa4b5eec63
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
varken/tautulli.py
View file

@ -1,12 +1,11 @@
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
from influxdb.exceptions import InfluxDBClientError from influxdb.exceptions import InfluxDBClientError
from varken.structures import TautulliStream from varken.structures import TautulliStream
from varken.helpers import hashit, connection_handler from varken.helpers import hashit, connection_handler, itemgetter_with_default
class TautulliAPI(object): class TautulliAPI(object):
@ -41,10 +40,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) fields = itemgetter_with_default(*TautulliStream._field_defaults)
try: try:
sessions = [TautulliStream(*fields(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