- if col is self._unitNameColumn:
- selectedUnitColumn = 0
- self._unitNameColumn.set_sort_indicator(True)
- self._unitValueColumn.set_sort_indicator(False)
- self._unitSymbolColumn.set_sort_indicator(False)
- self._unitNameColumn.set_sort_order(not self._unit_sort_direction)
- elif col is self._unitValueColumn:
- selectedUnitColumn = 1
- self._unitNameColumn.set_sort_indicator(False)
- self._unitValueColumn.set_sort_indicator(True)
- self._unitSymbolColumn.set_sort_indicator(False)
- self._unitValueColumn.set_sort_order(not self._value_sort_direction)
- elif col is self._unitSymbolColumn:
- selectedUnitColumn = 2
- self._unitNameColumn.set_sort_indicator(False)
- self._unitValueColumn.set_sort_indicator(False)
- self._unitSymbolColumn.set_sort_indicator(True)
- self._unitSymbolColumn.set_sort_order(not self._units_sort_direction)
+ self._unitNameColumn.set_sort_indicator(False)
+ self._unitValueColumn.set_sort_indicator(False)
+ self._unitSymbolColumn.set_sort_indicator(False)
+ for selectedUnitColumn, (maybeCol, sortDirection) in enumerate((
+ (self._unitNameColumn, self._unit_sort_direction),
+ (self._unitValueColumn, self._value_sort_direction),
+ (self._unitSymbolColumn, self._units_sort_direction),
+ )):
+ if col is maybeCol:
+ col.set_sort_indicator(True)
+ col.set_sort_order(not sortDirection)
+ break