Merge pull request #4 from Stunkymonkey/single-service-check

Added Parsing option for testing a single Systemd-service

bin/pynagsystemd.py

@@ -8,6 +8,7 @@
 """
 import io
 import subprocess
+import argparse
 
 import nagiosplugin
 
@@ -39,6 +40,36 @@ def probe(self):
         return [nagiosplugin.Metric('systemd', (True, None), context='systemd')]
 
 
+class ServiceStatus(nagiosplugin.Resource):
+    name = 'SYSTEMD'
+
+    def probe(self):
+        # Execute systemctl is-active and get output
+        global service
+        try:
+            p = subprocess.Popen(['systemctl', 'is-active', service],
+                                 stderr=subprocess.PIPE,
+                                 stdin=subprocess.PIPE,
+                                 stdout=subprocess.PIPE)
+            pres, err = p.communicate()
+        except OSError as e:
+            raise nagiosplugin.CheckError(e)
+
+        if err:
+            raise nagiosplugin.CheckError(err)
+        if pres:
+            result = ""
+            for line in io.StringIO(pres.decode('utf-8')):
+                result = "%s %s" % (result, line.split(' ')[0])
+            result = result.strip()
+            if result == "active":
+                return [nagiosplugin.Metric('systemd', (True, None), context='systemd')]
+            else:
+                return [nagiosplugin.Metric('systemd', (False, service), context='systemd')]
+
+        return [nagiosplugin.Metric('systemd', (False, "No Service given"), context='systemd')]
+
+
 class SystemdContext(nagiosplugin.Context):
     def __init__(self):
         super(SystemdContext, self).__init__('systemd')
@@ -52,10 +83,22 @@ def evaluate(self, metric, resource):
 
 
 def main():
-    check = nagiosplugin.Check(
-        SystemdStatus(),
-        SystemdContext())
-
+    global service
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-s", "--service", type=str, dest="service", help="Name of the Service that is beeing tested")
+
+    args = parser.parse_args()
+    service = str(args.service)
+
+    if service == "None":
+        check = nagiosplugin.Check(
+            SystemdStatus(),
+            SystemdContext())
+    else:
+        check = nagiosplugin.Check(
+            ServiceStatus(),
+            SystemdContext())
     check.main()