#!/usr/bin/env python3

import os
import sys
from comet_server import CometServer

def config():
	print('graph_title Comet Job Status 24h')
	print('graph_args --base 1000 -l 0 ')
	print('graph_category Comet')
	print('skipped.label Skipped')
	print('skipped.draw AREASTACK')
	print('skipped.min 0')
	print('skipped.colour a562d3')
	print('running.label Running')
	print('running.draw AREASTACK')
	print('running.min 0')
	print('running.colour b9b9b9')
	print('missed.label Missed')
	print('missed.draw AREASTACK')
	print('missed.min 0')
	print('missed.colour dcdcdc')
	print('warning.label Warning')
	print('warning.draw AREASTACK')
	print('warning.min 0')
	print('warning.colour fa9545')
	print('error.label Error')
	print('error.draw AREASTACK')
	print('error.min 0')
	print('error.colour cf4848')
	print('success.label Success')
	print('success.draw AREASTACK')
	print('success.min 0')
	print('success.colour 58b154')

def main():
	cs = CometServer()
	statusCount = cs.getJobsStatus()
	print("success.value " + str(statusCount["Success"]))
	print("error.value " + str(statusCount["Error"]))
	print("warning.value " + str(statusCount["Warning"]))
	print("missed.value " + str(statusCount["Missed"]))
	print("running.value " + str(statusCount["Running"]))
	print("skipped.value " + str(statusCount["Skipped"]))

if __name__ == '__main__':
	if len(sys.argv) > 1 and sys.argv[1] == 'config':
		config()
	else:
		main()