mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from PySide6.QtCore import Qt
|
|
from PySide6.QtWidgets import QGridLayout, QScrollArea, QVBoxLayout, QWidget
|
|
|
|
from game.dcs.groundunittype import GroundUnitType
|
|
from game.purchaseadapter import GroundUnitPurchaseAdapter
|
|
from game.theater import ControlPoint
|
|
from game.theater.player import Player
|
|
from qt_ui.models import GameModel
|
|
from qt_ui.windows.basemenu.UnitTransactionFrame import UnitTransactionFrame
|
|
|
|
|
|
class QArmorRecruitmentMenu(UnitTransactionFrame[GroundUnitType]):
|
|
def __init__(self, cp: ControlPoint, game_model: GameModel):
|
|
super().__init__(
|
|
game_model,
|
|
GroundUnitPurchaseAdapter(
|
|
cp, game_model.game.coalition_for(cp.captured), game_model.game
|
|
),
|
|
)
|
|
self.cp = cp
|
|
self.game_model = game_model
|
|
self.purchase_groups = {}
|
|
self.bought_amount_labels = {}
|
|
self.existing_units_labels = {}
|
|
|
|
main_layout = QVBoxLayout()
|
|
|
|
scroll_content = QWidget()
|
|
task_box_layout = QGridLayout()
|
|
scroll_content.setLayout(task_box_layout)
|
|
row = 0
|
|
|
|
unit_types = list(
|
|
set(self.game_model.game.faction_for(player=Player.BLUE).ground_units)
|
|
)
|
|
unit_types.sort(key=lambda u: u.display_name)
|
|
for row, unit_type in enumerate(unit_types):
|
|
self.add_purchase_row(unit_type, task_box_layout, row)
|
|
stretch = QVBoxLayout()
|
|
stretch.addStretch()
|
|
task_box_layout.addLayout(stretch, row, 0)
|
|
|
|
scroll_content.setLayout(task_box_layout)
|
|
scroll = QScrollArea()
|
|
scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
|
|
scroll.setVerticalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOn)
|
|
scroll.setWidgetResizable(True)
|
|
scroll.setWidget(scroll_content)
|
|
main_layout.addWidget(scroll)
|
|
self.setLayout(main_layout)
|