Kapisi/bin/generate-monitoring.py

#!/usr/bin/env python3
# File: generate-pihole-dns-dhcp.py
#
# Description: This file generates the DNS and DHCP files for pihole.
#
# Package: AniNIX/Kapisi
# Copyright: WTFPL
#
# Author: DarkFeather <darkfeather@aninix.net>

import os
import subprocess
import sys
import re
import yaml
from kapisi_lib import *

rolepath='../roles/Sharingan/files'
monfilepath=rolepath+"/monit/checks/availability"

def WriteMonitoringEntry(entryset):
    ### Create the ping-based monitoring entry
    # param entryset: Entries matched from the inventory
    global monfile

    with open(monfilepath,'a') as monfile:

        # Write host entries
        for host in entryset:
            try:
                monfile.write('check program ' + host + '_ping_mon with path "/usr/lib/monitoring-plugins/check_ping -H ' + entryset[host][2] + ' -w 100,50% -c 1000,100% -p 3 -t 60 -4"\n')
                monfile.write('    if status != 0 for 3 times within 5 cycles then exec "/etc/monit.d/scripts/critical ' + entryset[host][2] + ' is not online."\n\n')
            except:
                print(host + ' is not complete for monitoring.')

def WriteSSHMonitoringEntry(entryset):
    ### Create the ping-based monitoring entry
    # param entryset: Entries matched from the inventory
    global monfile

    with open(monfilepath,'a') as monfile:

        # Write host entries
        for host in entryset:
            try:
                monfile.write('check program ' + host + '_ssh_mon with path "/usr/lib/monitoring-plugins/check_ssh -H ' + entryset[host][2] + '"\n')
                monfile.write('    if status != 0 for 3 times within 5 cycles then exec "/etc/monit.d/scripts/critical ' + host + ' is not responding to SSH."\n\n')
            except:
                print(host + ' is not complete for monitoring.')

def GenerateFiles(file):
    ### Open the file and parse it
    # param file: the file to work on
    global monfilepath

    if not os.path.isdir(rolepath):
        os.mkdir(rolepath)

    # Parse the yaml
    entryset = TrackIPEntries(file,searchstring='all.children.managed.**.ip')

    if os.path.isfile(monfilepath): os.remove(monfilepath)

    WriteSSHMonitoringEntry(entryset)
    WriteMonitoringEntry(entryset)

if __name__ == '__main__':
    if len(sys.argv) != 2:
        print("You need to supply an inventory file.")
        sys.exit(1)
    GenerateFiles(sys.argv[1])
    sys.exit(0)