+class NodeFilterProxyModel(QSortFilterProxyModel):
+ def __init__(self, parent=None):
+ QSortFilterProxyModel.__init__(self, parent)
+ self.hostname_filter_regex = None
+ self.nodestatus_filter = None
+
+ def setHostNameFilter(self, hostname):
+ self.hostname_filter_regex = QRegExp(hostname)
+ self.invalidateFilter()
+
+ def setNodeStatusFilter(self, status):
+ if (status == "all"):
+ self.nodestatus_filter = None
+ else:
+ self.nodestatus_filter = status
+ self.invalidateFilter()
+
+ def filterAcceptsRow(self, sourceRow, source_parent):
+ kind_data = self.sourceModel().index(sourceRow, KIND_COLUMN, source_parent).data().toString()
+ if (kind_data == "node"):
+ if self.hostname_filter_regex:
+ name_data = self.sourceModel().index(sourceRow, NAME_COLUMN, source_parent).data().toString()
+ if (self.hostname_filter_regex.indexIn(name_data) < 0):
+ return False
+ if self.nodestatus_filter:
+ nodestatus_data = self.sourceModel().index(sourceRow, NODE_STATUS_COLUMN, source_parent).data().toString()
+ if (nodestatus_data != self.nodestatus_filter):
+ return False
+ return True
+
+ def lessThan(self, left, right):
+ l_str = str(left.data().toString())
+ r_str = str(right.data().toString())
+
+ # make sure default_tags appears before everything else
+ if l_str.startswith(default_tags):
+ return True
+
+ if r_str.startswith(default_tags):
+ return False
+
+ return (l_str < r_str)
+
+