#!/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 Ombi movie requests")
    print("graph_title Ombi movie requests")
    print("graph_category ombi_movie_requests")

    print("ombi_requests_movies_requested.label requested movies")
    print("ombi_requests_movies_available.label available movies")
    print("ombi_requests_movies_unavailable.label unavailable movies")

    exit()

response = get(
    "{}/{}".format(environ["host"], "api/v1/Request/movie"),
    headers={"Accept": "application/json", "apikey": environ["apikey"]}
)

response.raise_for_status()

available = 0
requested = 0
unavailable = 0

for item in response.json():
    if item["available"] == True:
        available = available + 1
    else:
        unavailable = unavailable + 1

    requested = requested + 1

print "ombi_requests_movies_requested.value", requested
print "ombi_requests_movies_available.value", available
print "ombi_requests_movies_unavailable.value", unavailable