Add transport aircraft to US factions.

game/db.py

@@ -9,9 +9,12 @@ from dcs.helicopters import (
     AH_1W,
     AH_64A,
     AH_64D,
+    CH_47D,
+    CH_53E,
     HelicopterType,
     Ka_50,
     Mi_24V,
+    Mi_26,
     Mi_28N,
     Mi_8MT,
     OH_58D,
@@ -43,6 +46,7 @@ from dcs.planes import (
     Bf_109K_4,
     C_101CC,
     C_130,
+    C_17A,
     E_3A,
     E_2C,
     FA_18C_hornet,
@@ -441,6 +445,10 @@ PRICES = {
     AH_64D: 30,
     OH_58D: 6,
     SH_60B: 6,
+    CH_47D: 4,
+    CH_53E: 4,
+    UH_60A: 4,
+    Mi_26: 4,
     # Bombers
     B_52H: 35,
     B_1B: 50,
@@ -464,6 +472,7 @@ PRICES = {
     E_2C: 50,
     C_130: 25,
     Hercules: 25,
+    C_17A: 20,
     # WW2
     P_51D_30_NA: 18,
     P_51D: 16,
@@ -729,42 +738,42 @@ Following tasks are present:
 """
 UNIT_BY_TASK = {
     CAP: [
+        A_4E_C,
+        Bf_109K_4,
+        C_101CC,
+        FA_18C_hornet,
+        FW_190A8,
+        FW_190D9,
+        F_14A_135_GR,
+        F_14B,
+        F_15C,
+        F_16A,
+        F_16C_50,
+        F_22A,
+        F_4E,
         F_5E_3,
-        Su_27,
-        Su_33,
-        Su_57,
+        I_16,
+        JF_17,
+        J_11A,
+        M_2000C,
         MiG_19P,
         MiG_21Bis,
         MiG_23MLD,
         MiG_25PD,
         MiG_29A,
+        MiG_29G,
         MiG_29S,
         MiG_31,
-        FA_18C_hornet,
-        F_15C,
-        F_22A,
-        F_14A_135_GR,
-        F_14B,
-        F_16A,
-        F_16C_50,
-        M_2000C,
         Mirage_2000_5,
-        P_51D_30_NA,
         P_51D,
-        MiG_29G,
-        Su_30,
-        J_11A,
-        JF_17,
-        F_4E,
-        C_101CC,
-        I_16,
-        Bf_109K_4,
-        FW_190D9,
-        FW_190A8,
-        SpitfireLFMkIXCW,
-        SpitfireLFMkIX,
-        A_4E_C,
+        P_51D_30_NA,
         SA342Mistral,
+        SpitfireLFMkIX,
+        SpitfireLFMkIXCW,
+        Su_27,
+        Su_30,
+        Su_33,
+        Su_57,
     ],
     CAS: [
         AH_1W,
@@ -782,6 +791,7 @@ UNIT_BY_TASK = {
         F_117A,
         F_15E,
         F_86F_Sabre,
+        Hercules,
         Ju_88A4,
         Ka_50,
         L_39ZA,
@@ -797,10 +807,11 @@ UNIT_BY_TASK = {
         P_47D_30bl1,
         P_47D_40,
         RQ_1A_Predator,
-        S_3B,
         SA342L,
         SA342M,
         SA342Minigun,
+        SH_60B,
+        S_3B,
         Su_17M4,
         Su_24M,
         Su_24MR,
@@ -813,19 +824,33 @@ UNIT_BY_TASK = {
         Tu_22M3,
         Tu_95MS,
         UH_1H,
-        SH_60B,
         WingLoong_I,
-        Hercules,
     ],
-    Transport: [IL_76MD, An_26B, An_30M, Yak_40, C_130],
+    Transport: [
+        An_26B,
+        An_30M,
+        CH_47D,
+        CH_53E,
+        C_130,
+        C_17A,
+        IL_76MD,
+        Mi_26,
+        UH_60A,
+        Yak_40,
+    ],
     Refueling: [
         IL_78M,
-        KC_135,
         KC130,
-        S_3B_Tanker,
         KC135MPRS,
+        KC_135,
+        S_3B_Tanker,
+    ],
+    AWACS: [
+        A_50,
+        E_2C,
+        E_3A,
+        KJ_2000,
     ],
-    AWACS: [E_3A, E_2C, A_50, KJ_2000],
     PinpointStrike: [
         Armor.APC_MTLB,
         Armor.APC_MTLB,

game/theater/base.py

@@ -4,7 +4,7 @@ import math
 import typing
 from typing import Dict, Type
 
-from dcs.task import AWACS, CAP, CAS, Embarking, PinpointStrike, Task
+from dcs.task import AWACS, CAP, CAS, Embarking, PinpointStrike, Task, Transport
 from dcs.unittype import FlyingType, UnitType, VehicleType
 from dcs.vehicles import AirDefence, Armor
 
@@ -152,6 +152,7 @@ class Base:
                 or for_task == CAS
                 or for_task == CAP
                 or for_task == Embarking
+                or for_task == Transport
             ):
                 target_dict = self.aircraft
             elif for_task == PinpointStrike: