#!/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 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)