+ print pcuid
+ return self.pcuview(None, pcuid, **dict(exceptions=tg_exceptions))
+
+ def nodeaction(self, **data):
+ print "NODEACTION------------------"
+ 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. Allow at most 5 minutes. Then 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
+
+ @expose(template="monitorweb.templates.simpleview")
+ def simpleview(self, **data):
+ return self.pre_view(**data)
+
+ @expose(template="monitorweb.templates.simpleview")
+ def pcuview(self, **data):
+ return self.pre_view(**data)
+
+ @expose(template="monitorweb.templates.detailview")
+ def detailview(self, **data):
+ return self.pre_view(**data)
+
+
+ def pre_view(self, **data):
+ session_clear_all()
+
+ loginbase=None
+ loginbase_list=[]
+ hostname=None
+ pcuid=None
+ since=20
+ # if objtype is not None, then treat 'hostname' or 'loginbase' as a search pattern
+ objtype=None
+
+ exceptions = None
+ sitequery=[]
+ nodequery=[]
+ pcuquery=[]
+ actions=[]
+ actions_list=[]
+
+ for key in data:
+ print key, data[key]
+
+ if 'query' in data:
+ obj = data['query']
+ fields = obj.split(":")
+ if len(fields) > 1:
+ objtype = fields[0]
+ obj = fields[1].replace("*", "%")
+ print "obj: %s"% obj
+
+ if len(obj.split(".")) > 1 or objtype == "node":
+ hostname = obj
+ else:
+ loginbase = obj
+
+ if 'loginbase' in data:
+ loginbase = data['loginbase']
+
+ if 'hostname' in data:
+ hostname = data['hostname']
+
+ if 'pcuid' in data:
+ try: pcuid = int(data['pcuid'])
+ except: pcuid = None
+
+ if 'since' in data:
+ try: since = int(since)
+ except: since = 20
+
+ if pcuid:
+ print "pcuid: %s" % pcuid
+ pcu = FindbadPCURecord.get_latest_by(plc_pcuid=pcuid)
+ loginbase_list += [ PlcSite.query.get(pcu.plc_pcu_stats['site_id']).plc_site_stats['login_base'] ]
+
+ if hostname:
+ if not objtype:
+ nodes = [ FindbadNodeRecord.get_latest_by(hostname=hostname) ]
+ else:
+ nodes = FindbadNodeRecord.query.filter(FindbadNodeRecord.hostname.like(hostname))
+
+ for node in nodes:
+ lb = PlcSite.query.get(node.plc_node_stats['site_id']).plc_site_stats['login_base']
+ if lb not in loginbase_list:
+ loginbase_list += [ lb ]
+
+ if loginbase:
+ if not objtype:
+ loginbase_list = [ loginbase ]
+ else:
+ loginbase_list = HistorySiteRecord.query.filter(HistorySiteRecord.loginbase.like(loginbase))
+ loginbase_list = [ l.loginbase for l in loginbase_list ]
+
+
+ if loginbase_list:
+ for loginbase in loginbase_list:
+ actions = ActionRecord.query.filter_by(loginbase=loginbase
+ ).filter(ActionRecord.date_created >= datetime.now() - timedelta(since)
+ ).order_by(ActionRecord.date_created.desc())
+ actions_list += [ a for a in actions ]
+ site = HistorySiteRecord.by_loginbase(loginbase)
+ if site:
+ sitequery.append(site)
+ # NOTE: because a single pcu may be assigned to multiple hosts,
+ # track unique pcus by their plc_pcuid, then turn dict into list
+ pcus = {}
+ for node in FindbadNodeRecord.query.filter_by(loginbase=loginbase):
+ # NOTE: reformat some fields.
+ agg = prep_node_for_display(node)
+ nodequery += [agg]
+ if agg.pcu:
+ pcus[agg.pcu.pcu.plc_pcuid] = agg.pcu
+
+ for pcuid_key in pcus:
+ pcuquery += [pcus[pcuid_key]]
+
+ actionlist_widget = ActionListWidget(template='monitorweb.templates.actionlist_template')
+ return dict(sitequery=sitequery, pcuquery=pcuquery, nodequery=nodequery, actions=actions_list, actionlist_widget=actionlist_widget, since=since, exceptions=exceptions)
+
+
+ # TODO: add form validation