- for counter in range(len(unit_data.UNIT_CATEGORIES)):
- if selections['selected_category'] == unit_data.UNIT_CATEGORIES[counter]:
- # Restore the previously selected category.
- self._categoryView.set_cursor(counter, self._categoryColumn, False)
- self._categoryView.grab_focus()
- historical_catergory_found = True
-
- if not historical_catergory_found:
- print "Couldn't find saved category, using default."
- #If historical records were not kept then default to
- # put the focus on the first category
- self._categoryView.set_cursor(0, self._categoryColumn, False)
- self._categoryView.grab_focus()
+ selectedCategoryName = selections['selected_category']
+ try:
+ categoryIndex = unit_data.UNIT_CATEGORIES.index(selectedCategoryName)
+ except ValueError:
+ _moduleLogger.warn("Unknown category: %s" % selectedCategoryName)
+
+ self._categorySelectionButton.set_label(selectedCategoryName)
+ self._categoryView.set_cursor(categoryIndex, self._categoryColumn, False)
+ self._categoryView.grab_focus()