lemmon_tools.py

Go to the documentation of this file.

# SERVERS = ["lem.hep.caltech.edu", "ec2-52-10-107-248.us-west-2.compute.amazonaws.com", "52.10.107.248"]
SERVERS = ["lem.hep.caltech.edu"]
PORTS   = range(1024, 1032)
########################### 
# machine information
def runMachineCheck():
    status = {}
    import socket, time
    machine = socket.gethostname()
    status["machine"] = machine
    this_time = time.localtime()
    time_str = time.strftime("%Y-%m-%d %H:%M:%S", this_time)
    status["time"]      = time_str
    status["year"]      = this_time.tm_year
    status["month"]     = this_time.tm_mon
    status["day"]       = this_time.tm_mday
    status["hour"]      = this_time.tm_hour
    status["minute"]    = this_time.tm_min
    status["second"]    = this_time.tm_sec
    return status
########################### 
# run LEM status check
def runStatusCheck(MAX_ATTEMPTS=3, verbose=False):
    status = runMachineCheck()
    import urllib2
    results = {}
    for server in SERVERS:
        for port in PORTS:
            url = "http://%s:%i/status"%(server,port)
            results[url] = {"status"        :False,
                            "status_text"   :False,
                            "exception"     :False,
                            "exception_text":False}
            if verbose: print "lem  : determining status of %s"%url
            attempts = 0
            while (attempts < MAX_ATTEMPTS):
                try:
                    response = urllib2.urlopen(url)
                    content = response.read()
                    if verbose: print "lem  : ---\n%s\nlem  : ---\n"%content
                    results[url]["status"]         = "good"
                    results[url]["status_text"]    = content
                    results[url]["exception"]      = False
                    results[url]["exception_text"] = False
                    break
                except urllib2.URLError as exception:
                    attempts += 1
                    if verbose: 
                        print "lem  :     attempt[%i/%i] failed"%(attempts,MAX_ATTEMPTS)
                        print "lem  :     exception type: ",type(exception)
                        print "lem  : ---\n%s\nlem  : ---\n"%str(exception.reason)
                    results[url]["status"]         = "bad"
                    results[url]["status_text"]    = False
                    results[url]["exception"]      = str(type(exception))
                    results[url]["exception_text"] = str(exception.reason)
    status["results"] = results
    return status
########################### 
# append a status to a file
def appendToFile(status, prefix="/nusoft/app/web/htdoc/nova/lem/"):
    import os, json
    output_name = "%s/lemmon.json"%(prefix)
    if not os.path.exists(output_name):
        print "tools: output file %s doesn't exist, making a fresh file"%output_name
        statuses = {}
        statuses["servers"] = SERVERS
        statuses["ports"]   = PORTS
        statuses["machine"] = []
        statuses["time"]    = []
        statuses["year"]    = []
        statuses["month"]   = []
        statuses["day"]     = []
        statuses["hour"]    = []
        statuses["minute"]  = []
        statuses["second"]  = []
        statuses["results"] = []
    else:
        print "tools: loading output file"
        statuses = json.load(open(output_name,"r"))
    print "tools: Appending to list of %i statuses"%len(statuses["time"])
    statuses = appendStatus(statuses, status)
    with open(output_name, "w") as outfile:
        json.dump(statuses, outfile)
    print "tools: written to %s"%output_name
########################### 
# append status to statuses
def appendStatus(history, new):
    history["machine"].append(          new["machine"])
    history["time"].append(             new["time"])
    history["year"].append(             new["year"])
    history["month"].append(            new["month"])
    history["day"].append(              new["day"])
    history["hour"].append(             new["hour"])
    history["minute"].append(           new["minute"])
    history["second"].append(           new["second"])
    history["results"].append(          new["results"])
    return history