def methodSelectionChanged(self, state: int, method_name):
state = Qt.CheckState(state)
if state == Qt.Checked:
self.selected_methods.add(method_name)
elif method_name in self.selected_methods:
self.selected_methods.remove(method_name)
self.update_scores()
def nextCheckState(self, state, index):
# type: (Qt.CheckState, QModelIndex) -> Qt.CheckState
"""
Return the next check state for index.
"""
constraint = index.data(HasConstraintRole)
flags = index.flags()
if flags & Qt.ItemIsUserTristate and constraint:
return Qt.PartiallyChecked if state == Qt.Checked else Qt.Checked
elif flags & Qt.ItemIsUserTristate:
return Qt.CheckState((state + 1) % 3)
else:
return Qt.Unchecked if state == Qt.Checked else Qt.Checked
