ping_servers.py

from __future__ import print_function

import os  # Load the Library Module
import subprocess  # Load the Library Module
import sys  # Load the Library Module

# Script Name		: ping_servers.py
# Author				: Craig Richards
# Created				: 9th May 2012
# Last Modified		: 14th May 2012
# Version				: 1.1
# Modifications		: 1.1 - 14th May 2012 - CR Changed it to use the config directory to store the server files
# Description	 : This script will, depending on the arguments supplied will ping the
# servers associated with that application group.

filename = sys.argv[0]  # Sets a variable for the script name
if (
    "-h" in sys.argv or "--h" in sys.argv or "-help" in sys.argv or "--help" in sys.argv
):  # Help Menu if called
    print(
        """
You need to supply the application group for the servers you want to ping, i.e.
    dms
    swaps

Followed by the site i.e.
    155
    bromley"""
    )
    sys.exit(0)
else:

    if (
        len(sys.argv) < 3
    ):  # If no arguments are passed,display the help/instructions on how to run the script
        sys.exit(
            "\nYou need to supply the app group. Usage : "
            + filename
            + " followed by the application group i.e. \n \t dms or \n \t swaps \n "
            "then the site i.e. \n \t 155 or \n \t bromley"
        )

    appgroup = sys.argv[1]  # Set the variable appgroup as the first argument you supply
    site = sys.argv[2]  # Set the variable site as the second argument you supply

    if os.name == "posix":  # Check the os, if it's linux then
        myping = "ping -c 2 "  # This is the ping command
    elif os.name in ("nt", "dos", "ce"):  # Check the os, if it's windows then
        myping = "ping -n 2 "  # This is the ping command

    if "dms" in sys.argv:  # If the argument passed is dms then
        appgroup = "dms"  # Set the variable appgroup to dms
    elif "swaps" in sys.argv:  # Else if the argment passed is swaps then
        appgroup = "swaps"  # Set the variable appgroup to swaps

    if "155" in sys.argv:  # If the argument passed is 155 then
        site = "155"  # Set the variable site to 155
    elif "bromley" in sys.argv:  # Else if the argument passed is bromley
        site = "bromley"  # Set the variable site to bromley

logdir = os.getenv("logs")  # Set the variable logdir by getting the OS environment logs
logfile = (
    "ping_" + appgroup + "_" + site + ".log"
)  # Set the variable logfile, using the arguments passed to create the logfile
logfilename = os.path.join(
    logdir, logfile
)  # Set the variable logfilename by joining logdir and logfile together
confdir = os.getenv(
    "my_config"
)  # Set the variable confdir from the OS environment variable - 1.2
conffile = appgroup + "_servers_" + site + ".txt"  # Set the variable conffile - 1.2
conffilename = os.path.join(
    confdir, conffile
)  # Set the variable conffilename by joining confdir and conffile together - 1.2

f = open(logfilename, "w")  # Open a logfile to write out the output
for server in open(conffilename):  # Open the config file and read each line - 1.2
    ret = subprocess.call(
        myping + server, shell=True, stdout=f, stderr=subprocess.STDOUT
    )  # Run the ping command for each server in the list.
    if ret == 0:  # Depending on the response
        f.write(
            server.strip() + " is alive" + "\n"
        )  # Write out that you can receive a reponse
    else:
        f.write(
            server.strip() + " did not respond" + "\n"
        )  # Write out you can't reach the box

print("\n\tYou can see the results in the logfile : " + logfilename)
# Show the location of the logfile