86 lines
1.8 KiB
Python
Executable file
86 lines
1.8 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import logging
|
|
import sys
|
|
import ovh
|
|
|
|
from lib 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()
|