From a312edabb8366b550a91fdbbe672b58827ce5541 Mon Sep 17 00:00:00 2001 From: samwiseg0 Date: Mon, 19 Nov 2018 17:40:12 -0500 Subject: [PATCH] Fix relay lookup failure using geoip --- tautulli.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tautulli.py b/tautulli.py index e92fb4f..9fc06be 100644 --- a/tautulli.py +++ b/tautulli.py @@ -2,12 +2,15 @@ import os import tarfile import urllib.request import time -from datetime import datetime, timezone import geoip2.database -from influxdb import InfluxDBClient import requests import configuration +from geoip2.errors import AddressNotFoundError +from influxdb import InfluxDBClient +from datetime import datetime, timezone + + CURRENT_TIME = datetime.now(timezone.utc).astimezone().isoformat() PAYLOAD = {'apikey': configuration.tautulli_api_key, 'cmd': 'get_activity'} @@ -74,7 +77,7 @@ INFLUX_PAYLOAD = [ for session in SESSIONS.keys(): try: geodata = geo_lookup(SESSIONS[session]['ip_address_public']) - except (ValueError, geoip2.errors.AddressNotFoundError): + except (ValueError, AddressNotFoundError): if configuration.tautulli_failback_ip: geodata = geo_lookup(configuration.tautulli_failback_ip) else: