1 import turbogears as tg
2 from turbogears import controllers, expose, flash, exception_handler
3 from turbogears import widgets
4 from cherrypy import request, response
6 # from monitorweb import model
8 # log = logging.getLogger("monitorweb.controllers")
10 from monitor.database.info.model import *
11 #from monitor.database.zabbixapi.model import *
12 #from monitor.database.dborm import zab_session as session
13 #from monitor.database.dborm import zab_metadata as metadata
14 from monitor_xmlrpc import MonitorXmlrpcServer
16 from monitor import reboot
17 from monitor import scanapi
20 from monitor.wrapper.plccache import plcdb_hn2lb as site_hn2lb
22 from monitorweb.templates.links import *
25 # make it easier group objects without invoking the elixir auto-write feature.
29 def query_to_dict(query):
30 """ take a url query string and chop it up """
32 query_fields = query.split('&')
33 for f in query_fields:
34 (k,v) = urllib.splitvalue(f)
39 def format_ports(data, pcumodel=None):
44 supported_ports=reboot.model_to_object(pcumodel).supported_ports
46 # ports of a production node
47 supported_ports=[22,80,806]
49 if data and len(data.keys()) > 0 :
50 for port in supported_ports:
52 state = data[str(port)]
56 if state == "filtered":
59 retval.append( (port, state) )
62 retval = [( "Closed/Filtered", "state" )]
64 if filtered_length == len(supported_ports):
65 retval = [( "All Filtered", "state" )]
69 def format_pcu_shortstatus(pcu):
72 if pcu.reboot_trial_status == str(0):
74 elif pcu.reboot_trial_status == "NetDown" or pcu.reboot_trial_status == "Not_Run":
75 status = pcu.reboot_trial_status
81 def prep_pcu_for_display(pcu):
86 agg.loginbase = PlcSite.query.get(pcu.plc_pcu_stats['site_id']).plc_site_stats['login_base']
88 agg.loginbase = "unknown"
90 agg.ports = format_ports(pcu.port_status, pcu.plc_pcu_stats['model'])
91 agg.status = format_pcu_shortstatus(pcu)
93 #print pcu.entry_complete
94 agg.entry_complete_str = pcu.entry_complete
95 #pcu.entry_complete_str += "".join([ f[0] for f in pcu.entry_complete.split() ])
96 if pcu.dns_status == "NOHOSTNAME":
97 agg.dns_short_status = 'NoHost'
98 elif pcu.dns_status == "DNS-OK":
99 agg.dns_short_status = 'Ok'
100 elif pcu.dns_status == "DNS-NOENTRY":
101 agg.dns_short_status = 'NoEntry'
102 elif pcu.dns_status == "NO-DNS-OR-IP":
103 agg.dns_short_status = 'NoHostOrIP'
104 elif pcu.dns_status == "DNS-MISMATCH":
105 agg.dns_short_status = 'Mismatch'
108 class NodeWidget(widgets.Widget):
111 def prep_node_for_display(node):
116 pcu = FindbadPCURecord.get_latest_by(plc_pcuid=node.plc_pcuid)
118 agg.pcu_status = pcu.reboot_trial_status
119 agg.pcu_short_status = format_pcu_shortstatus(pcu)
120 agg.pcu = prep_pcu_for_display(pcu)
122 agg.pcu_short_status = "none"
123 agg.pcu_status = "nodata"
127 agg.pcu_status = "nopcu"
128 agg.pcu_short_status = "none"
132 if node.kernel_version:
133 agg.kernel = node.kernel_version.split()[2]
138 agg.loginbase = PlcSite.query.get(node.plc_node_stats['site_id']).plc_site_stats['login_base']
140 agg.loginbase = "unknown"
143 agg.site = HistorySiteRecord.by_loginbase(agg.loginbase)
145 # TODO: need a cleaner fix for this...
146 agg.site = HistorySiteRecord.by_loginbase("pl")
148 agg.site = HistorySiteRecord.by_loginbase("ple")
150 agg.history = HistoryNodeRecord.by_hostname(node.hostname)
152 agg.ports = format_ports(node.port_status)
155 exists = node.plc_node_stats['last_contact']
157 # TODO: this should not assign to the fb object!
158 node.plc_node_stats = {'last_contact' : None}
163 class Root(controllers.RootController, MonitorXmlrpcServer):
164 @expose(template="monitorweb.templates.welcome")
166 # log.debug("Happy TurboGears Controller Responding For Duty")
167 flash("Welcome To MyOps!")
168 return dict(now=time.ctime())
170 @expose(template="monitorweb.templates.nodelist")
171 def node(self, filter='boot'):
172 print "NODE------------------"
173 print "befor-len: ", len( [ i for i in session] )
174 session.flush(); session.clear()
175 print "after-len: ", len( [ i for i in session] )
176 fbquery = FindbadNodeRecord.get_all_latest()
178 filtercount = {'down' : 0, 'boot': 0, 'debug' : 0, 'diagnose' : 0, 'disabled': 0,
179 'neverboot' : 0, 'pending' : 0, 'all' : 0, None : 0}
181 # NOTE: reformat some fields.
182 agg = prep_node_for_display(node)
190 if agg.history.status in ['down', 'offline']:
191 if node.plc_node_stats and node.plc_node_stats['last_contact'] != None:
192 filtercount['down'] += 1
194 filtercount['neverboot'] += 1
195 elif agg.history.status in ['good', 'online']:
196 filtercount['boot'] += 1
197 elif agg.history.status in ['debug', 'monitordebug']:
198 filtercount['debug'] += 1
200 # TODO: need a better fix. filtercount
201 # doesn't maps to GetBootStates() on
202 # 4.3 so this one fails quite often.
203 if filtercount.has_key(agg.history.status):
204 filtercount[agg.history.status] += 1
208 if filter == "neverboot":
209 if not node.plc_node_stats or node.plc_node_stats['last_contact'] == None:
211 elif filter == "all":
213 elif filter == agg.history.status:
215 elif filter == 'boot':
219 widget = NodeWidget(template='monitorweb.templates.node_template')
220 return dict(now=time.ctime(), query=query, fc=filtercount, nodewidget=widget)
222 def nodeaction_handler(self, tg_exceptions=None):
223 """Handle any kind of error."""
224 print "NODEACTION_HANDLER------------------"
226 if 'pcuid' in request.params:
227 pcuid = request.params['pcuid']
229 refurl = request.headers.get("Referer",link("pcu"))
232 # TODO: do this more intelligently...
233 uri_fields = urllib.splitquery(refurl)
234 if uri_fields[1] is not None:
235 val = query_to_dict(uri_fields[1])
238 elif 'hostname' in val:
239 pcuid = FindbadNodeRecord.get_latest_by(hostname=val['hostname']).plc_pcuid
245 cherry_trail = cherrypy._cputil.get_object_trail()
246 for i in cherry_trail:
250 return self.pcuview(None, pcuid, **dict(exceptions=tg_exceptions))
252 def nodeaction(self, **data):
253 print "NODEACTION------------------"
254 for item in data.keys():
255 print "%s %s" % ( item, data[item] )
257 if 'hostname' in data:
258 hostname = data['hostname']
260 flash("No hostname given in submitted data")
263 if 'submit' in data or 'type' in data:
265 action = data['submit']
267 action = data['type']
269 flash("No submit action given in submitted data")
272 if action == "Reboot":
273 print "REBOOT: %s" % hostname
274 ret = reboot.reboot_str(str(hostname))
276 if ret: raise RuntimeError("Error using PCU: " + str(ret))
277 flash("Reboot appeared to work. Allow at most 5 minutes. Then run ExternalScan to check current status.")
279 elif action == "ExternalScan":
280 scanapi.externalprobe(str(hostname))
281 flash("External Scan Successful!")
282 elif action == "InternalScan":
283 scanapi.internalprobe(str(hostname))
284 flash("Internal Scan Successful!")
287 raise RuntimeError("Unknown action given")
290 # TODO: add form validation
291 @expose(template="monitorweb.templates.pcuview")
292 @exception_handler(nodeaction_handler,"isinstance(tg_exceptions,RuntimeError)")
293 def pcuview(self, loginbase=None, pcuid=None, hostname=None, since=20, **data):
294 print "PCUVIEW------------------"
295 print "befor-len: ", len( [ i for i in session] )
296 session.flush(); session.clear()
297 print "after-len: ", len( [ i for i in session] )
304 try: since = int(since)
310 if 'submit' in data.keys() or 'type' in data.keys():
311 if hostname: data['hostname'] = hostname
312 self.nodeaction(**data)
313 if 'exceptions' in data:
314 exceptions = data['exceptions']
317 print "pcuid: %s" % pcuid
318 pcu = FindbadPCURecord.get_latest_by(plc_pcuid=pcuid)
319 loginbase = PlcSite.query.get(pcu.plc_pcu_stats['site_id']).plc_site_stats['login_base']
322 node = FindbadNodeRecord.get_latest_by(hostname=hostname)
323 loginbase = PlcSite.query.get(node.plc_node_stats['site_id']).plc_site_stats['login_base']
326 actions = ActionRecord.query.filter_by(loginbase=loginbase
327 ).filter(ActionRecord.date_created >= datetime.now() - timedelta(since)
328 ).order_by(ActionRecord.date_created.desc())
329 actions = [ a for a in actions ]
330 sitequery = [HistorySiteRecord.by_loginbase(loginbase)]
332 for node in FindbadNodeRecord.query.filter_by(loginbase=loginbase):
333 # NOTE: reformat some fields.
334 agg = prep_node_for_display(node)
336 if agg.pcu: #.pcu.plc_pcuid: # not None
337 #pcu = FindbadPCURecord.get_latest_by(plc_pcuid=agg.plc_pcuid)
338 #prep_pcu_for_display(pcu)
339 pcus[agg.pcu.pcu.plc_pcuid] = agg.pcu
341 for pcuid_key in pcus:
342 pcuquery += [pcus[pcuid_key]]
344 return dict(sitequery=sitequery, pcuquery=pcuquery, nodequery=nodequery, actions=actions, since=since, exceptions=exceptions)
346 @expose(template="monitorweb.templates.pcuhistory")
347 def pcuhistory(self, pcu_id=None):
350 fbnode = HistoryPCURecord.get_by(plc_pcuid=pcu_id)
351 l = fbnode.versions[-100:]
354 #prep_node_for_display(node)
357 return dict(query=query, pcu_id=pcu_id)
359 @expose(template="monitorweb.templates.nodehistory")
360 def nodehistory(self, hostname=None):
363 #fbnode = FindbadNodeRecord.get_by(hostname=hostname)
364 ## TODO: add links for earlier history if desired.
365 #l = fbnode.versions[-100:]
368 # prep_node_for_display(node)
371 fbnode = HistoryNodeRecord.get_by(hostname=hostname)
372 l = fbnode.versions[-100:]
375 #prep_node_for_display(node)
378 return dict(query=query, hostname=hostname)
380 @expose(template="monitorweb.templates.sitehistory")
381 def sitehistory(self, loginbase=None):
384 fbsite = HistorySiteRecord.get_by(loginbase=loginbase)
385 # TODO: add links for earlier history if desired.
386 l = fbsite.versions[-100:]
390 return dict(query=query, loginbase=loginbase)
393 @expose(template="monitorweb.templates.pculist")
394 def pcu(self, filter='all'):
395 print "PCUVIEW------------------"
396 print "befor-len: ", len( [ i for i in session] )
397 session.flush(); session.clear()
398 print "after-len: ", len( [ i for i in session] )
399 fbquery = FindbadPCURecord.get_all_latest()
401 filtercount = {'ok' : 0, 'NetDown': 0, 'Not_Run' : 0, 'pending' : 0, 'all' : 0}
405 if node.reboot_trial_status == str(0):
406 filtercount['ok'] += 1
407 elif node.reboot_trial_status == 'NetDown' or node.reboot_trial_status == 'Not_Run':
408 filtercount[node.reboot_trial_status] += 1
410 filtercount['pending'] += 1
412 pcuagg = prep_pcu_for_display(node)
417 elif filter == "ok" and node.reboot_trial_status == str(0):
419 elif filter == node.reboot_trial_status:
421 elif filter == "pending":
422 # TODO: look in message logs...
423 if node.reboot_trial_status != str(0) and \
424 node.reboot_trial_status != 'NetDown' and \
425 node.reboot_trial_status != 'Not_Run':
429 return dict(query=query, fc=filtercount)
431 @expose(template="monitorweb.templates.sitelist")
432 def site(self, filter='all'):
433 print "SITE------------------"
434 print "befor-len: ", len( [ i for i in session] )
435 session.flush(); session.clear()
436 print "after-len: ", len( [ i for i in session] )
437 filtercount = {'good' : 0, 'down': 0, 'online':0, 'offline' : 0, 'new' : 0, 'pending' : 0, 'all' : 0}
438 fbquery = HistorySiteRecord.query.all()
442 filtercount['all'] += 1
443 if site.new and site.slices_used == 0 and not site.enabled:
444 filtercount['new'] += 1
445 elif not site.enabled:
446 filtercount['pending'] += 1
447 elif site.status in ['good', 'online']:
448 filtercount['good'] += 1
449 elif site.status in ['down', 'offline']:
450 filtercount['down'] += 1
455 elif filter == 'new' and site.new and site.slices_used == 0 and not site.enabled:
457 elif filter == "pending" and not site.enabled:
459 elif filter == 'good' and site.status in ['good', 'online']:
461 elif filter == 'down' and site.status in ['down', 'offline']:
464 return dict(query=query, fc=filtercount)
466 @expose(template="monitorweb.templates.actionsummary")
467 def actionsummary(self, since=7):
468 from monitor.wrapper.emailTxt import mailtxt
470 types = filter(lambda x: 'notice' in x, dir(mailtxt))
473 try: since = int(since)
477 acts = ActionRecord.query.filter(ActionRecord.action_type==t
478 ).filter(ActionRecord.date_created >= datetime.now() - timedelta(since))
479 results[t] = acts.count()
480 return dict(results=results)
482 @expose(template="monitorweb.templates.actionlist")
483 def actionlist(self, action_type='down_notice', since=7, loginbase=None):
485 try: since = int(since)
489 acts = ActionRecord.query.filter_by(loginbase=loginbase
490 ).filter(ActionRecord.date_created >= datetime.now() - timedelta(since)
491 ).order_by(ActionRecord.date_created.desc())
493 acts = ActionRecord.query.filter(ActionRecord.action_type==action_type
494 ).filter(ActionRecord.date_created >= datetime.now() - timedelta(since)
495 ).order_by(ActionRecord.date_created.desc())
496 query = [ a for a in acts ]
498 return dict(actions=query, action_type=action_type, since=since)