From fa4b5eec6360238711dcc7eef2b65af067fc7b83 Mon Sep 17 00:00:00 2001 From: JonnyWong16 Date: Mon, 7 Oct 2019 13:02:22 -0700 Subject: [PATCH] Use itemgetter_with_default helper function for Tautulli sessions --- varken/tautulli.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/varken/tautulli.py b/varken/tautulli.py index af0d979..effbfc7 100644 --- a/varken/tautulli.py +++ b/varken/tautulli.py @@ -1,12 +1,11 @@ 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 from influxdb.exceptions import InfluxDBClientError 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): @@ -41,10 +40,10 @@ class TautulliAPI(object): if session.get('_cache_time'): del session['_cache_time'] - fields = itemgetter(*TautulliStream._fields) + fields = itemgetter_with_default(*TautulliStream._field_defaults) try: - sessions = [TautulliStream(*fields(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