From b750099b16ffadd200da75e519c18b597c5ba6d4 Mon Sep 17 00:00:00 2001
From: Raffson <Raffson@users.noreply.github.com>
Date: Thu, 18 May 2023 13:14:22 +0200
Subject: [PATCH] Add missing tasks in aircraft yamls

---
 game/migrator.py                        |  1 +
 resources/units/aircraft/A-7E.yaml      | 11 ++++++++++-
 resources/units/aircraft/EA-18G.yaml    | 15 +++++++++++++++
 resources/units/aircraft/F-16A MLU.yaml | 11 +++++++++++
 resources/units/aircraft/F-16A.yaml     |  1 +
 5 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/game/migrator.py b/game/migrator.py
index bb9da5ca..77153d0e 100644
--- a/game/migrator.py
+++ b/game/migrator.py
@@ -95,6 +95,7 @@ class Migrator:
     def _update_squadrons(self) -> None:
         for cp in self.game.theater.controlpoints:
             for s in cp.squadrons:
+                print(s.aircraft, s.aircraft.task_priorities)
                 preferred_task = max(
                     s.aircraft.task_priorities,
                     key=lambda x: s.aircraft.task_priorities[x],
diff --git a/resources/units/aircraft/A-7E.yaml b/resources/units/aircraft/A-7E.yaml
index b0117d3a..06c2ba56 100644
--- a/resources/units/aircraft/A-7E.yaml
+++ b/resources/units/aircraft/A-7E.yaml
@@ -13,4 +13,13 @@ price: 10
 role: Carrier-based Attack
 gunfighter: true
 variants:
-  A-7E Corsair II: {}
\ No newline at end of file
+  A-7E Corsair II: {}
+tasks:
+  BAI: 660
+  CAS: 660
+  DEAD: 390
+  OCA/Aircraft: 660
+  OCA/Runway: 360
+  SEAD: 110
+  SEAD Escort: 110
+  Strike: 360
diff --git a/resources/units/aircraft/EA-18G.yaml b/resources/units/aircraft/EA-18G.yaml
index bd23bba7..bd489827 100644
--- a/resources/units/aircraft/EA-18G.yaml
+++ b/resources/units/aircraft/EA-18G.yaml
@@ -36,3 +36,18 @@ utc_kneeboard: true
 #   HelmetMountedDevice: 1
 #   InnerBoard: 0
 #   OuterBoard: 0
+tasks:
+  Anti-ship: 150
+  BAI: 740
+  BARCAP: 450
+  CAS: 740
+  DEAD: 440
+  Escort: 450
+  Fighter sweep: 450
+  Intercept: 450
+  OCA/Aircraft: 740
+  OCA/Runway: 600
+  SEAD: 430
+  SEAD Escort: 430
+  Strike: 600
+  TARCAP: 450
diff --git a/resources/units/aircraft/F-16A MLU.yaml b/resources/units/aircraft/F-16A MLU.yaml
index f476550a..06f894de 100644
--- a/resources/units/aircraft/F-16A MLU.yaml	
+++ b/resources/units/aircraft/F-16A MLU.yaml	
@@ -4,3 +4,14 @@ price: 18
 max_range: 350
 variants:
   F-16A MLU: null
+tasks:
+  Anti-ship: 150
+  BARCAP: 420
+  CAS: 340
+  DEAD: 210
+  Escort: 420
+  Fighter sweep: 420
+  Intercept: 420
+  OCA/Runway: 570
+  Strike: 570
+  TARCAP: 420
diff --git a/resources/units/aircraft/F-16A.yaml b/resources/units/aircraft/F-16A.yaml
index 919c01b1..3ee749c7 100644
--- a/resources/units/aircraft/F-16A.yaml
+++ b/resources/units/aircraft/F-16A.yaml
@@ -5,6 +5,7 @@ variants:
   F-16A: null
 tasks:
   BARCAP: 420
+  CAS: 340
   Escort: 420
   Fighter sweep: 420
   Intercept: 420