#!/usr/bin/python

from sys import argv, exit
from subprocess import check_output
from re import search, sub
from json import loads, dumps
from os import environ
from requests import get

if len(argv) > 1:
    print("graph_vlabel Tautulli activity")
    print("graph_title Tautulli activity")
    print("graph_category tautulli_activity")

    print("tautulli_stream_count.label Stream count")
    print("tautulli_total_bandwidth.label Total bandwidth")
    print("tautulli_stream_count_transcode.label Stream count transcode")
    print("tautulli_wan_bandwidth.label WAN bandwidth")
    print("tautulli_stream_count_direct_play.label Stream count directplay")
    print("tautulli_lan_bandwidth.label LAN bandwidth")
    print("tautulli_stream_count_direct_stream.label Stream count direct stream")

    exit()

response = get(
    "{}/api/v2?apikey={}&cmd={}".format(environ["host"], environ["apikey"], "get_activity"),
    headers={"Accept": "application/json"}
)
response.raise_for_status()

print "tautulli_stream_count.value", response.json()["response"]["data"]["stream_count"]
print "tautulli_total_bandwidth.value", response.json()["response"]["data"]["total_bandwidth"]
print "tautulli_stream_count_transcode.value", response.json()["response"]["data"]["stream_count_transcode"]
print "tautulli_wan_bandwidth.value", response.json()["response"]["data"]["wan_bandwidth"]
print "tautulli_stream_count_direct_play.value", response.json()["response"]["data"]["stream_count_direct_play"]
print "tautulli_lan_bandwidth.value", response.json()["response"]["data"]["lan_bandwidth"]
print "tautulli_stream_count_direct_stream.value", response.json()["response"]["data"]["stream_count_direct_stream"]