refactor: Use python packages

Signed-off-by: Julien Riou <julien@riou.xyz>

check_ovhcloud/__init__.py

@@ -0,0 +1,37 @@
+import logging
+import os
+
+from nagiosplugin import Context
+from nagiosplugin.state import Critical, Ok, Unknown, Warn
+
+
+class BooleanContext(Context):
+    def __init__(self, name, expected=True, warning=False, critical=False):
+        super().__init__(name)
+        self.expected = expected
+        self.warning = warning
+        self.critical = critical
+
+    def evaluate(self, metric, resource):
+        if not metric.value is self.expected:
+            result_type = Ok
+            if self.critical:
+                result_type = Critical
+            elif self.warning:
+                result_type = Warn
+            return self.result_cls(
+                result_type, f"{metric.name} is not {self.expected}", metric
+            )
+        else:
+            return self.result_cls(Ok, None, metric)
+
+
+def show_version():
+    with open(
+        os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "VERSION"), "r"
+    ) as fd:
+        print(fd.read().strip())
+
+
+def setup_logging(args):
+    logging.basicConfig(format="%(levelname)s: %(message)s", level=args.loglevel)

check_ovhcloud/ping/__init__.py

@@ -0,0 +1,86 @@
+#!/usr/bin/env python3
+
+import argparse
+import logging
+import sys
+import ovh
+
+from check_ovhcloud import setup_logging, show_version, BooleanContext
+
+from nagiosplugin import (
+    Check,
+    Metric,
+    Resource,
+    Summary,
+)
+from nagiosplugin.state import Unknown
+
+logger = logging.getLogger(__name__)
+
+
+def parse_arguments():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-v",
+        "--verbose",
+        dest="loglevel",
+        action="store_const",
+        const=logging.INFO,
+        help="Print more output",
+    )
+    parser.add_argument(
+        "-d",
+        "--debug",
+        dest="loglevel",
+        action="store_const",
+        const=logging.DEBUG,
+        default=logging.WARNING,
+        help="Print even more output",
+    )
+
+    parser.add_argument(
+        "--version",
+        dest="show_version",
+        action="store_true",
+        help="Print version and exit",
+    )
+
+    args = parser.parse_args()
+    return args
+
+
+class Ping(Resource):
+    def __init__(self, client):
+        self.client = client
+
+    def probe(self):
+        try:
+            me = self.client.get(f"/me")
+            return [Metric(me["firstname"], True, context="ping")]
+        except Exception as err:
+            return [Metric(str(err), False, context="ping")]
+
+
+def main():
+    args = parse_arguments()
+    setup_logging(args)
+
+    if args.show_version:
+        show_version()
+        return
+
+    try:
+        check = Check(
+            Ping(client=ovh.Client()),
+            BooleanContext("ping", expected=True, critical=True),
+            Summary(),
+        )
+        check.main()
+    except Exception as err:
+        print(f"Failed to execute check: {str(err)}")
+        logger.debug(err, exc_info=True)
+        sys.exit(Unknown.code)
+
+
+if __name__ == "__main__":
+    main()

check_ovhcloud/voip/__init__.py

@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+
+import argparse
+import logging
+import sys
+import ovh
+from datetime import datetime, timezone
+
+from check_ovhcloud import setup_logging, show_version
+
+from nagiosplugin import (
+    Check,
+    Metric,
+    Resource,
+    ScalarContext,
+    Summary,
+)
+from nagiosplugin.state import Unknown
+
+logger = logging.getLogger(__name__)
+
+
+def parse_arguments():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-v",
+        "--verbose",
+        dest="loglevel",
+        action="store_const",
+        const=logging.INFO,
+        help="Print more output",
+    )
+    parser.add_argument(
+        "-d",
+        "--debug",
+        dest="loglevel",
+        action="store_const",
+        const=logging.DEBUG,
+        default=logging.WARNING,
+        help="Print even more output",
+    )
+
+    parser.add_argument(
+        "--version",
+        dest="show_version",
+        action="store_true",
+        help="Print version and exit",
+    )
+
+    parser.add_argument(
+        "--registration-warning",
+        "-w",
+        dest="registration_warning",
+        type=int,
+        default=7200,
+        help="Raise warning if last line registration time is higher than this threshold (in seconds)",
+    )
+
+    parser.add_argument(
+        "--registration-critical",
+        "-c",
+        dest="registration_critical",
+        type=int,
+        default=86400,
+        help="Raise critical if last line registration time is higher than this threshold (in seconds)",
+    )
+
+    args = parser.parse_args()
+    return args
+
+
+class Voip(Resource):
+    def __init__(self, client):
+        self.client = client
+
+    def probe(self):
+        metrics = []
+        billing_accounts = self.client.get(f"/telephony")
+        for billing_account in billing_accounts:
+            service_names = self.client.get(f"/telephony/{billing_account}/line")
+            for service_name in service_names:
+                last_registration = self.client.get(
+                    f"/telephony/{billing_account}/line/{service_name}/lastRegistrations"
+                )[0]
+                last_registration_delta = int(
+                    (
+                        datetime.now(tz=timezone.utc)
+                        - datetime.fromisoformat(last_registration["datetime"])
+                    ).total_seconds()
+                )
+                metrics.append(
+                    Metric(
+                        f"{service_name} last registration",
+                        last_registration_delta,
+                        context="last_registration",
+                        uom="s",
+                    )
+                )
+        return metrics
+
+
+class VoipSummary(Summary):
+    def problem(self, results):
+        return ", ".join(
+            [
+                f"{result.metric.name} {result.state}: {result.hint}"
+                for result in results
+                if str(result.state) != "ok"
+            ]
+        )
+
+
+def main():
+    args = parse_arguments()
+    setup_logging(args)
+
+    if args.show_version:
+        show_version()
+        return
+
+    try:
+        check = Check(
+            Voip(client=ovh.Client()),
+            ScalarContext(
+                "last_registration",
+                warning=args.registration_warning,
+                critical=args.registration_critical,
+            ),
+            VoipSummary(),
+        )
+        check.main()
+    except Exception as err:
+        print(f"Failed to execute check: {str(err)}")
+        logger.debug(err, exc_info=True)
+        sys.exit(Unknown.code)
+
+
+if __name__ == "__main__":
+    main()