Properly get enum instead of value on strategy

selecter
This commit is contained in:
walterroach 2020-10-29 23:38:14 -05:00 committed by Dan Albert
parent 0b2fbddbc5
commit 5adc92c601

View File

@ -14,8 +14,8 @@ class QGroundForcesStrategySelector(QComboBox):
self.cp.stances[enemy_cp.id] = CombatStance.DEFENSIVE
for i, stance in enumerate(CombatStance):
self.addItem(stance.name, userData=stance)
if self.cp.stances[enemy_cp.id] == stance:
self.addItem(stance.name, userData=CombatStance(i))
if self.cp.stances[enemy_cp.id] == CombatStance(i):
self.setCurrentIndex(i)
self.currentTextChanged.connect(self.on_change)