import json
import yaml
import urllib2

def run(params):
    try:
        url = params.get('url')
        bases_servidor = params.get('databases')
        limite = params.get('limite')
        total_server = 0
        databases = 0

        f = urllib2.urlopen(url)
        load = json.load(f)

        for i in load['result']:
            databases += i['databases']
            total_server += 1

        capacity = total_server * bases_servidor
        bases_instaladas = databases

        if capacity - limite > bases_instaladas:
            return [0, 'OK: Capacity robbs OK']
        elif (capacity - bases_instaladas) < (limite / 2):
            return [2, "CRITICAL: Capacity atingido: Instalar novas robbs"]
        else:
            return [1, "WARNING: Capacity atingido: Instalar novas robbs"]

    except Exception, e:
        return [1, 'WARNING: {0}'.format(str(e))]

def  __run__(params):
    return run(params)