Allow selection of auto-assigned mission types.

Fixes https://github.com/dcs-liberation/dcs_liberation/issues/1176

qt_ui/models.py

@@ -18,7 +18,7 @@ from game.squadrons import Squadron, Pilot
 from game.theater.missiontarget import MissionTarget
 from game.transfers import TransferOrder
 from gen.ato import AirTaskingOrder, Package
-from gen.flights.flight import Flight
+from gen.flights.flight import Flight, FlightType
 from gen.flights.traveltime import TotEstimator
 from qt_ui.uiconstants import AIRCRAFT_ICONS
 
@@ -467,6 +467,15 @@ class SquadronModel(QAbstractListModel):
             pilot.send_on_leave()
         self.endResetModel()
 
+    def is_auto_assignable(self, task: FlightType) -> bool:
+        return task in self.squadron.auto_assignable_mission_types
+
+    def set_auto_assignable(self, task: FlightType, auto_assignable: bool) -> None:
+        if auto_assignable:
+            self.squadron.auto_assignable_mission_types.add(task)
+        else:
+            self.squadron.auto_assignable_mission_types.remove(task)
+
 
 class GameModel:
     """A model for the Game object.

qt_ui/windows/SquadronDialog.py

@@ -1,4 +1,5 @@
 import logging
+from typing import Callable
 
 from PySide2.QtCore import (
     QItemSelectionModel,
@@ -13,9 +14,13 @@ from PySide2.QtWidgets import (
     QVBoxLayout,
     QPushButton,
     QHBoxLayout,
+    QGridLayout,
+    QLabel,
+    QCheckBox,
 )
 
 from game.squadrons import Pilot
+from gen.flights.flight import FlightType
 from qt_ui.delegates import TwoColumnRowDelegate
 from qt_ui.models import SquadronModel
 
@@ -61,6 +66,31 @@ class PilotList(QListView):
         self.setSelectionBehavior(QAbstractItemView.SelectItems)
 
 
+class AutoAssignedTaskControls(QVBoxLayout):
+    def __init__(self, squadron_model: SquadronModel) -> None:
+        super().__init__()
+        self.squadron_model = squadron_model
+
+        self.addWidget(QLabel("Auto-assignable mission types"))
+
+        def make_callback(toggled_task: FlightType) -> Callable[[bool], None]:
+            def callback(checked: bool) -> None:
+                self.on_toggled(toggled_task, checked)
+
+            return callback
+
+        for task in squadron_model.squadron.mission_types:
+            checkbox = QCheckBox(text=task.value)
+            checkbox.setChecked(squadron_model.is_auto_assignable(task))
+            checkbox.toggled.connect(make_callback(task))
+            self.addWidget(checkbox)
+
+        self.addStretch()
+
+    def on_toggled(self, task: FlightType, checked: bool) -> None:
+        self.squadron_model.set_auto_assignable(task, checked)
+
+
 class SquadronDialog(QDialog):
     """Dialog window showing a squadron."""
 
@@ -75,11 +105,17 @@ class SquadronDialog(QDialog):
         layout = QVBoxLayout()
         self.setLayout(layout)
 
+        columns = QHBoxLayout()
+        layout.addLayout(columns)
+
+        auto_assigned_tasks = AutoAssignedTaskControls(squadron_model)
+        columns.addLayout(auto_assigned_tasks)
+
         self.pilot_list = PilotList(squadron_model)
         self.pilot_list.selectionModel().selectionChanged.connect(
             self.on_selection_changed
         )
-        layout.addWidget(self.pilot_list)
+        columns.addWidget(self.pilot_list)
 
         button_panel = QHBoxLayout()
         button_panel.addStretch()