Skip to content

Commit ead9ed6

Browse files
committed
Fix to PowerSelector for nested powersets eg Corruptor -> Dual Pistols
1 parent d8d1b39 commit ead9ed6

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

UI/PowerSelector.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -70,10 +70,18 @@ def __init__(self, parent, powerselector: PowerSelector):
7070

7171
self.Popups = []
7272
for power in powers:
73-
popuppower = PopupPower(powerpanel, powerset, power)
74-
popuppower.CB.SetValue(power in self.PowerSelector.Powers)
75-
powersizer.Add(popuppower, 0, wx.EXPAND)
76-
self.Popups.append(popuppower)
73+
if isinstance(power, dict):
74+
for plist in power.values():
75+
for p in plist:
76+
popuppower = PopupPower(powerpanel, powerset, p)
77+
popuppower.CB.SetValue(p in self.PowerSelector.Powers)
78+
powersizer.Add(popuppower, 0, wx.EXPAND)
79+
self.Popups.append(popuppower)
80+
else:
81+
popuppower = PopupPower(powerpanel, powerset, power)
82+
popuppower.CB.SetValue(power in self.PowerSelector.Powers)
83+
powersizer.Add(popuppower, 0, wx.EXPAND)
84+
self.Popups.append(popuppower)
7785

7886
powerpanel.SetSizerAndFit(powersizer)
7987

0 commit comments

Comments
 (0)