X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=web%2FMonitorWeb%2Fmonitorweb%2Fcontrollers.py;h=f4832baab2f66a5794fe967f0e1e0305e6be706b;hb=f6ae4843ec52f237b8c01c9fdcc9130a34518944;hp=be94316d554b4635047bcc6657fa0527361877b7;hpb=786822939c1d244e30172b997f91dd4aafb2ff24;p=monitor.git

diff --git a/web/MonitorWeb/monitorweb/controllers.py b/web/MonitorWeb/monitorweb/controllers.py
index be94316..f4832ba 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")
@@ -12,35 +15,135 @@ from monitor.database.dborm import zab_metadata as metadata
 from pcucontrol import reboot
 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
 
+def prep_pcu_for_display(pcu):
+		
+	try:
+		pcu.loginbase = site_id2lb[pcu.plc_pcu_stats['site_id']]
+	except:
+		pcu.loginbase = "unknown"
+
+	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 = 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]
+	else:
+		node.kernel = ""
+
+	try:
+		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")
 	def index(self):
@@ -49,6 +152,17 @@ class Root(controllers.RootController):
 		flash("Your application is now running")
 		return dict(now=time.ctime())
 
+	@expose(template="monitorweb.templates.pcuview")
+	def nodeview(self, hostname=None):
+		nodequery=[]
+		if hostname:
+			for node in FindbadNodeRecord.get_latest_by(hostname=hostname):
+				# NOTE: reformat some fields.
+				prep_node_for_display(node)
+				nodequery += [node]
+
+		return self.pcuview(None, hostname) # dict(nodequery=nodequery)
+
 	@expose(template="monitorweb.templates.nodelist")
 	def node(self, filter='BOOT'):
 		import time
@@ -57,34 +171,13 @@ class Root(controllers.RootController):
 		filtercount = {'DOWN' : 0, 'BOOT': 0, 'DEBUG' : 0, 'neverboot' : 0, 'pending' : 0, 'all' : 0}
 		for node in fbquery:
 			# NOTE: reformat some fields.
-			if node.plc_pcuid:
-				pcu = FindbadPCURecord.get_latest_by(plc_pcuid=node.plc_pcuid).first()
-				if pcu:
-					node.pcu_status = pcu.reboot_trial_status
-				else:
-					node.pcu_status = "nodata"
-				node.pcu_short_status = format_pcu_shortstatus(pcu)
-
-			else:
-				node.pcu_status = "nopcu"
-				node.pcu_short_status = "none"
-
-			if node.kernel_version:
-				node.kernel = node.kernel_version.split()[2]
-			else:
-				node.kernel = ""
-
-			try:
-				node.loginbase = site_id2lb[node.plc_node_stats['site_id']]
-			except:
-				node.loginbase = "unknown"
-
+			prep_node_for_display(node)
 
 			# NOTE: count filters
 			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
@@ -97,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...
@@ -105,7 +198,136 @@ 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."""
+		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
+
+		cherry_trail = cherrypy._cputil.get_object_trail()
+		for i in cherry_trail:
+			print "trail: ", i
+
+		print pcuid
+		return self.pcuview(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: " + 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")
+	@exception_handler(nodeaction_handler,"isinstance(tg_exceptions,RuntimeError)")
+	def pcuview(self, loginbase=None, pcuid=None, hostname=None, **data):
+		sitequery=[]
+		pcuquery=[]
+		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]
+			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'):
@@ -122,14 +344,8 @@ class Root(controllers.RootController):
 				filtercount[node.reboot_trial_status] += 1
 			else:
 				filtercount['pending'] += 1
-				
-			try:
-				node.loginbase = site_id2lb[node.plc_pcu_stats['site_id']]
-			except:
-				node.loginbase = "unknown"
 
-			node.ports = format_ports(node)
-			node.status = format_pcu_shortstatus(node)
+			prep_pcu_for_display(node)
 
 			# NOTE: apply filter
 			if filter == "all":
@@ -148,6 +364,18 @@ class Root(controllers.RootController):
 				
 		return dict(query=query, fc=filtercount)
 
+	@expose(template="monitorweb.templates.siteview")
+	def siteview(self, loginbase='pl'):
+		# get site query
+		sitequery = [HistorySiteRecord.by_loginbase(loginbase)]
+		nodequery = []
+		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]
+		return dict(sitequery=sitequery, nodequery=nodequery, fc={})
+
 	@expose(template="monitorweb.templates.sitelist")
 	def site(self, filter='all'):
 		filtercount = {'good' : 0, 'down': 0, 'new' : 0, 'pending' : 0, 'all' : 0}
@@ -180,7 +408,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 ' + \