X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=web%2FMonitorWeb%2Fmonitorweb%2Fcontrollers.py;h=bb0580b068302ae8d947a92172004ab4edb5fa1f;hb=4df3989d4aee44cb6e36169ac69ba097436f7b15;hp=9202cacb3e4e6157e8ff7e547ffcf675aeadb2b6;hpb=d389a66c8cc1819095018e741352aff301cfcb4c;p=monitor.git diff --git a/web/MonitorWeb/monitorweb/controllers.py b/web/MonitorWeb/monitorweb/controllers.py index 9202cac..bb0580b 100644 --- a/web/MonitorWeb/monitorweb/controllers.py +++ b/web/MonitorWeb/monitorweb/controllers.py @@ -1,5 +1,8 @@ import turbogears as tg -from turbogears import controllers, expose, flash +from turbogears import controllers, expose, flash, exception_handler +from turbogears import widgets +from cherrypy import request, response +import cherrypy # from monitorweb import model # import logging # log = logging.getLogger("monitorweb.controllers") @@ -14,31 +17,60 @@ from monitor.wrapper.plccache import plcdb_id2lb as site_id2lb from monitor.wrapper.plccache import plcdb_hn2lb as site_hn2lb from monitor.wrapper.plccache import plcdb_lb2hn as site_lb2hn -def format_ports(pcu): +from monitorweb.templates.links import * + +from monitor import scanapi + + +def query_to_dict(query): + """ take a url query string and chop it up """ + val = {} + query_fields = query.split('&') + for f in query_fields: + (k,v) = urllib.splitvalue(f) + val[k] = v + + return val + +def format_ports(data, pcumodel=None): retval = [] - if pcu.port_status and len(pcu.port_status.keys()) > 0 : - obj = reboot.model_to_object(pcu.plc_pcu_stats['model']) - for port in obj.supported_ports: + filtered_length=0 + + if pcumodel: + supported_ports=reboot.model_to_object(pcumodel).supported_ports + else: + # ports of a production node + supported_ports=[22,80,806] + + if data and len(data.keys()) > 0 : + for port in supported_ports: try: - state = pcu.port_status[str(port)] + state = data[str(port)] except: state = "unknown" + + if state == "filtered": + filtered_length += 1 retval.append( (port, state) ) if retval == []: retval = [( "Closed/Filtered", "state" )] + if filtered_length == len(supported_ports): + retval = [( "All Filtered", "state" )] + return retval def format_pcu_shortstatus(pcu): status = "error" - if pcu.reboot_trial_status == str(0): - status = "ok" - elif pcu.reboot_trial_status == "NetDown" or pcu.reboot_trial_status == "Not_Run": - status = pcu.reboot_trial_status - else: - status = "error" + if pcu: + if pcu.reboot_trial_status == str(0): + status = "Ok" + elif pcu.reboot_trial_status == "NetDown" or pcu.reboot_trial_status == "Not_Run": + status = pcu.reboot_trial_status + else: + status = "error" return status @@ -49,21 +81,44 @@ def prep_pcu_for_display(pcu): except: pcu.loginbase = "unknown" - pcu.ports = format_ports(pcu) + pcu.ports = format_ports(pcu.port_status, pcu.plc_pcu_stats['model']) pcu.status = format_pcu_shortstatus(pcu) + #print pcu.entry_complete + pcu.entry_complete_str = pcu.entry_complete + #pcu.entry_complete_str += "".join([ f[0] for f in pcu.entry_complete.split() ]) + if pcu.dns_status == "NOHOSTNAME": + pcu.dns_short_status = 'NoHost' + elif pcu.dns_status == "DNS-OK": + pcu.dns_short_status = 'Ok' + elif pcu.dns_status == "DNS-NOENTRY": + pcu.dns_short_status = 'NoEntry' + elif pcu.dns_status == "NO-DNS-OR-IP": + pcu.dns_short_status = 'NoHostOrIP' + elif pcu.dns_status == "DNS-MISMATCH": + pcu.dns_short_status = 'Mismatch' + +class NodeWidget(widgets.Widget): + pass + def prep_node_for_display(node): if node.plc_pcuid: pcu = FindbadPCURecord.get_latest_by(plc_pcuid=node.plc_pcuid).first() if pcu: node.pcu_status = pcu.reboot_trial_status + node.pcu_short_status = format_pcu_shortstatus(pcu) + node.pcu = pcu + prep_pcu_for_display(node.pcu) else: + node.pcu_short_status = "none" node.pcu_status = "nodata" - node.pcu_short_status = format_pcu_shortstatus(pcu) + node.pcu = None else: node.pcu_status = "nopcu" node.pcu_short_status = "none" + node.pcu = None + if node.kernel_version: node.kernel = node.kernel_version.split()[2] @@ -74,7 +129,20 @@ def prep_node_for_display(node): node.loginbase = site_id2lb[node.plc_node_stats['site_id']] except: node.loginbase = "unknown" - + + if node.loginbase: + node.site = HistorySiteRecord.by_loginbase(node.loginbase) + + node.history = HistoryNodeRecord.by_hostname(node.hostname) + + node.ports = format_ports(node.port_status) + + try: + exists = node.plc_node_stats['last_contact'] + except: + node.plc_node_stats = {'last_contact' : None} + + class Root(controllers.RootController): @expose(template="monitorweb.templates.welcome") @@ -84,7 +152,7 @@ class Root(controllers.RootController): flash("Your application is now running") return dict(now=time.ctime()) - @expose(template="monitorweb.templates.nodeview") + @expose(template="monitorweb.templates.pcuview") def nodeview(self, hostname=None): nodequery=[] if hostname: @@ -93,7 +161,7 @@ class Root(controllers.RootController): prep_node_for_display(node) nodequery += [node] - return dict(nodequery=nodequery) + return self.pcuview(None, hostname) # dict(nodequery=nodequery) @expose(template="monitorweb.templates.nodelist") def node(self, filter='BOOT'): @@ -109,7 +177,7 @@ class Root(controllers.RootController): if node.observed_status != 'DOWN': filtercount[node.observed_status] += 1 else: - if node.plc_node_stats['last_contact'] != None: + if node.plc_node_stats and node.plc_node_stats['last_contact'] != None: filtercount[node.observed_status] += 1 else: filtercount['neverboot'] += 1 @@ -122,7 +190,7 @@ class Root(controllers.RootController): else: query.append(node) elif filter == "neverboot": - if node.plc_node_stats['last_contact'] == None: + if not node.plc_node_stats or node.plc_node_stats['last_contact'] == None: query.append(node) elif filter == "pending": # TODO: look in message logs... @@ -130,18 +198,142 @@ class Root(controllers.RootController): elif filter == "all": query.append(node) - return dict(now=time.ctime(), query=query, fc=filtercount) + widget = NodeWidget(template='monitorweb.templates.node_template') + return dict(now=time.ctime(), query=query, fc=filtercount, nodewidget=widget) + + def nodeaction_handler(self, tg_exceptions=None): + """Handle any kind of error.""" + + if 'pcuid' in request.params: + pcuid = request.params['pcuid'] + else: + refurl = request.headers.get("Referer",link("pcu")) + print refurl + + # TODO: do this more intelligently... + uri_fields = urllib.splitquery(refurl) + if uri_fields[1] is not None: + val = query_to_dict(uri_fields[1]) + if 'pcuid' in val: + pcuid = val['pcuid'] + elif 'hostname' in val: + pcuid = FindbadNodeRecord.get_latest_by(hostname=val['hostname']).first().plc_pcuid + else: + pcuid=None + else: + pcuid=None + cherry_trail = cherrypy._cputil.get_object_trail() + for i in cherry_trail: + print "trail: ", i + + print pcuid + return self.pcuview(None, pcuid, **dict(exceptions=tg_exceptions)) + + def nodeaction(self, **data): + for item in data.keys(): + print "%s %s" % ( item, data[item] ) + + if 'hostname' in data: + hostname = data['hostname'] + else: + flash("No hostname given in submitted data") + return + + if 'submit' in data or 'type' in data: + try: + action = data['submit'] + except: + action = data['type'] + else: + flash("No submit action given in submitted data") + return + + if action == "Reboot": + print "REBOOT: %s" % hostname + ret = reboot.reboot_str(str(hostname)) + print ret + if ret: raise RuntimeError("Error using PCU: " + str(ret)) + flash("Reboot appeared to work. All at most 5 minutes. Run ExternalScan to check current status.") + + elif action == "ExternalScan": + scanapi.externalprobe(str(hostname)) + flash("External Scan Successful!") + elif action == "InternalScan": + scanapi.internalprobe(str(hostname)) + flash("Internal Scan Successful!") + else: + # unknown action + raise RuntimeError("Unknown action given") + return + + # TODO: add form validation @expose(template="monitorweb.templates.pcuview") - def pcuview(self, pcuid=None): + @exception_handler(nodeaction_handler,"isinstance(tg_exceptions,RuntimeError)") + def pcuview(self, loginbase=None, pcuid=None, hostname=None, **data): + sitequery=[] pcuquery=[] - if pcuid: + nodequery=[] + exceptions = None + + for key in data: + print key, data[key] + + if 'submit' in data.keys() or 'type' in data.keys(): + if hostname: data['hostname'] = hostname + self.nodeaction(**data) + if 'exceptions' in data: + exceptions = data['exceptions'] + + if loginbase: + sitequery = [HistorySiteRecord.by_loginbase(loginbase)] + pcus = {} + for plcnode in site_lb2hn[loginbase]: + for node in FindbadNodeRecord.get_latest_by(hostname=plcnode['hostname']): + # NOTE: reformat some fields. + prep_node_for_display(node) + nodequery += [node] + if node.plc_pcuid: # not None + pcu = FindbadPCURecord.get_latest_by(plc_pcuid=node.plc_pcuid).first() + prep_pcu_for_display(pcu) + pcus[node.plc_pcuid] = pcu + + for pcuid_key in pcus: + pcuquery += [pcus[pcuid_key]] + + if pcuid and hostname is None: + print "pcuid: %s" % pcuid for pcu in FindbadPCURecord.get_latest_by(plc_pcuid=pcuid): # NOTE: count filter prep_pcu_for_display(pcu) pcuquery += [pcu] - - return dict(pcuquery=pcuquery) + if 'site_id' in pcu.plc_pcu_stats: + sitequery = [HistorySiteRecord.by_loginbase(pcu.loginbase)] + + if 'nodenames' in pcu.plc_pcu_stats: + for nodename in pcu.plc_pcu_stats['nodenames']: + print "query for %s" % nodename + q = FindbadNodeRecord.get_latest_by(hostname=nodename) + node = q.first() + print "%s" % node.port_status + print "%s" % node.to_dict() + print "%s" % len(q.all()) + if node: + prep_node_for_display(node) + nodequery += [node] + + if hostname and pcuid is None: + for node in FindbadNodeRecord.get_latest_by(hostname=hostname): + # NOTE: reformat some fields. + prep_node_for_display(node) + sitequery = [node.site] + nodequery += [node] + if node.plc_pcuid: # not None + pcu = FindbadPCURecord.get_latest_by(plc_pcuid=node.plc_pcuid).first() + prep_pcu_for_display(pcu) + pcuquery += [pcu] + + return dict(sitequery=sitequery, pcuquery=pcuquery, nodequery=nodequery, exceptions=exceptions) @expose(template="monitorweb.templates.pculist") def pcu(self, filter='all'): @@ -222,7 +414,7 @@ class Root(controllers.RootController): session.bind = metadata.bind filtercount = {'active' : 0, 'acknowledged': 0, 'all' : 0} # With Acknowledgement - sql_ack = 'SELECT DISTINCT h.host,t.description,t.priority,t.lastchange,a.message '+ \ + sql_ack = 'SELECT DISTINCT h.host,t.description,t.priority,t.lastchange,a.message,e.eventid '+ \ ' FROM triggers t,hosts h,items i,functions f, hosts_groups hg,escalations e,acknowledges a ' + \ ' WHERE f.itemid=i.itemid ' + \ ' AND h.hostid=i.hostid ' + \