75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
import time
|
|
from threading import Timer
|
|
from subprocess import run
|
|
from lib.logger import logging
|
|
|
|
log = logging.getLogger('check')
|
|
|
|
class Check:
|
|
def run_check(self):
|
|
self.last_exec_start = time.asctime()
|
|
log.debug(f'start command {self.command} at {self.last_exec_start}')
|
|
try:
|
|
runcheck = run(self.command, capture_output=True)
|
|
|
|
# for nagios checks split text and perfdata
|
|
perfdata = None
|
|
if self.nagios_check:
|
|
parts = runcheck.stdout.decode('utf-8').split('|')
|
|
output_text = parts[0]
|
|
perfdata = parts[1]
|
|
else:
|
|
output_text = runcheck.stdout.decode('utf-8')
|
|
|
|
self.output = {
|
|
"rc": runcheck.returncode,
|
|
"stdout": runcheck.stdout.decode('utf-8'),
|
|
"stderr": runcheck.stderr.decode('utf-8'),
|
|
"output_text": output_text
|
|
}
|
|
if perfdata:
|
|
self.output["perfdata"] = perfdata
|
|
|
|
if runcheck.returncode == 0:
|
|
self.state = "OK"
|
|
elif runcheck.returncode == 1:
|
|
self.state = "WARNING"
|
|
else:
|
|
self.state = "CRITICAL"
|
|
self.last_exec_finish = time.asctime()
|
|
log.debug(f'finished command {self.command} at {self.last_exec_start}')
|
|
|
|
except:
|
|
log.error(f'error trying to execute {self.command}')
|
|
self.state = "CRITICAL"
|
|
|
|
self.timer = Timer(interval=self.interval, function=self.run_check)
|
|
self.timer.daemon = True
|
|
self.timer.start()
|
|
|
|
def __init__(self, configuration, check):
|
|
defaults = configuration.get('defaults')
|
|
self.name = check['name']
|
|
self.command = check['command']
|
|
self.nagios_check = check.get('nagios_check', False)
|
|
self.interval = check.get('interval', defaults.get('interval', 300))
|
|
|
|
# pre define variables for check output
|
|
self.timer = None
|
|
self.state = None
|
|
self.output = {}
|
|
self.last_exec_finish = None
|
|
self.last_exec_start = None
|
|
|
|
self.run_check()
|
|
|
|
def get_values(self):
|
|
values = {
|
|
'name': self.name,
|
|
'command': self.command,
|
|
'last_exec_start': self.last_exec_start,
|
|
'last_exec_finish': self.last_exec_finish,
|
|
'output': self.output,
|
|
'state': self.state
|
|
}
|
|
return values
|