docker_images/uptime/check/check.py

48 lines
1.4 KiB
Python

# -*- coding: utf-8 -*-
"""
Uptime check, at the moment HTTPS only domains
Usage of own healthchecks.io instance and ENV
"""
from os import environ
from requests import get, post, RequestException
hc_root = environ.get('HC_ROOT', '')
# HOSTS are in form hostname|uuid:hostname2|uuid...
hosts = environ.get('HOSTS', '').split(':')
# Domains are in form of example.org,example.net:hostname.org:another.net
domains = environ.get('DOMAINS', '').split(':')
config = {}
for host in zip(hosts, domains):
host_data = host[0].split('|')
config[host_data[0]] = {'url': '%s/ping/%s' % (hc_root, host_data[1]),
'domains': host[1].split(',')}
def check_domain(domain):
try:
r = get('https://%s' % domain)
except Exception as e:
return e
return r.status_code
def ping_with_info(url, fail=False, data=None):
try:
if fail:
url = '%s/fail' % url
post(url, timeout=10, data=data)
except RequestException as e:
print("Ping failed: %s" % e)
for host, settings in config.items():
print('Checking %s : %s' % (', '.join(settings['domains']), host))
fail = False
data = []
for domain in settings['domains']:
status = check_domain(domain)
if not fail and status != 200:
fail = True
data.append('%s : %s' % (domain, status))
ping_with_info(settings['url'], fail=fail, data='\n'.join(data))