iamthefij/minitor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Switch to python logging

Browse Source
This commit is contained in:
IamTheFij 2018-02-16 10:03:12 -08:00
parent 045bdee781
commit adcd659316
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
minitor/main.py
View File

@ -1,3 +1,4 @@
import logging
import sys import sys
from argparse import ArgumentParser from argparse import ArgumentParser
from datetime import datetime from datetime import datetime
@ -9,7 +10,10 @@ from time import sleep
import yamlenv import yamlenv
# TODO: validate on start
logging.basicConfig(level=logging.INFO)
logging.getLogger(__name__).addHandler(logging.NullHandler())
def read_yaml(path): def read_yaml(path):
"""Loads config from a YAML file with env interpolation""" """Loads config from a YAML file with env interpolation"""
@ -151,6 +155,9 @@ class Minitor(object):
state = None state = None
check_interval = None check_interval = None
def __init__(self):
self.logger = logging.getLogger(__name__)
def setup(self, config_path): def setup(self, config_path):
"""Load all setup from YAML file at provided path""" """Load all setup from YAML file at provided path"""
config = read_yaml(config_path) config = read_yaml(config_path)
@ -214,12 +221,12 @@ class Minitor(object):
try: try:
result = monitor.check() result = monitor.check()
if result is not None: if result is not None:
print('{}: {}'.format( self.logger.info('%s: %s',
monitor.name, monitor.name,
'SUCCESS' if result else 'FAILURE' 'SUCCESS' if result else 'FAILURE'
)) )
except MinitorAlert as minitor_alert: except MinitorAlert as minitor_alert:
print(minitor_alert) self.logger.warn(minitor_alert)
self.alert_for_monitor(monitor) self.alert_for_monitor(monitor)
sleep(self.check_interval) sleep(self.check_interval)