+ self.updateView()
+
+ def submitFinished(self):
+ self.disconnect(self.process, SIGNAL('finished()'), self.submitFinished)
+
+ faultString = self.process.getFaultString()
+ if not faultString:
+ self.setStatus("<font color='green'>Slice data submitted.</font>")
+ else:
+ self.setStatus("<font color='red'>Slice submit failed: %s</font>" % (faultString))
+
+ # no need to do that anymore
+ # QTimer.singleShot(1000, self.refresh)
+ self.updateView()
+ self.parent().signalAll("rspecUpdated")
+
+ def refreshResourcesFinished(self):
+ self.disconnect(self.process, SIGNAL('finished()'), self.refreshResourcesFinished)
+
+ self.setStatus("Refreshing slice RSpec.")
+ self.connect(self.process, SIGNAL('finished()'), self.refreshRSpecFinished)
+ self.process.retrieveRspec()
+
+ def refreshRSpecFinished(self):
+ self.disconnect(self.process, SIGNAL('finished()'), self.refreshRSpecFinished)
+ self.setStatus("<font color='green'>Slice data refreshed.</font>", timeout=5000)
+ 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, resources, item, depth = 0):
+ change = False
+ 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 = resources.get_node_element(hostname)
+
+ if resource_node==None:
+ print "Error: Failed to find %s in resources rspec" % hostname
+ else:
+ rspec.merge_node(resource_node, testbed)
+ rspec.add_slivers(str(hostname), testbed)
+ change = True
+ elif status == node_status['remove']:
+ print "Remove hostname: %s" % hostname
+ rspec.remove_slivers(str(hostname), testbed)
+ change = True
+ 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
+ node = self.itemText(item.parent())
+ testbed = self.itemText(item.parent().parent())
+ if status == tag_status['add']:
+ print "Add tag to (%s, %s): %s/%s " % (testbed, node, tag, value)
+ if node.startsWith(default_tags):
+ rspec.add_default_sliver_attribute(tag, value, testbed)
+ else:
+ rspec.add_sliver_attribute(node, tag, value, testbed)
+ change = True
+ elif status == tag_status['remove']:
+ print "Remove tag from (%s, %s): %s/%s " % (testbed, node, tag, value)
+ if node.startsWith(default_tags):
+ rspec.remove_default_sliver_attribute(tag, value, testbed)
+ else:
+ rspec.remove_sliver_attribute(node, tag, value, testbed)
+ change = True
+
+ children = item.rowCount()
+ for row in range(0, children):
+ status = self.process_subtree(rspec, resources, item.child(row), depth + 1)
+ change = change or status
+
+ return change
+
+ def submit(self):
+ if self.checkRunningProcess():
+ return
+
+ rspec = SfiData().getSliceRSpec()
+ resources = SfiData().getResourcesRSpec()
+ change = self.process_subtree(rspec, resources, self.nodeModel.invisibleRootItem())
+
+ if not change:
+ self.setStatus("<font color=red>No change in slice data. Not submitting!</font>", timeout=3000)
+ return
+
+ self.connect(self.process, SIGNAL('finished()'), self.submitFinished)
+
+ self.process.applyRSpec(rspec)
+ self.setStatus("Sending slice data (RSpec). This will take some time...")
+
+ def renew(self):
+ dlg = RenewWindow(parent=self)
+ dlg.exec_()