1 files changed, 78 insertions, 0 deletions
diff --git a/plugins/server_utils.py b/plugins/server_utils.py
new file mode 100755
index 0000000..b3767bd
--- /dev/null
+++ b/plugins/server_utils.py
@@ -0,0 +1,78 @@
+#!/usr/bin/python
+# discovery.py Nagios plugin to discover Gluster entities using NRPE
+# Copyright (C) 2014 Red Hat Inc
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
+#
+import datetime
+from pynag import Model
+
+from glusternagios import utils
+import submit_external_command
+
+
+serviceCmdPath = utils.CommandPath("service", "/sbin/service", )
+
+
+def restartNagios():
+    now = datetime.datetime.now()
+    cmdStr = "[%s] RESTART_PROGRAM\n" % (now)
+    submit_external_command.submitExternalCommand(cmdStr)
+
+
+def isNagiosRunning():
+    (rc, out, err) = utils.execCmd([serviceCmdPath.cmd, 'nagios', 'status'])
+    if rc == 0:
+        return True
+    else:
+        return False
+
+
+def getServiceConfig(serviceDesc, hostName):
+    serviceConfig = Model.Service.objects.filter(
+        service_description=serviceDesc, host_name=hostName)
+    if serviceConfig:
+        return serviceConfig[0]
+    else:
+        return None
+
+
+def getServiceConfigByHost(hostName):
+    serviceConfigs = Model.Service.objects.filter(host_name=hostName)
+    return serviceConfigs
+
+
+def getHostConfigByName(hostName):
+    hostConfigs = Model.Host.objects.filter(host_name=hostName)
+    if hostConfigs:
+        return hostConfigs[0]
+    else:
+        return None
+
+
+def getHostConfigsForCluster(clusterName):
+    hostgroup = getHostGroup(clusterName)
+    if hostgroup:
+        return hostgroup.get_effective_hosts()
+    else:
+        return []
+
+
+def getHostGroup(name):
+    hostgroup = Model.Hostgroup.objects.filter(hostgroup_name=name)
+    if hostgroup:
+        return hostgroup[0]
+    else:
+        return None