split dialogs and war/campaign/round classes to files
This commit is contained in:
Maxime Réaux
parent
55abdccc64
commit
019e62565f
22 changed files with 562 additions and 502 deletions
|
|
@ -1,17 +1,10 @@
|
|||
from typing import cast, Callable, List
|
||||
from typing import Callable, List
|
||||
from pathlib import Path
|
||||
import calendar
|
||||
|
||||
from PyQt6 import QtWidgets
|
||||
from PyQt6.QtCore import Qt, QPoint
|
||||
from PyQt6.QtWidgets import (
|
||||
QWidget,
|
||||
QDialog,
|
||||
QFileDialog,
|
||||
QTreeWidgetItem,
|
||||
QMenu,
|
||||
QComboBox,
|
||||
)
|
||||
from PyQt6.QtWidgets import QWidget, QFileDialog, QTreeWidgetItem, QMenu
|
||||
from PyQt6.QtGui import QCloseEvent
|
||||
|
||||
from warchron.constants import ROLE_TYPE, ROLE_ID, ItemType
|
||||
|
|
@ -21,49 +14,17 @@ from warchron.controller.dtos import (
|
|||
WarDTO,
|
||||
WarParticipantDTO,
|
||||
ObjectiveDTO,
|
||||
CampaignDTO,
|
||||
CampaignParticipantDTO,
|
||||
SectorDTO,
|
||||
RoundDTO,
|
||||
ChoiceDTO,
|
||||
BattleDTO,
|
||||
)
|
||||
from warchron.view.ui.ui_main_window import Ui_MainWindow
|
||||
from warchron.view.ui.ui_player_dialog import Ui_playerDialog
|
||||
from warchron.view.ui.ui_war_dialog import Ui_warDialog
|
||||
from warchron.view.ui.ui_campaign_dialog import Ui_campaignDialog
|
||||
from warchron.view.ui.ui_objective_dialog import Ui_objectiveDialog
|
||||
from warchron.view.ui.ui_war_participant_dialog import Ui_warParticipantDialog
|
||||
from warchron.view.ui.ui_campaign_participant_dialog import (
|
||||
Ui_campaignParticipantDialog,
|
||||
from warchron.view.helpers import (
|
||||
format_campaign_label,
|
||||
format_round_label,
|
||||
format_war_label,
|
||||
)
|
||||
from warchron.view.ui.ui_sector_dialog import Ui_sectorDialog
|
||||
from warchron.view.ui.ui_choices_dialog import Ui_choicesDialog
|
||||
from warchron.view.ui.ui_battle_result_dialog import Ui_battleResultDialog
|
||||
|
||||
# utils...
|
||||
|
||||
|
||||
def select_if_exists(combo: QComboBox, value: str | None) -> None:
|
||||
if value is None:
|
||||
return
|
||||
idx = combo.findData(value)
|
||||
if idx != -1:
|
||||
combo.setCurrentIndex(idx)
|
||||
|
||||
|
||||
def format_war_label(war: WarDTO) -> str:
|
||||
return f"{war.name} ({war.year})"
|
||||
|
||||
|
||||
def format_campaign_label(camp: CampaignDTO) -> str:
|
||||
return f"{camp.name} ({calendar.month_name[camp.month]})"
|
||||
|
||||
|
||||
def format_round_label(index: int) -> str:
|
||||
if index is None:
|
||||
return ""
|
||||
return f"Round {index}"
|
||||
from warchron.view.ui.ui_main_window import Ui_MainWindow
|
||||
|
||||
|
||||
class View(QtWidgets.QMainWindow, Ui_MainWindow):
|
||||
|
|
@ -517,285 +478,3 @@ class View(QtWidgets.QMainWindow, Ui_MainWindow):
|
|||
table.setItem(row, 1, player_1_item)
|
||||
table.setItem(row, 2, player_2_item)
|
||||
table.resizeColumnsToContents()
|
||||
|
||||
|
||||
class PlayerDialog(QDialog):
|
||||
def __init__(
|
||||
self, parent: QWidget | None = None, *, default_name: str = ""
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_playerDialog = Ui_playerDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
self.ui.playerName.setText(default_name)
|
||||
|
||||
def get_player_name(self) -> str:
|
||||
return self.ui.playerName.text().strip()
|
||||
|
||||
|
||||
class WarDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
default_name: str = "",
|
||||
default_year: int | None = None,
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_warDialog = Ui_warDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
self.ui.warName.setText(default_name)
|
||||
if default_year is not None:
|
||||
self.ui.warYear.setValue(default_year)
|
||||
|
||||
def get_war_name(self) -> str:
|
||||
return self.ui.warName.text().strip()
|
||||
|
||||
def get_war_year(self) -> int:
|
||||
return int(self.ui.warYear.value())
|
||||
|
||||
|
||||
class CampaignDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
default_name: str = "",
|
||||
default_month: int | None = None,
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_campaignDialog = Ui_campaignDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
self.ui.campaignName.setText(default_name)
|
||||
if default_month is not None:
|
||||
self.ui.campaignMonth.setValue(default_month)
|
||||
|
||||
def get_campaign_name(self) -> str:
|
||||
return self.ui.campaignName.text().strip()
|
||||
|
||||
def get_campaign_month(self) -> int:
|
||||
return int(self.ui.campaignMonth.value())
|
||||
|
||||
|
||||
class ObjectiveDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
*,
|
||||
default_name: str = "",
|
||||
default_description: str | None = "",
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_objectiveDialog = Ui_objectiveDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
self.ui.objectiveName.setText(default_name)
|
||||
self.ui.objectiveDescription.setPlainText(default_description)
|
||||
|
||||
def get_objective_name(self) -> str:
|
||||
return self.ui.objectiveName.text().strip()
|
||||
|
||||
def get_objective_description(self) -> str:
|
||||
return self.ui.objectiveDescription.toPlainText().strip()
|
||||
|
||||
|
||||
class WarParticipantDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
*,
|
||||
players: List[ParticipantOption],
|
||||
default_player_id: str | None = None,
|
||||
default_faction: str | None = "",
|
||||
editable_player: bool = True,
|
||||
):
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_warParticipantDialog = Ui_warParticipantDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
for player in players:
|
||||
self.ui.playerComboBox.addItem(player.name, player.id)
|
||||
select_if_exists(self.ui.playerComboBox, default_player_id)
|
||||
self.ui.playerComboBox.setEnabled(editable_player)
|
||||
self.ui.faction.setText(default_faction)
|
||||
|
||||
def get_player_id(self) -> str:
|
||||
return cast(str, self.ui.playerComboBox.currentData())
|
||||
|
||||
def get_participant_faction(self) -> str:
|
||||
return self.ui.faction.text().strip()
|
||||
|
||||
|
||||
class CampaignParticipantDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
*,
|
||||
participants: List[ParticipantOption],
|
||||
default_participant_id: str | None = None,
|
||||
default_leader: str | None = "",
|
||||
default_theme: str | None = "",
|
||||
editable_player: bool = True,
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_campaignParticipantDialog = Ui_campaignParticipantDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
for part in participants:
|
||||
self.ui.playerComboBox.addItem(part.name, part.id)
|
||||
select_if_exists(self.ui.playerComboBox, default_participant_id)
|
||||
self.ui.playerComboBox.setEnabled(editable_player)
|
||||
self.ui.leader.setText(default_leader)
|
||||
self.ui.theme.setText(default_theme)
|
||||
|
||||
def get_player_id(self) -> str:
|
||||
return cast(str, self.ui.playerComboBox.currentData())
|
||||
|
||||
def get_participant_leader(self) -> str:
|
||||
return self.ui.leader.text().strip()
|
||||
|
||||
def get_participant_theme(self) -> str:
|
||||
return self.ui.theme.text().strip()
|
||||
|
||||
|
||||
class SectorDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
*,
|
||||
default_name: str = "",
|
||||
rounds: List[RoundDTO],
|
||||
default_round_id: str | None = None,
|
||||
objectives: List[ObjectiveDTO],
|
||||
default_major_id: str | None = None,
|
||||
default_minor_id: str | None = None,
|
||||
default_influence_id: str | None = None,
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_sectorDialog = Ui_sectorDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
self.ui.majorComboBox.addItem("(none)", None)
|
||||
self.ui.minorComboBox.addItem("(none)", None)
|
||||
self.ui.influenceComboBox.addItem("(none)", None)
|
||||
self.ui.sectorName.setText(default_name)
|
||||
for index, rnd in enumerate(rounds, start=1):
|
||||
self.ui.roundComboBox.addItem(format_round_label(index), rnd.id)
|
||||
select_if_exists(self.ui.roundComboBox, default_round_id)
|
||||
for obj in objectives:
|
||||
self.ui.majorComboBox.addItem(obj.name, obj.id)
|
||||
self.ui.minorComboBox.addItem(obj.name, obj.id)
|
||||
self.ui.influenceComboBox.addItem(obj.name, obj.id)
|
||||
select_if_exists(self.ui.majorComboBox, default_major_id)
|
||||
select_if_exists(self.ui.minorComboBox, default_minor_id)
|
||||
select_if_exists(self.ui.influenceComboBox, default_influence_id)
|
||||
|
||||
def get_sector_name(self) -> str:
|
||||
return self.ui.sectorName.text().strip()
|
||||
|
||||
def get_round_id(self) -> str:
|
||||
return cast(str, self.ui.roundComboBox.currentData())
|
||||
|
||||
def get_major_id(self) -> str:
|
||||
return cast(str, self.ui.majorComboBox.currentData())
|
||||
|
||||
def get_minor_id(self) -> str:
|
||||
return cast(str, self.ui.minorComboBox.currentData())
|
||||
|
||||
def get_influence_id(self) -> str:
|
||||
return cast(str, self.ui.influenceComboBox.currentData())
|
||||
|
||||
|
||||
class ChoicesDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
*,
|
||||
participants: List[ParticipantOption],
|
||||
default_participant_id: str | None = None,
|
||||
sectors: List[SectorDTO],
|
||||
default_priority_id: str | None = None,
|
||||
default_secondary_id: str | None = None,
|
||||
default_comment: str | None = None,
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_choicesDialog = Ui_choicesDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
for part in participants:
|
||||
self.ui.playerComboBox.addItem(part.name, part.id)
|
||||
select_if_exists(self.ui.playerComboBox, default_participant_id)
|
||||
self.ui.playerComboBox.setEnabled(False)
|
||||
self.ui.priorityComboBox.addItem("(none)", None)
|
||||
self.ui.secondaryComboBox.addItem("(none)", None)
|
||||
for sect in sectors:
|
||||
self.ui.priorityComboBox.addItem(sect.name, sect.id)
|
||||
self.ui.secondaryComboBox.addItem(sect.name, sect.id)
|
||||
select_if_exists(self.ui.priorityComboBox, default_priority_id)
|
||||
select_if_exists(self.ui.secondaryComboBox, default_secondary_id)
|
||||
self.ui.choiceComment.setPlainText(default_comment)
|
||||
|
||||
def get_participant_id(self) -> str:
|
||||
return cast(str, self.ui.playerComboBox.currentData())
|
||||
|
||||
def get_priority_id(self) -> str:
|
||||
return cast(str, self.ui.priorityComboBox.currentData())
|
||||
|
||||
def get_secondary_id(self) -> str:
|
||||
return cast(str, self.ui.secondaryComboBox.currentData())
|
||||
|
||||
def get_comment(self) -> str:
|
||||
return self.ui.choiceComment.toPlainText().strip()
|
||||
|
||||
|
||||
class BattlesDialog(QDialog):
|
||||
def __init__(
|
||||
self,
|
||||
parent: QWidget | None = None,
|
||||
*,
|
||||
sectors: List[SectorDTO],
|
||||
default_sector_id: str | None = None,
|
||||
players: List[ParticipantOption],
|
||||
default_player_1_id: str | None = None,
|
||||
default_player_2_id: str | None = None,
|
||||
default_winner_id: str | None = None,
|
||||
default_score: str | None = None,
|
||||
default_victory_condition: str | None = None,
|
||||
default_comment: str | None = None,
|
||||
) -> None:
|
||||
super().__init__(parent)
|
||||
self.ui: Ui_battleResultDialog = Ui_battleResultDialog()
|
||||
self.ui.setupUi(self) # type: ignore
|
||||
for sect in sectors:
|
||||
self.ui.sectorComboBox.addItem(sect.name, sect.id)
|
||||
select_if_exists(self.ui.sectorComboBox, default_sector_id)
|
||||
self.ui.sectorComboBox.setEnabled(False)
|
||||
|
||||
self.ui.player1ComboBox.addItem("(none)", None)
|
||||
self.ui.player2ComboBox.addItem("(none)", None)
|
||||
for play in players:
|
||||
self.ui.player1ComboBox.addItem(play.name, play.id)
|
||||
self.ui.player2ComboBox.addItem(play.name, play.id)
|
||||
select_if_exists(self.ui.player1ComboBox, default_player_1_id)
|
||||
select_if_exists(self.ui.player2ComboBox, default_player_2_id)
|
||||
self.ui.winnerComboBox.addItem("(none)", None)
|
||||
for play in players:
|
||||
if play.id in (default_player_1_id, default_player_2_id):
|
||||
self.ui.winnerComboBox.addItem(play.name, play.id)
|
||||
select_if_exists(self.ui.winnerComboBox, default_winner_id)
|
||||
self.ui.score.setText(default_score)
|
||||
self.ui.victoryCondition.setText(default_victory_condition)
|
||||
self.ui.battleComment.setPlainText(default_comment)
|
||||
|
||||
def get_sector_id(self) -> str:
|
||||
return cast(str, self.ui.sectorComboBox.currentData())
|
||||
|
||||
def get_player_1_id(self) -> str:
|
||||
return cast(str, self.ui.player1ComboBox.currentData())
|
||||
|
||||
def get_player_2_id(self) -> str:
|
||||
return cast(str, self.ui.player2ComboBox.currentData())
|
||||
|
||||
def get_winner_id(self) -> str:
|
||||
return cast(str, self.ui.winnerComboBox.currentData())
|
||||
|
||||
def get_score(self) -> str:
|
||||
return self.ui.score.text().strip()
|
||||
|
||||
def get_victory_condition(self) -> str:
|
||||
return self.ui.victoryCondition.text().strip()
|
||||
|
||||
def get_comment(self) -> str:
|
||||
return self.ui.battleComment.toPlainText().strip()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue