split controller
This commit is contained in:
Maxime Réaux
parent
701f6b3292
commit
7792a76f8e
11 changed files with 1212 additions and 987 deletions
246
src/warchron/controller/app_controller.py
Normal file
246
src/warchron/controller/app_controller.py
Normal file
|
|
@ -0,0 +1,246 @@
|
|||
from pathlib import Path
|
||||
|
||||
from PyQt6.QtWidgets import QMessageBox
|
||||
|
||||
from warchron.model.model import Model
|
||||
from warchron.model.exception import (
|
||||
DeletionForbidden,
|
||||
DeletionRequiresConfirmation,
|
||||
UpdateRequiresConfirmation,
|
||||
)
|
||||
from warchron.view.view import View
|
||||
from warchron.constants import ItemType, RefreshScope
|
||||
from warchron.controller.navigation_controller import NavigationController
|
||||
from warchron.controller.player_controller import PlayerController
|
||||
from warchron.controller.war_controller import WarController
|
||||
from warchron.controller.campaign_controller import CampaignController
|
||||
from warchron.controller.round_controller import RoundController
|
||||
|
||||
|
||||
class AppController:
|
||||
def __init__(self, model: Model, view: View) -> None:
|
||||
self.model: Model = model
|
||||
self.view: View = view
|
||||
self.navigation = NavigationController(self)
|
||||
self.players = PlayerController(self)
|
||||
self.wars = WarController(self)
|
||||
self.campaigns = CampaignController(self)
|
||||
self.rounds = RoundController(self)
|
||||
self.current_file: Path | None = None
|
||||
self.view.on_close_callback = self.on_app_close
|
||||
self.is_dirty: bool = False
|
||||
self.__connect()
|
||||
self.navigation.refresh_players_view()
|
||||
self.navigation.refresh_wars_view()
|
||||
self.update_window_title()
|
||||
self.view.on_tree_selection_changed = self.navigation.on_tree_selection_changed
|
||||
self.view.on_add_campaign = self.campaigns.add_campaign
|
||||
self.view.on_add_round = self.rounds.add_round
|
||||
|
||||
def __connect(self) -> None:
|
||||
self.view.actionExit.triggered.connect(self.view.close)
|
||||
self.view.actionNew.triggered.connect(self.new)
|
||||
self.view.actionOpen.triggered.connect(self.open_file)
|
||||
self.view.actionSave.triggered.connect(self.save)
|
||||
self.view.actionSave_as.triggered.connect(self.save_as)
|
||||
self.view.addPlayerBtn.clicked.connect(self.players.add_player)
|
||||
self.view.addWarBtn.clicked.connect(self.wars.add_war)
|
||||
self.view.addObjectiveBtn.clicked.connect(self.wars.add_objective)
|
||||
self.view.addWarParticipantBtn.clicked.connect(self.wars.add_war_participant)
|
||||
self.view.addSectorBtn.clicked.connect(self.campaigns.add_sector)
|
||||
self.view.addCampaignParticipantBtn.clicked.connect(
|
||||
self.campaigns.add_campaign_participant
|
||||
)
|
||||
self.view.on_edit_item = self.edit_item
|
||||
self.view.on_delete_item = self.delete_item
|
||||
|
||||
def on_app_close(self) -> bool:
|
||||
if self.is_dirty:
|
||||
reply = QMessageBox.question(
|
||||
self.view,
|
||||
"Unsaved changes",
|
||||
"You have unsaved changes. Do you want to save before quitting?",
|
||||
QMessageBox.StandardButton.Yes
|
||||
| QMessageBox.StandardButton.No
|
||||
| QMessageBox.StandardButton.Cancel,
|
||||
)
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
self.save()
|
||||
elif reply == QMessageBox.StandardButton.Cancel:
|
||||
return False
|
||||
return True
|
||||
|
||||
# Menu bar methods
|
||||
|
||||
def new(self) -> None:
|
||||
if self.is_dirty:
|
||||
reply = QMessageBox.question(
|
||||
self.view,
|
||||
"Unsaved changes",
|
||||
"Discard current campaign?",
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
self.model.new()
|
||||
self.current_file = None
|
||||
self.is_dirty = False
|
||||
self.navigation.refresh_players_view()
|
||||
self.navigation.refresh_wars_view()
|
||||
self.update_window_title()
|
||||
|
||||
def open_file(self) -> None:
|
||||
if self.is_dirty:
|
||||
reply = QMessageBox.question(
|
||||
self.view,
|
||||
"Unsaved changes",
|
||||
"Discard current campaign?",
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
path = self.view.ask_open_file()
|
||||
if not path:
|
||||
return
|
||||
self.model.load(path)
|
||||
self.current_file = path
|
||||
self.is_dirty = False
|
||||
self.navigation.refresh_players_view()
|
||||
self.navigation.refresh_wars_view()
|
||||
self.update_window_title()
|
||||
|
||||
def save(self) -> None:
|
||||
if not self.current_file:
|
||||
self.save_as()
|
||||
return
|
||||
self.model.save(self.current_file)
|
||||
self.is_dirty = False
|
||||
self.update_window_title()
|
||||
|
||||
def save_as(self) -> None:
|
||||
path = self.view.ask_save_file()
|
||||
if not path:
|
||||
return
|
||||
self.current_file = path
|
||||
self.model.save(path)
|
||||
self.is_dirty = False
|
||||
self.update_window_title()
|
||||
|
||||
# Display methods
|
||||
|
||||
def update_window_title(self) -> None:
|
||||
base = "WarChron"
|
||||
if self.current_file:
|
||||
base += f" - {self.current_file.name}"
|
||||
else:
|
||||
base += " - New file"
|
||||
if self.is_dirty:
|
||||
base = base + " *"
|
||||
self.view.setWindowTitle(base)
|
||||
|
||||
# Command methods
|
||||
|
||||
def edit_item(self, item_type: str, item_id: str) -> None:
|
||||
try:
|
||||
if item_type == ItemType.PLAYER:
|
||||
self.players.edit_player(item_id)
|
||||
self.navigation.refresh(RefreshScope.PLAYERS_LIST)
|
||||
elif item_type == ItemType.WAR:
|
||||
self.wars.edit_war(item_id)
|
||||
self.navigation.refresh_and_select(
|
||||
RefreshScope.WARS_TREE, item_type=ItemType.WAR, item_id=item_id
|
||||
)
|
||||
elif item_type == ItemType.CAMPAIGN:
|
||||
self.campaigns.edit_campaign(item_id)
|
||||
self.navigation.refresh_and_select(
|
||||
RefreshScope.WARS_TREE, item_type=ItemType.CAMPAIGN, item_id=item_id
|
||||
)
|
||||
elif item_type == ItemType.OBJECTIVE:
|
||||
self.wars.edit_objective(item_id)
|
||||
self.navigation.refresh(RefreshScope.WAR_DETAILS)
|
||||
elif item_type == ItemType.WAR_PARTICIPANT:
|
||||
self.wars.edit_war_participant(item_id)
|
||||
self.navigation.refresh(RefreshScope.WAR_DETAILS)
|
||||
elif item_type == ItemType.SECTOR:
|
||||
self.campaigns.edit_sector(item_id)
|
||||
self.navigation.refresh(RefreshScope.CAMPAIGN_DETAILS)
|
||||
elif item_type == ItemType.CAMPAIGN_PARTICIPANT:
|
||||
self.campaigns.edit_campaign_participant(item_id)
|
||||
self.navigation.refresh(RefreshScope.CAMPAIGN_DETAILS)
|
||||
elif item_type == ItemType.CHOICE:
|
||||
self.rounds.edit_round_choice(item_id)
|
||||
self.navigation.refresh(RefreshScope.ROUND_DETAILS)
|
||||
elif item_type == ItemType.BATTLE:
|
||||
self.rounds.edit_round_battle(item_id)
|
||||
self.navigation.refresh(RefreshScope.ROUND_DETAILS)
|
||||
self.is_dirty = True
|
||||
except UpdateRequiresConfirmation as e:
|
||||
reply = QMessageBox.question(
|
||||
self.view,
|
||||
"Confirm update",
|
||||
e.message,
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
e.apply_update()
|
||||
self.navigation.refresh(RefreshScope.CAMPAIGN_DETAILS)
|
||||
|
||||
def delete_item(self, item_type: str, item_id: str) -> None:
|
||||
reply = QMessageBox.question(
|
||||
self.view,
|
||||
"Confirm deletion",
|
||||
"Are you sure you want to delete this item?",
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply != QMessageBox.StandardButton.Yes:
|
||||
return
|
||||
try:
|
||||
if item_type == ItemType.PLAYER:
|
||||
self.model.remove_player(item_id)
|
||||
self.navigation.refresh(RefreshScope.PLAYERS_LIST)
|
||||
elif item_type == ItemType.WAR:
|
||||
self.model.remove_war(item_id)
|
||||
self.navigation.refresh(RefreshScope.WARS_TREE)
|
||||
elif item_type == ItemType.CAMPAIGN:
|
||||
war = self.model.get_war_by_campaign(item_id)
|
||||
war_id = war.id
|
||||
self.model.remove_campaign(item_id)
|
||||
self.navigation.refresh_and_select(
|
||||
RefreshScope.WARS_TREE, item_type=ItemType.WAR, item_id=war_id
|
||||
)
|
||||
elif item_type == ItemType.OBJECTIVE:
|
||||
self.model.remove_objective(item_id)
|
||||
self.navigation.refresh(RefreshScope.WAR_DETAILS)
|
||||
elif item_type == ItemType.WAR_PARTICIPANT:
|
||||
self.model.remove_war_participant(item_id)
|
||||
self.navigation.refresh(RefreshScope.WAR_DETAILS)
|
||||
elif item_type == ItemType.SECTOR:
|
||||
self.model.remove_sector(item_id)
|
||||
self.navigation.refresh(RefreshScope.CAMPAIGN_DETAILS)
|
||||
elif item_type == ItemType.CAMPAIGN_PARTICIPANT:
|
||||
self.model.remove_campaign_participant(item_id)
|
||||
self.navigation.refresh(RefreshScope.CAMPAIGN_DETAILS)
|
||||
elif item_type == ItemType.ROUND:
|
||||
camp = self.model.get_campaign_by_round(item_id)
|
||||
camp_id = camp.id
|
||||
self.model.remove_round(item_id)
|
||||
self.navigation.refresh_and_select(
|
||||
RefreshScope.WARS_TREE, item_type=ItemType.CAMPAIGN, item_id=camp_id
|
||||
)
|
||||
self.is_dirty = True
|
||||
except DeletionForbidden as e:
|
||||
QMessageBox.warning(
|
||||
self.view,
|
||||
"Deletion forbidden",
|
||||
e.reason,
|
||||
)
|
||||
except DeletionRequiresConfirmation as e:
|
||||
reply = QMessageBox.question(
|
||||
self.view,
|
||||
"Confirm deletion",
|
||||
e.message,
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
||||
)
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
e.cleanup_action()
|
||||
self.navigation.refresh(RefreshScope.CAMPAIGN_DETAILS)
|
||||
Loading…
Add table
Add a link
Reference in a new issue