+ faultString = self.process.getFaultString()
+ if not faultString:
+ self.setStatus("<font color='green'>Slice data submitted.</font>")
+ QTimer.singleShot(2500, self.refresh)
+ else:
+ self.setStatus("<font color='red'>Slice submit failed: %s</font>" % (faultString))
+
+ def refreshResourcesFinished(self):
+ self.disconnect(self.process, SIGNAL('finished()'), self.refreshResourcesFinished)
+
+ faultString = self.process.getFaultString()
+ if not faultString:
+ self.setStatus("Refreshing slice RSpec.")
+ self.connect(self.process, SIGNAL('finished()'), self.refreshRSpecFinished)
+ self.process.retrieveRspec()
+ else:
+ self.setStatus("<font color='red'>Resources refresh failed: %s</font>" % (faultString))
+
+ def refreshRSpecFinished(self):
+ self.disconnect(self.process, SIGNAL('finished()'), self.refreshRSpecFinished)
+
+ faultString = self.process.getFaultString()
+ if not faultString:
+ self.setStatus("<font color='green'>Slice data refreshed.</font>", timeout=5000)
+ else:
+ self.setStatus("<font color='red'>Slice refresh failed: %s</font>" % (faultString))
+
+ self.updateView()
+ self.parent().signalAll("rspecUpdated")
+
+ def setStatus(self, msg, timeout=None):
+ self.parent().setStatus(msg, timeout)
+
+ def checkRunningProcess(self):
+ if self.process.isRunning():
+ self.setStatus("<font color='red'>There is already a process running. Please wait.</font>")
+ return True
+ return False
+
+ def search(self, search_string):
+ self.filterModel.setHostNameFilter(str(search_string))
+
+ def filter(self, filter_string):
+ self.filterModel.setNodeStatusFilter(str(filter_string))
+
+ def itemStatus(self, item):
+ statusItem = item.parent().child(item.row(), MEMBERSHIP_STATUS_COLUMN)
+ return str(statusItem.data(Qt.DisplayRole).toString())
+
+ def itemText(self, item):
+ return str(item.data(Qt.DisplayRole).toString())
+
+ # Recursively walk the tree, making changes to the RSpec
+ def process_subtree(self, rspec, rspec_node_names, resource_node_names, item, depth = 0):
+ changeAdd = 0
+ changeRemove = 0
+ changeAddTag = 0
+ changeRemoveTag = 0
+ model = self.nodeModel
+
+ if depth in [0, 1]:
+ pass
+ elif depth == 2: # Hostname
+ hostname = self.itemText(item)
+ testbed = self.itemText(item.parent())
+ status = self.itemStatus(item)
+ if status == node_status['add']:
+ print "Add hostname: %s" % hostname
+
+ resource_node = resource_node_names.get(hostname, None)
+
+ if resource_node==None:
+ print "Error: Failed to find %s in resources rspec" % hostname
+ else:
+ if not (hostname in rspec_node_names):
+ network_name = Xrn(resource_node['component_manager_id']).get_hrn()
+ rspec.version.add_network(network_name)
+ rspec.version.add_nodes([resource_node])
+ rspec.version.add_slivers([str(hostname)])
+ changeAdd += 1
+ elif status == node_status['remove']:
+ print "Remove hostname: %s" % hostname
+ rspec.version.remove_slivers([str(hostname)])
+ changeRemove += 1
+ elif depth == 3: # Tag
+ tag, value = self.itemText(item).split(": ")
+ status = self.itemStatus(item)
+ tag = "%s" % tag # Prevent weird error from lxml
+ value = "%s" % value # Prevent weird error from lxml
+ hostname = self.itemText(item.parent())
+ testbed = self.itemText(item.parent().parent())
+ if status == tag_status['add']:
+ print "Add tag to (%s, %s): %s/%s " % (testbed, hostname, tag, value)
+ if hostname.startswith(default_tags):
+ rspec.version.add_default_sliver_attribute(tag, value, testbed)
+ else:
+ node = rspec_node_names.get(hostname, None)
+ if node:
+ rspec.version.add_sliver_attribute(node['component_id'], tag, value, testbed)
+ changeAddTag += 1
+ elif status == tag_status['remove']:
+ print "Remove tag from (%s, %s): %s/%s " % (testbed, hostname, tag, value)
+ if hostname.startswith(default_tags):
+ rspec.version.remove_default_sliver_attribute(tag, value, testbed)
+ else:
+ node = rspec_node_names.get(hostname, None)
+ if node:
+ rspec.version.remove_sliver_attribute(node['component_id'], tag, value, testbed)
+ changeRemoveTag += 1
+
+ children = item.rowCount()
+ for row in range(0, children):
+ (c, ca, cr, cat, crt) = self.process_subtree(rspec, rspec_node_names, resource_node_names, item.child(row), depth + 1)
+ changeAdd = changeAdd + ca
+ changeRemove = changeRemove + cr
+ changeAddTag = changeAddTag + cat
+ changeRemoveTag = changeRemoveTag + crt
+
+ change = changeAdd + changeRemove + changeAddTag + changeRemoveTag
+
+ return (change, changeAdd, changeRemove, changeAddTag, changeRemoveTag)
+
+ def submit(self):
+ if self.checkRunningProcess():
+ return
+
+ rspec = SfiData().getSliceRSpec()
+ resources = SfiData().getResourcesRSpec()
+
+ resource_node_names = self.nodesByName(resources.version.get_nodes())
+ rspec_node_names = self.nodesByName(rspec.version.get_nodes_with_slivers())
+
+ (change, changeAdd, changeRemove, changeAddTag, changeRemoveTag) = \
+ self.process_subtree(rspec, rspec_node_names, resource_node_names, self.nodeModel.invisibleRootItem())
+
+ if (change <= 0):
+ self.setStatus("<font color=red>No change in slice data. Not submitting!</font>", timeout=3000)
+ return
+
+ if (changeRemove > 0) and (rspec.version.get_nodes_with_slivers() == []):
+ # removing the last sliver in a slice requires us to call deleteSlivers
+ self.connect(self.process, SIGNAL('finished()'), self.deleteSliversFinished)
+ self.process.deleteSlivers()
+ return
+
+ # Several aggregates have issues with the <statistics> section in the
+ # rspec, so make sure it's not there.
+ stats_elems = rspec.xml.xpath("//statistics")
+ if len(stats_elems)>0:
+ stats_elem = stats_elems[0]
+ parent = stats_elem.xpath("..")[0]
+ parent.remove(stats_elem)
+
+ self.connect(self.process, SIGNAL('finished()'), self.submitFinished)
+ self.process.applyRSpec(rspec)
+ self.setStatus("Sending slice data (RSpec). This will take some time...")
+
+ def submit_pg_compat(self):
+ if self.checkRunningProcess():
+ return
+
+ rspec = SfiData().getSliceRSpec()
+ resources = SfiData().getResourcesRSpec()
+
+ (change, changeAdd, changeRemove, changeAddTag, changeRemoveTag) = \
+ self.process_subtree(rspec, rspec, resources, self.nodeModel.invisibleRootItem())
+
+ if (change <= 0):
+ self.setStatus("<font color=red>No change in slice data. Not submitting!</font>", timeout=3000)
+ return
+
+ dlg = ClientSliceManager(self)
+ dlg.submit_pg_compat(rspec)
+ dlg.exec_()
+
+ self.setStatus("<font color='green'>Finished submitting. %d/%d aggs succeeded.</font>" %
+ (dlg.submit_aggSuccessCount,dlg.submit_aggSuccessCount+dlg.submit_aggFailCount))
+ QTimer.singleShot(2500, self.refresh)
+
+ def renew(self):
+ dlg = RenewWindow(parent=self)
+ dlg.exec_()
+
+ def refresh(self):