mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
1d7f1082ea
Logging before we've made it to the logging setup was causing the root logger to be permanently configured to the default (warning) log level, so we weren't getting any info or debug logs any more. Defer the campaign data load until it is needed rather than doing it at import time. I've also cleaned up a bit so we only load each campaign once, rather than re-loading the campaign to create the theater again after the wizard is finished.
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
from PySide2 import QtGui
|
|
from PySide2.QtCore import QItemSelectionModel
|
|
from PySide2.QtGui import QStandardItem, QStandardItemModel
|
|
from PySide2.QtWidgets import QAbstractItemView, QListView
|
|
|
|
import qt_ui.uiconstants as CONST
|
|
from theater import ConflictTheater
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Campaign:
|
|
name: str
|
|
icon_name: str
|
|
theater: ConflictTheater
|
|
|
|
@classmethod
|
|
def from_json(cls, path: Path) -> Campaign:
|
|
with path.open() as campaign_file:
|
|
data = json.load(campaign_file)
|
|
|
|
sanitized_theater = data["theater"].replace(" ", "")
|
|
return cls(data["name"], f"Terrain_{sanitized_theater}",
|
|
ConflictTheater.from_json(data))
|
|
|
|
|
|
def load_campaigns() -> List[Campaign]:
|
|
campaign_dir = Path("resources\\campaigns")
|
|
campaigns = []
|
|
for path in campaign_dir.iterdir():
|
|
try:
|
|
logging.debug(f"Loading campaign from {path}...")
|
|
campaign = Campaign.from_json(path)
|
|
campaigns.append(campaign)
|
|
except RuntimeError:
|
|
logging.exception(f"Unable to load campaign from {path}")
|
|
|
|
return sorted(campaigns, key=lambda x: x.name)
|
|
|
|
|
|
class QCampaignItem(QStandardItem):
|
|
|
|
def __init__(self, campaign: Campaign) -> None:
|
|
super(QCampaignItem, self).__init__()
|
|
self.setIcon(QtGui.QIcon(CONST.ICONS[campaign.icon_name]))
|
|
self.setEditable(False)
|
|
self.setText(campaign.name)
|
|
|
|
|
|
class QCampaignList(QListView):
|
|
|
|
def __init__(self, campaigns: List[Campaign]) -> None:
|
|
super(QCampaignList, self).__init__()
|
|
self.model = QStandardItemModel(self)
|
|
self.setModel(self.model)
|
|
self.setMinimumWidth(250)
|
|
self.setMinimumHeight(350)
|
|
self.campaigns = []
|
|
self.setSelectionBehavior(QAbstractItemView.SelectItems)
|
|
self.setup_content(campaigns)
|
|
|
|
def setup_content(self, campaigns: List[Campaign]) -> None:
|
|
for campaign in campaigns:
|
|
self.campaigns.append(campaign)
|
|
item = QCampaignItem(campaign)
|
|
self.model.appendRow(item)
|
|
self.setSelectedCampaign(0)
|
|
self.repaint()
|
|
|
|
def setSelectedCampaign(self, row):
|
|
self.selectionModel().clearSelection()
|
|
index = self.model.index(row, 0)
|
|
if not index.isValid():
|
|
index = self.model.index(0, 0)
|
|
self.selectionModel().setCurrentIndex(index, QItemSelectionModel.Select)
|
|
self.repaint()
|
|
|
|
def clear_layout(self):
|
|
self.model.removeRows(0, self.model.rowCount()) |