web/MonitorWeb/monitorweb/controllers.py

   1 import turbogears as tg
   2 from turbogears import controllers, expose, flash
   3 # from monitorweb import model
   4 # import logging
   5 # log = logging.getLogger("monitorweb.controllers")
   6 import re
   7 from monitor.database.info.model import *
   8 from monitor.database.zabbixapi.model import *
   9 from monitor.database.dborm import zab_session as session
  10 from monitor.database.dborm import zab_metadata as metadata
  11
  12 from pcucontrol import reboot
  13 from monitor.wrapper.plccache import plcdb_id2lb as site_id2lb
  14 from monitor.wrapper.plccache import plcdb_hn2lb as site_hn2lb
  15 from monitor.wrapper.plccache import plcdb_lb2hn as site_lb2hn
  16
  17 def format_ports(pcu):
  18         retval = []
  19         if pcu.port_status and len(pcu.port_status.keys()) > 0 :
  20                 obj = reboot.model_to_object(pcu.plc_pcu_stats['model'])
  21                 for port in obj.supported_ports:
  22                         try:
  23                                 state = pcu.port_status[str(port)]
  24                         except:
  25                                 state = "unknown"
  26
  27                         retval.append( (port, state) )
  28
  29         if retval == []:
  30                 retval = [( "Closed/Filtered", "state" )]
  31
  32         return retval
  33
  34 def format_pcu_shortstatus(pcu):
  35         status = "error"
  36         if pcu:
  37                 if pcu.reboot_trial_status == str(0):
  38                         status = "ok"
  39                 elif pcu.reboot_trial_status == "NetDown" or pcu.reboot_trial_status == "Not_Run":
  40                         status = pcu.reboot_trial_status
  41                 else:
  42                         status = "error"
  43
  44         return status
  45
  46 def prep_pcu_for_display(pcu):
  47
  48         try:
  49                 pcu.loginbase = site_id2lb[pcu.plc_pcu_stats['site_id']]
  50         except:
  51                 pcu.loginbase = "unknown"
  52
  53         pcu.ports = format_ports(pcu)
  54         pcu.status = format_pcu_shortstatus(pcu)
  55
  56 def prep_node_for_display(node):
  57         if node.plc_pcuid:
  58                 pcu = FindbadPCURecord.get_latest_by(plc_pcuid=node.plc_pcuid).first()
  59                 if pcu:
  60                         node.pcu_status = pcu.reboot_trial_status
  61                 else:
  62                         node.pcu_status = "nodata"
  63                 node.pcu_short_status = format_pcu_shortstatus(pcu)
  64
  65         else:
  66                 node.pcu_status = "nopcu"
  67                 node.pcu_short_status = "none"
  68
  69         if node.kernel_version:
  70                 node.kernel = node.kernel_version.split()[2]
  71         else:
  72                 node.kernel = ""
  73
  74         try:
  75                 node.loginbase = site_id2lb[node.plc_node_stats['site_id']]
  76         except:
  77                 node.loginbase = "unknown"
  78
  79
  80 class Root(controllers.RootController):
  81         @expose(template="monitorweb.templates.welcome")
  82         def index(self):
  83                 import time
  84                 # log.debug("Happy TurboGears Controller Responding For Duty")
  85                 flash("Your application is now running")
  86                 return dict(now=time.ctime())
  87
  88         @expose(template="monitorweb.templates.nodeview")
  89         def nodeview(self, hostname=None):
  90                 nodequery=[]
  91                 if hostname:
  92                         for node in FindbadNodeRecord.get_latest_by(hostname=hostname):
  93                                 # NOTE: reformat some fields.
  94                                 prep_node_for_display(node)
  95                                 nodequery += [node]
  96
  97                 return dict(nodequery=nodequery)
  98
  99         @expose(template="monitorweb.templates.nodelist")
 100         def node(self, filter='BOOT'):
 101                 import time
 102                 fbquery = FindbadNodeRecord.get_all_latest()
 103                 query = []
 104                 filtercount = {'DOWN' : 0, 'BOOT': 0, 'DEBUG' : 0, 'neverboot' : 0, 'pending' : 0, 'all' : 0}
 105                 for node in fbquery:
 106                         # NOTE: reformat some fields.
 107                         prep_node_for_display(node)
 108
 109                         # NOTE: count filters
 110                         if node.observed_status != 'DOWN':
 111                                 filtercount[node.observed_status] += 1
 112                         else:
 113                                 if node.plc_node_stats['last_contact'] != None:
 114                                         filtercount[node.observed_status] += 1
 115                                 else:
 116                                         filtercount['neverboot'] += 1
 117
 118                         # NOTE: apply filter
 119                         if filter == node.observed_status:
 120                                 if filter == "DOWN":
 121                                         if node.plc_node_stats['last_contact'] != None:
 122                                                 query.append(node)
 123                                 else:
 124                                         query.append(node)
 125                         elif filter == "neverboot":
 126                                 if node.plc_node_stats['last_contact'] == None:
 127                                         query.append(node)
 128                         elif filter == "pending":
 129                                 # TODO: look in message logs...
 130                                 pass
 131                         elif filter == "all":
 132                                 query.append(node)
 133
 134                 return dict(now=time.ctime(), query=query, fc=filtercount)
 135
 136         @expose(template="monitorweb.templates.pcuview")
 137         def pcuview(self, pcuid=None):
 138                 pcuquery=[]
 139                 if pcuid:
 140                         for pcu in FindbadPCURecord.get_latest_by(plc_pcuid=pcuid):
 141                                 # NOTE: count filter
 142                                 prep_pcu_for_display(pcu)
 143                                 pcuquery += [pcu]
 144
 145                 return dict(pcuquery=pcuquery)
 146
 147         @expose(template="monitorweb.templates.pculist")
 148         def pcu(self, filter='all'):
 149                 import time
 150                 fbquery = FindbadPCURecord.get_all_latest()
 151                 query = []
 152                 filtercount = {'ok' : 0, 'NetDown': 0, 'Not_Run' : 0, 'pending' : 0, 'all' : 0}
 153                 for node in fbquery:
 154
 155                         # NOTE: count filter
 156                         if node.reboot_trial_status == str(0):
 157                                 filtercount['ok'] += 1
 158                         elif node.reboot_trial_status == 'NetDown' or node.reboot_trial_status == 'Not_Run':
 159                                 filtercount[node.reboot_trial_status] += 1
 160                         else:
 161                                 filtercount['pending'] += 1
 162
 163                         prep_pcu_for_display(node)
 164
 165                         # NOTE: apply filter
 166                         if filter == "all":
 167                                 query.append(node)
 168                         elif filter == "ok" and node.reboot_trial_status == str(0):
 169                                 query.append(node)
 170                         elif filter == node.reboot_trial_status:
 171                                 query.append(node)
 172                         elif filter == "pending":
 173                                 # TODO: look in message logs...
 174                                 if node.reboot_trial_status != str(0) and \
 175                                         node.reboot_trial_status != 'NetDown' and \
 176                                         node.reboot_trial_status != 'Not_Run':
 177
 178                                         query.append(node)
 179
 180                 return dict(query=query, fc=filtercount)
 181
 182         @expose(template="monitorweb.templates.siteview")
 183         def siteview(self, loginbase='pl'):
 184                 # get site query
 185                 sitequery = [HistorySiteRecord.by_loginbase(loginbase)]
 186                 nodequery = []
 187                 for plcnode in site_lb2hn[loginbase]:
 188                         for node in FindbadNodeRecord.get_latest_by(hostname=plcnode['hostname']):
 189                                 # NOTE: reformat some fields.
 190                                 prep_node_for_display(node)
 191                                 nodequery += [node]
 192                 return dict(sitequery=sitequery, nodequery=nodequery, fc={})
 193
 194         @expose(template="monitorweb.templates.sitelist")
 195         def site(self, filter='all'):
 196                 filtercount = {'good' : 0, 'down': 0, 'new' : 0, 'pending' : 0, 'all' : 0}
 197                 fbquery = HistorySiteRecord.query.all()
 198                 query = []
 199                 for site in fbquery:
 200                         # count filter
 201                         filtercount['all'] += 1
 202                         if site.new and site.slices_used == 0 and not site.enabled:
 203                                 filtercount['new'] += 1
 204                         elif not site.enabled:
 205                                 filtercount['pending'] += 1
 206                         else:
 207                                 filtercount[site.status] += 1
 208
 209                         # apply filter
 210                         if filter == "all":
 211                                 query.append(site)
 212                         elif filter == 'new' and site.new and site.slices_used == 0 and not site.enabled:
 213                                 query.append(site)
 214                         elif filter == "pending" and not site.enabled:
 215                                 query.append(site)
 216                         elif filter == site.status:
 217                                 query.append(site)
 218
 219                 return dict(query=query, fc=filtercount)
 220
 221         @expose(template="monitorweb.templates.actionlist")
 222         def action(self, filter='all'):
 223                 session.bind = metadata.bind
 224                 filtercount = {'active' : 0, 'acknowledged': 0, 'all' : 0}
 225                 # With Acknowledgement
 226                 sql_ack = 'SELECT DISTINCT h.host,t.description,t.priority,t.lastchange,a.message '+ \
 227               ' FROM triggers t,hosts h,items i,functions f, hosts_groups hg,escalations e,acknowledges a ' + \
 228               ' WHERE f.itemid=i.itemid ' + \
 229                   ' AND h.hostid=i.hostid ' + \
 230                   ' AND hg.hostid=h.hostid ' + \
 231                   ' AND t.triggerid=f.triggerid ' + \
 232                   ' AND t.triggerid=e.triggerid ' + \
 233                   ' AND a.eventid=e.eventid ' + \
 234                   ' AND t.status=' + str(defines.TRIGGER_STATUS_ENABLED) + \
 235                   ' AND i.status=' + str(defines.ITEM_STATUS_ACTIVE) + \
 236                   ' AND h.status=' + str(defines.HOST_STATUS_MONITORED) + \
 237                   ' AND t.value=' + str(defines.TRIGGER_VALUE_TRUE) + \
 238               ' ORDER BY t.lastchange DESC';
 239
 240                 # WithOUT Acknowledgement
 241                 sql_noack = 'SELECT DISTINCT h.host,t.description,t.priority,t.lastchange,e.eventid ' + \
 242               ' FROM triggers t,hosts h,items i,functions f, hosts_groups hg,escalations e,acknowledges a ' + \
 243               ' WHERE f.itemid=i.itemid ' + \
 244                   ' AND h.hostid=i.hostid ' + \
 245                   ' AND hg.hostid=h.hostid ' + \
 246                   ' AND t.triggerid=f.triggerid ' + \
 247                   ' AND t.triggerid=e.triggerid ' + \
 248                   ' AND e.eventid not in (select eventid from acknowledges) ' + \
 249                   ' AND t.status=' + str(defines.TRIGGER_STATUS_ENABLED) + \
 250                   ' AND i.status=' + str(defines.ITEM_STATUS_ACTIVE) + \
 251                   ' AND h.status=' + str(defines.HOST_STATUS_MONITORED) + \
 252                   ' AND t.value=' + str(defines.TRIGGER_VALUE_TRUE) + \
 253               ' ORDER BY t.lastchange DESC';
 254                 # for i in session.execute(sql): print i
 255
 256                 query=[]
 257                 replace = re.compile(' {.*}')
 258                 for sql,ack in [(sql_ack,True), (sql_noack,False)]:
 259                         result = session.execute(sql)
 260                         for row in result:
 261                                 try:
 262                                         newrow = [ site_hn2lb[row[0].lower()] ] + [ r for r in row ]
 263                                 except:
 264                                         print site_hn2lb.keys()
 265                                         newrow = [ "unknown" ] + [ r for r in row ]
 266
 267                                 newrow[2] = replace.sub("", newrow[2]) # strip {.*} expressions
 268
 269                                 # NOTE: filter count
 270                                 filtercount['all'] += 1
 271                                 if not ack: # for unacknowledged
 272                                         filtercount['active'] += 1
 273                                         if filter == 'active':
 274                                                 query.append(newrow)
 275                                 else:
 276                                         filtercount['acknowledged'] += 1
 277                                         if filter == 'acknowledged':
 278                                                 query.append(newrow)
 279
 280                                 if filter != "acknowledged" and filter != "active":
 281                                         query.append(newrow)
 282
 283                 return dict(query=query, fc=filtercount)