add checkrt to indicate when a site has new or open tickets

[monitor.git] / nagios / plugins / checkrt.py

diff --git a/nagios/plugins/checkrt.py b/nagios/plugins/checkrt.py
new file mode 100755 (executable)
index 0000000..befb1e3
--- /dev/null
+++ b/nagios/plugins/checkrt.py
@@ -0,0 +1,72 @@
+#!/usr/bin/python
+
+import os
+import time
+import sys
+import auth
+
+def argv_to_dict(argv):
+    """
+        NOTE: very bare-bones, no error checking, will fail easily.
+    """
+    d = {}
+    prev=None
+    for a in argv:
+        if "--" == a[0:2]:
+            prev = a[2:]
+        elif "-" == a[0:1]:
+            prev = a[1:]
+        else:
+            d[prev] = a
+    return d
+
+def main():
+    d = argv_to_dict(sys.argv[1:])
+
+    if 'pattern' in d or 'p' in d:
+        try:
+            pattern = d['pattern']
+        except:
+            pattern = d['p']
+    else:
+        print "UNKNOWN: Argument error"
+        sys.exit(3)
+
+
+    # TODO: check that RT is configured correctly
+    os.environ["RTSERVER"] = auth.RTSERVER
+    os.environ["RTUSER"] = auth.RTUSER
+    os.environ["RTPASSWD"] = auth.RTPASSWD
+    os.environ["RTDEBUG"] = auth.RTDEBUG
+
+    # TODO: may need to add a timeout
+    # NOTE: RT3.8
+    query = "Subject like '%%%s%%' and Queue='Monitor' and ( Status='new' or Status='open' )" % pattern
+    cmd = """rt ls -s -t ticket "%s" 2>&1 """ % query
+    cmd = cmd + """| grep -vi "no match" | wc -l """
+
+    out = os.popen(cmd, 'r')
+    open_tickets = out.read()
+
+    try:
+        open_tickets_i = int(open_tickets)
+    except:
+        print "UNKNOWN: failed to convert %s to open ticket count" % open_tickets
+        sys.exit(3)
+
+    if open_tickets_i == 0:
+        print "OK: no open tickets for site"
+        sys.exit(0)
+    elif open_tickets_i != 0:
+        print "WARNING: %s open tickets" % open_tickets_i
+        sys.exit(1)
+    else:
+        print "FAKE-CRITICAL: RT check failed"
+        sys.exit(2)
+
+
+if __name__ == '__main__':
+    f = open("/tmp/checkpcu", 'a')
+    f.write("checkpcu %s %s\n" % (time.time(), " ".join(sys.argv[1:])))
+    f.close()
+    main()