From b38d271f10b9dcf5c38faaa5038e33e382969ea4 Mon Sep 17 00:00:00 2001
From: Dan Albert <dan@gingerhq.net>
Date: Wed, 26 May 2021 19:47:53 -0700
Subject: [PATCH] Show player/AI status for pilots.

https://github.com/dcs-liberation/dcs_liberation/issues/276
---
 qt_ui/windows/SquadronDialog.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/qt_ui/windows/SquadronDialog.py b/qt_ui/windows/SquadronDialog.py
index 9e620ee2..4b4408aa 100644
--- a/qt_ui/windows/SquadronDialog.py
+++ b/qt_ui/windows/SquadronDialog.py
@@ -25,14 +25,17 @@ class PilotDelegate(TwoColumnRowDelegate):
         return index.data(SquadronModel.PilotRole)
 
     def text_for(self, index: QModelIndex, row: int, column: int) -> str:
+        pilot = self.pilot(index)
         if (row, column) == (0, 0):
             return self.squadron_model.data(index, Qt.DisplayRole)
         elif (row, column) == (0, 1):
-            flown = self.pilot(index).record.missions_flown
+            flown = pilot.record.missions_flown
             missions = "missions" if flown != 1 else "mission"
             return f"{flown} {missions} flown"
         elif (row, column) == (1, 0):
-            return "Alive" if self.pilot(index).alive else "Dead"
+            return "Player" if pilot.player else "AI"
+        elif (row, column) == (1, 1):
+            return "Alive" if pilot.alive else "Dead"
         return ""