#!/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 libraries")
    print("graph_title Tautulli libraries")
    print("graph_category tautulli_libraries")

    print("tautulli_shows_de.label Serien")
    print("tautulli_shows_de_seasons.label Staffeln")
    print("tautulli_shows_de_episodes.label Folgen")
    print("tautulli_shows_de_4k.label Serien 4k")
    print("tautulli_shows_de_4k_seasons.label Staffeln 4k")
    print("tautulli_shows_de_4k_episodes.label Folgen 4k")
    print("tautulli_shows_en.label Series")
    print("tautulli_shows_en_seasons.label Seasons")
    print("tautulli_shows_en_episodes.label Episodes")
    print("tautulli_shows_en_4k.label Series 4k")
    print("tautulli_shows_en_4k_seasons.label Seasons 4k")
    print("tautulli_shows_en_4k_episodes.label Episodes 4k")
    print("tautulli_movies_de.label Movies")
    print("tautulli_movies_de_4k.label Filme 4k")
    print("tautulli_movies_en.label Movies")
    print("tautulli_movies_en_4k.label Movies")
    
    exit()

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

for item in response.json()['response']['data']:
    if item["section_id"] == environ["shows_de"]:
        print "tautulli_shows_de.value", item["count"]
        print "tautulli_shows_de_seasons.value", item["parent_count"]
        print "tautulli_shows_de_episodes.value", item["child_count"]
    
    elif item["section_id"] == environ["shows_de_4k"]:
        print "tautulli_shows_de_4k.value", item["count"]
        print "tautulli_shows_de_4k_seasons.value", item["parent_count"]
        print "tautulli_shows_de_4k_episodes.value", item["child_count"]

    elif item["section_id"] == environ["shows_en"]:
        print "tautulli_shows_en.value", item["count"]
        print "tautulli_shows_en_seasons.value", item["parent_count"]
        print "tautulli_shows_en_episodes.value", item["child_count"]

    elif item["section_id"] == environ["shows_en_4k"]:
        print "tautulli_shows_en_4k.value", item["count"]
        print "tautulli_shows_en_4k_seasons.value", item["parent_count"]
        print "tautulli_shows_en_4k_episodes.value", item["child_count"]

    elif item["section_id"] == environ["movies_de"]:
        print "tautulli_movies_de.value", item["count"]

    elif item["section_id"] == environ["movies_de_4k"]:
        print "tautulli_movies_de_4k.value", item["count"]

    elif item["section_id"] == environ["movies_en"]:
        print "tautulli_movies_en.value", item["count"]

    elif item["section_id"] == environ["movies_en_4k"]:
        print "tautulli_movies_en_4k.value", item["count"]