103 lines
3.1 KiB
Python
Executable File
103 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import requests
|
|
import traceback
|
|
import time
|
|
import json
|
|
import urllib2
|
|
import pytz
|
|
import hashlib
|
|
from datetime import datetime, timedelta
|
|
from itertools import islice
|
|
|
|
from hosted import CONFIG, NODE
|
|
|
|
CONFIG.restart_on_update()
|
|
|
|
def current_time():
|
|
timezone = pytz.timezone("Europe/Berlin")
|
|
now = datetime.utcnow()
|
|
now = now.replace(tzinfo=pytz.utc)
|
|
now = now.astimezone(timezone)
|
|
now = now.replace(tzinfo=None)
|
|
return now
|
|
|
|
def to_unixtimestamp(dt):
|
|
return int(time.mktime(dt.timetuple()))
|
|
|
|
def regenerate():
|
|
now = current_time()
|
|
|
|
services = {
|
|
'generated': to_unixtimestamp(now),
|
|
'prettytime': now.strftime('%d.%m.%Y %H:%M:%S'),
|
|
'services': [],
|
|
}
|
|
|
|
try:
|
|
hosts = requests.get(CONFIG["url_hosts"], auth=(CONFIG["api_user"], CONFIG["api_password"]), verify=False).json()
|
|
serv = requests.get(CONFIG["url_services"], auth=(CONFIG["api_user"], CONFIG["api_password"]), verify=False).json()
|
|
|
|
if 'results' not in hosts:
|
|
raise KeyError('API call for hosts did not return any results')
|
|
if 'results' not in serv:
|
|
raise KeyError('API call for services did not return any results')
|
|
|
|
for host in hosts['results']:
|
|
if host['attrs']['downtime_depth'] > 0:
|
|
continue
|
|
|
|
if host['attrs']['problem']:
|
|
services['services'].append({
|
|
'host': host['attrs']['display_name'],
|
|
'service': '-- HOST --',
|
|
'state': 2,
|
|
'type': host['attrs']['state_type'],
|
|
'output': host['attrs']['last_check_result']['output'].splitlines(),
|
|
})
|
|
|
|
for svc in serv['results']:
|
|
if host['attrs']['downtime_depth'] > 0:
|
|
continue
|
|
|
|
if svc['attrs']['problem']:
|
|
services['services'].append({
|
|
'host': svc['attrs']['host_name'],
|
|
'service': svc['attrs']['display_name'],
|
|
'state': svc['attrs']['state'],
|
|
'type': svc['attrs']['state_type'],
|
|
'output': svc['attrs']['last_check_result']['output'].splitlines(),
|
|
})
|
|
|
|
services['services'].sort(key=lambda x: str(x['type'])+str(x['state'])+x['host']+x['service'])
|
|
services['services'].reverse()
|
|
except Exception as e:
|
|
services['services'].append({
|
|
'host': 'icinga2beamer',
|
|
'service': 'INTERNAL',
|
|
'state': 2,
|
|
'output': [repr(e)],
|
|
})
|
|
|
|
if len(services['services']) == 0:
|
|
services['services'].append({
|
|
'host': '',
|
|
'service': 'icinga2',
|
|
'state': 0,
|
|
'output': ['Everything is fine. Go get some coffee.'],
|
|
})
|
|
|
|
with file("services.json", "wb") as f:
|
|
f.write(json.dumps(services, ensure_ascii=False).encode("utf8"))
|
|
|
|
def main():
|
|
while 1:
|
|
try:
|
|
regenerate()
|
|
except Exception:
|
|
traceback.print_exc()
|
|
|
|
time.sleep(20)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|