weyne/dcs-retribution
1
0
Fork 0
mirror of https://github.com/dcs-retribution/dcs-retribution.git synced 2025-11-10 15:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix GroundObject Menu

Browse Source 
- fixed incorrect Buy button enable / disable
- update the game after repair
This commit is contained in:
RndName 2022-03-27 20:31:10 +02:00
parent 5f010bb94d
commit c9df8cc803
No known key found for this signature in database
GPG Key ID: 5EF516FD9537F7C0
2 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
qt_ui/windows/groundobject/QGroundObjectBuyMenu.py
View File

@ -174,6 +174,14 @@ class QGroundObjectTemplateLayout(QGroupBox):
self.add_theater_group(group_name, self.layout_model.force_group, groups) self.add_theater_group(group_name, self.layout_model.force_group, groups)
self.group_template_changed() self.group_template_changed()
@property
def cost(self) -> int:
return self.layout_model.price - self.current_group_value
@property
def affordable(self) -> bool:
return self.cost <= self.game.blue.budget
def add_theater_group( def add_theater_group(
self, group_name: str, force_group: ForceGroup, groups: list[TgoLayoutGroup] self, group_name: str, force_group: ForceGroup, groups: list[TgoLayoutGroup]
) -> None: ) -> None:
@ -193,15 +201,14 @@ class QGroundObjectTemplateLayout(QGroupBox):
def group_template_changed(self) -> None: def group_template_changed(self) -> None:
price = self.layout_model.price price = self.layout_model.price
self.buy_button.setText(f"Buy [${price}M][-${self.current_group_value}M]") self.buy_button.setText(f"Buy [${price}M][-${self.current_group_value}M]")
self.buy_button.setEnabled(price <= self.game.blue.budget) self.buy_button.setEnabled(self.affordable)
if self.buy_button.isEnabled(): if self.buy_button.isEnabled():
self.buy_button.setToolTip("Buy the group") self.buy_button.setToolTip(f"Buy the group for ${self.cost}M")
else: else:
self.buy_button.setToolTip("Not enough money to buy this group") self.buy_button.setToolTip("Not enough money to buy this group")
def buy_group(self) -> None: def buy_group(self) -> None:
price = self.layout_model.price if not self.affordable:
if price > self.game.blue.budget:
# Something went wrong. Buy button should be disabled! # Something went wrong. Buy button should be disabled!
logging.error("Not enough money to buy the group") logging.error("Not enough money to buy the group")
return return
@ -211,7 +218,7 @@ class QGroundObjectTemplateLayout(QGroupBox):
self.game.theater.heading_to_conflict_from(self.ground_object.position) self.game.theater.heading_to_conflict_from(self.ground_object.position)
or self.ground_object.heading or self.ground_object.heading
) )
self.game.blue.budget -= price - self.current_group_value self.game.blue.budget -= self.cost
self.ground_object.groups = [] self.ground_object.groups = []
for group_name, groups in self.layout_model.groups.items(): for group_name, groups in self.layout_model.groups.items():
for group in groups: for group in groups:

2
qt_ui/windows/groundobject/QGroundObjectMenu.py
View File

@ -238,7 +238,7 @@ class QGroundObjectMenu(QDialog):
logging.info("Removed destroyed units " + str(d)) logging.info("Removed destroyed units " + str(d))
logging.info(f"Repaired unit: {unit.unit_name}") logging.info(f"Repaired unit: {unit.unit_name}")
self.do_refresh_layout() self.update_game()
def rotate_tgo(self, heading: Heading) -> None: def rotate_tgo(self, heading: Heading) -> None:
self.ground_object.rotate(heading) self.ground_object.rotate(heading)