__main__.py

def main():
    import logging
    logging.basicConfig()
    logger = logging.getLogger()
    logger.setLevel(logging.DEBUG)
    logging.getLogger('urllib3').setLevel(logging.CRITICAL)
    logging.getLogger('paramiko').setLevel(logging.CRITICAL)


    import argparse
    import yaml
    parser = argparse.ArgumentParser()
    parser.add_argument('--config', default='./config.yml')
    parser.add_argument('--check', type=bool, default=False)
    args = parser.parse_args()
    

    from . import statuspagewrapper
    from . import influxdbwrapper
    from .influxdbwrapper import Client as influxWrapperClient
    from . import component_tests
    from .status import Status
    
    import datetime
    global ts
    ts = datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S")

    with open(args.config) as f:
        config = yaml.safe_load(f.read())

    influxClient = influxdbwrapper.Client.configure(config['influx'])
    spClient = statuspagewrapper.Client.configure(config['statuspageio'])

    spcomponents = {}
    for c in spClient.components.list():
        spcomponents[c['name']] = c
        logger.debug("found component {}".format(c['name']))
    
    for name,c in config['components'].items():
        msgs = []
        overall = Status.operational
        if name not in spcomponents:
            logger.error("Can't find component {}".format(name))
        if name in spcomponents and spcomponents[name]['status'] == 'under_maintenance':
            continue
        for t in c['tests']:
            testclass = component_tests.getTestClass(classname=t['class'],influxClient=influxClient,**t)
            result = testclass.test()
            logger.debug(result)
            overall = max(overall,result['status'])
            if result['status'] != Status.operational:
                msgs.append(result['msg'])
            if name in spcomponents:
                update_component(spClient,spcomponents[name],overall,"\n".join(msgs),check=args.check,description=c['desc'])
            #update_incident(spclient,spcomponents[name],overall,"\n".join(msgs),open_incidents)

def update_component(spclient,component,status,msg,check,description):
    from .status import Status

    if status is Status.operational:
        statusstr = 'operational'
    if status is Status.partial_outage:
        statusstr = 'partial_outage'
    if status is Status.major_outage:
        statusstr = 'major_outage'
    if status is Status.under_maintenance:
        statusstr = 'under_maintenance'
    if not check:
        if statusstr != component['status']:
            spclient.components.update(component.id, status = statusstr, description = "{}. Updated {}".format(description, ts))
    else:
        if statusstr == component['status']:
            print("no change {} {} {}".format(component['name'],component['status'], statusstr))
        else:
            print("change {} {} {}".format(component['name'],component['status'], statusstr))



if __name__ == '__main__':
    main()