diff --git a/qt_ui/windows/basemenu/base_defenses/QBaseDefenseGroupInfo.py b/qt_ui/windows/basemenu/base_defenses/QBaseDefenseGroupInfo.py index 94da80e5..370cf65a 100644 --- a/qt_ui/windows/basemenu/base_defenses/QBaseDefenseGroupInfo.py +++ b/qt_ui/windows/basemenu/base_defenses/QBaseDefenseGroupInfo.py @@ -35,10 +35,12 @@ class QBaseDefenseGroupInfo(QGroupBox): def buildLayout(self): unit_dict = {} - for i in range(self.unit_layout.count()): - item = self.unit_layout.itemAt(i) - if item is not None and item.widget() is not None: - self.unit_layout.removeItem(item) + for i in range(self.unit_layout.rowCount()): + for j in range(self.unit_layout.columnCount()): + item = self.unit_layout.itemAtPosition(i, j) + if item is not None and item.widget() is not None: + item.widget().setParent(None) + print("Remove " + str(i) + ", " + str(j)) for g in self.ground_object.groups: for u in g.units: @@ -58,6 +60,11 @@ class QBaseDefenseGroupInfo(QGroupBox): self.unit_layout.addWidget(QLabel(str(v) + " x " + "" + k + ""), i, 1) i = i + 1 + if len(unit_dict.items()) == 0: + self.unit_layout.addWidget(QLabel("/"), 0, 0) + + + self.setLayout(self.main_layout) def onManage(self):