diff --git a/README.md b/README.md index 92eae43..81b6956 100644 --- a/README.md +++ b/README.md @@ -12,6 +12,7 @@ our new monitoring system that does not suck. - history support -> possible grafana interface - easy to create and extend checks - Requirements: python3 + binaries for certain checks +- Minimal core logic - checks can check "anything" ### Planned features (game) diff --git a/check-cli.py b/check-cli.py index b92b55c..a12756e 100644 --- a/check-cli.py +++ b/check-cli.py @@ -2,18 +2,43 @@ import check +import argparse +import logging -checks = [] - -checks.append(check.DNSCheck("www.ungleich.ch")) -checks.append(check.DNSCheck("www.ungleich.ch", expected_result="2a0a:e5c0:0:2:400:b3ff:fe39:795c")) -checks.append(check.DNSCheck("www.ungleich.ch", rr_type="A", expected_result="185.203.112.17")) - -for num_server in range(1,6): - server="d{}.ungleich.ch".format(num_server) - - checks.append(check.DNSCheck("www.ungleich.ch", server=server)) +logging.basicConfig() +log = logging.getLogger(None) -for c in checks: - print("{} {}".format(c, c.check())) +def do_checks(): + checks = [] + + checks.append(check.DNSCheck("www.ungleich.ch")) + checks.append(check.DNSCheck("www.ungleich.ch", expected_result="2a0a:e5c0:0:2:400:b3ff:fe39:795c")) + checks.append(check.DNSCheck("www.ungleich.ch", rr_type="A", expected_result="185.203.112.17")) + + for num_server in range(1,6): + server="d{}.ungleich.ch".format(num_server) + + checks.append(check.DNSCheck("www.ungleich.ch", server=server)) + checks.append(check.DFCheck(server)) + + + for c in checks: + print("{} {}".format(c, c.check())) + +if __name__ == "__main__": + parser = argparse.ArgumentParser(description='checks') + parser.add_argument('--debug', help='Enable debug logging', action='store_true') + parser.add_argument('--verbose', help='Enable verbose logging', action='store_true') + + args = parser.parse_args() + + if args.debug: + log.setLevel(logging.DEBUG) + print("dbg") + elif args.verbose: + log.setLevel(logging.INFO) + else: + log.setLevel(logging.WARNING) + + do_checks() diff --git a/check.py b/check.py index 0fc01fa..a7f460c 100644 --- a/check.py +++ b/check.py @@ -1,4 +1,8 @@ import subprocess +import logging + +logging.basicConfig() +log = logging.getLogger("checks") class CheckException(Exception): pass @@ -82,3 +86,37 @@ class DNSCheck(BaseCheck): name = "{}@{}".format(name, self.server) return name + + +class DFCheck(BaseCheck): + def __init__(self, name, username="root", path="/"): + self.name = name + self.username = username + self.path = path + + self.command = self.create_command() + log.info("Command = {}".format(self.command)) + + def create_command(self): + base_command ='ssh {}@{}'.format(self.username, self.name).split() + base_command.append("df {}".format(self.path)) + + return base_command + + def check(self): + res = subprocess.run(self.command, + capture_output=True, + encoding="utf-8") + + if not res.returncode == 0: + return (False, "") + + return (True, res.stdout) + + def __repr__(self): + return "".format(self.__str__()) + + def __str__(self): + name = "{}:{}".format(self.name, self.path) + + return name