+ rspec_file = config.getSliceRSpecFile()
+ if os.path.exists(rspec_file):
+ xml = open(rspec_file).read()
+ return parse_rspec(xml)
+ return None
+
+ 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 filter(self, filter_string):
+ # for hierarchical models QSortFilterProxyModel applies the
+ # sort recursively. if the parent doesn't match the criteria
+ # we won't be able to match the children. so we need to match
+ # parent (by matching the network_names)
+ networks = ["^%s$" % n for n in self.network_names]
+ filters = networks + [str(filter_string)]
+ self.filterModel.setFilterRegExp(QRegExp('|'.join(filters)))
+
+ def itemStatus(self, item):
+ statusItem = item.parent().child(item.row(), 1)
+ return statusItem.data(Qt.DisplayRole).toString()
+
+ def itemText(self, item):
+ return item.data(Qt.DisplayRole).toString()
+
+ # Recursively walk the tree, making changes to the RSpec
+ def process_subtree(self, rspec, 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
+ 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, item.child(row), depth + 1)
+ change = change or status
+
+ return change
+
+ def submit(self):
+ if self.checkRunningProcess():
+ return
+
+ rspec = self.readSliceRSpec()
+ change = self.process_subtree(rspec, self.nodeModel.invisibleRootItem())
+
+ if not change:
+ self.setStatus("<font color=red>No change in slice data. Not submitting!</font>", timeout=3000)
+ return
+
+ self.disconnect(self.process, SIGNAL('finished()'), self.refreshFinished)
+ 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)
+ if (dlg.exec_() == QDialog.Accepted):
+ self.setStatus("Renewing Slice.")