Varken/tautulli.py

55 lines
1.8 KiB
Python
Raw Normal View History

2018-02-09 22:40:32 -08:00
import requests
import geohash
from datetime import datetime, timezone
from influxdb import InfluxDBClient
# noinspection PyUnresolvedReferences
from requests.packages.urllib3.exceptions import InsecureRequestWarning
# noinspection PyUnresolvedReferences
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
current_time = datetime.now(timezone.utc).astimezone().isoformat()
2018-02-09 22:54:27 -08:00
payload = {'apikey': 'xxxxxxxxxxxxxx', 'cmd': 'get_activity'}
activity = requests.get('https://plexpy.domain.tld/api/v2', params=payload).json()['response']['data']
2018-02-09 22:40:32 -08:00
sessions = {d['session_id']: d for d in activity['sessions']}
influx_payload = [
{
2018-02-09 23:46:58 -08:00
"measurement": "Tautulli",
2018-02-09 22:40:32 -08:00
"tags": {
2018-02-09 23:46:58 -08:00
"type": "stream_count"
2018-02-09 22:40:32 -08:00
},
"time": current_time,
"fields": {
2018-02-09 23:46:58 -08:00
"current_streams": int(activity['stream_count'])
2018-02-09 22:40:32 -08:00
}
}
]
for session in sessions.keys():
lookup = requests.get('http://freegeoip.net/json/{}'.format(sessions[session]['ip_address_public'])).json()
influx_payload.append(
{
2018-02-09 23:46:58 -08:00
"measurement": "Tautulli",
2018-02-09 22:40:32 -08:00
"tags": {
"type": "Session",
2018-02-09 23:46:58 -08:00
"region_code": lookup['region_code'],
"name": sessions[session]['friendly_name']
2018-02-09 22:40:32 -08:00
},
"time": current_time,
"fields": {
"name": sessions[session]['friendly_name'],
"title": sessions[session]['full_title'],
"quality": '{}p'.format(sessions[session]['video_resolution']),
"transcode_decision": sessions[session]['transcode_decision'],
"location": lookup['city'],
}
}
)
influx = InfluxDBClient('grafana.domain.tld', 8086, 'root', 'root', 'plex')
influx.write_points(influx_payload)