game loop; player budget; enemy progression; GUI WIP

shop/db.py

@@ -13,6 +13,7 @@ PRICES = {
     # planes
 
     Su_25T: 10,
+    Su_25: 10,
     A_10A: 15,
     A_10C: 20,
 
@@ -23,6 +24,10 @@ PRICES = {
 
     MiG_15bis: 10,
     MiG_21Bis: 13,
+    MiG_29A: 23,
+
+    IL_76MD: 20,
+    S_3B_Tanker: 20,
 
     # armor
 
@@ -40,5 +45,35 @@ PRICES = {
 UNIT_BY_TASK = {
     FighterSweep: [Su_27, Su_33, Su_25, F_15C, MiG_15bis, MiG_21Bis, MiG_29A, ],
     CAS: [Su_25T, A_10A, A_10C, ],
-    CAP: [Armor.MBT_T_90, Armor.MBT_T_80U, Armor.MBT_T_55, Armor.MBT_M1A2_Abrams, Armor.MBT_M60A3_Patton, Armor.ATGM_M1134_Stryker, Armor.APC_BTR_80, ]
+    CAP: [Armor.MBT_T_90, Armor.MBT_T_80U, Armor.MBT_T_55, Armor.MBT_M1A2_Abrams, Armor.MBT_M60A3_Patton, Armor.ATGM_M1134_Stryker, Armor.APC_BTR_80, ],
+    AirDefence: [AirDefence.AAA_ZU_23_on_Ural_375, ],
+    Transport: [IL_76MD, S_3B_Tanker, ],
 }
+
+UNIT_BY_COUNTRY = {
+    "Russia": [Su_25T, A_10C, Su_27, Su_33, Su_25, MiG_15bis, MiG_21Bis, MiG_29A, AirDefence.AAA_ZU_23_on_Ural_375, Armor.APC_BTR_80, Armor.MBT_T_90, Armor.MBT_T_80U, Armor.MBT_T_55, IL_76MD, ],
+    "USA": [F_15C, A_10C, Armor.MBT_M1A2_Abrams, Armor.MBT_M60A3_Patton, Armor.ATGM_M1134_Stryker, S_3B_Tanker],
+}
+
+
+def unit_task(unit: UnitType) -> Task:
+    for task, units in UNIT_BY_TASK.items():
+        if unit in units:
+            return task
+
+    assert False
+
+
+def find_unittype(for_task: Task, country_name: str) -> typing.List[UnitType]:
+    return [x for x in UNIT_BY_TASK[for_task] if x in UNIT_BY_COUNTRY[country_name]]
+
+
+def unit_type_name(unit_type) -> str:
+    return unit_type.id and unit_type.id or unit_type.name
+
+
+def task_name(task) -> str:
+    if task == AirDefence:
+        return "AirDefence"
+    else:
+        return task.name