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

Fix relay lookup failure using geoip

Browse source
This commit is contained in:
samwiseg0 2018-11-19 17:40:12 -05:00
parent cd56692834
commit a312edabb8
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tautulli.py
View file

@ -2,12 +2,15 @@ import os
import tarfile import tarfile
import urllib.request import urllib.request
import time import time
from datetime import datetime, timezone
import geoip2.database import geoip2.database
from influxdb import InfluxDBClient
import requests import requests
import configuration import configuration
from geoip2.errors import AddressNotFoundError
from influxdb import InfluxDBClient
from datetime import datetime, timezone
CURRENT_TIME = datetime.now(timezone.utc).astimezone().isoformat() CURRENT_TIME = datetime.now(timezone.utc).astimezone().isoformat()
PAYLOAD = {'apikey': configuration.tautulli_api_key, 'cmd': 'get_activity'} PAYLOAD = {'apikey': configuration.tautulli_api_key, 'cmd': 'get_activity'}
@ -74,7 +77,7 @@ INFLUX_PAYLOAD = [
for session in SESSIONS.keys(): for session in SESSIONS.keys():
try: try:
geodata = geo_lookup(SESSIONS[session]['ip_address_public']) geodata = geo_lookup(SESSIONS[session]['ip_address_public'])
except (ValueError, geoip2.errors.AddressNotFoundError): except (ValueError, AddressNotFoundError):
if configuration.tautulli_failback_ip: if configuration.tautulli_failback_ip:
geodata = geo_lookup(configuration.tautulli_failback_ip) geodata = geo_lookup(configuration.tautulli_failback_ip)
else: else: