Disallow partially specified generics.

game/unitdelivery.py

@@ -5,6 +5,8 @@ from collections import defaultdict
 from dataclasses import dataclass
 from typing import Optional, TYPE_CHECKING, Any
 
+from dcs.unittype import UnitType as DcsUnitType
+
 from game.theater import ControlPoint
 from .dcs.groundunittype import GroundUnitType
 from .dcs.unittype import UnitType
@@ -28,16 +30,16 @@ class PendingUnitDeliveries:
         self.destination = destination
 
         # Maps unit type to order quantity.
-        self.units: dict[UnitType, int] = defaultdict(int)
+        self.units: dict[UnitType[DcsUnitType], int] = defaultdict(int)
 
     def __str__(self) -> str:
         return f"Pending delivery to {self.destination}"
 
-    def order(self, units: dict[UnitType, int]) -> None:
+    def order(self, units: dict[UnitType[DcsUnitType], int]) -> None:
         for k, v in units.items():
             self.units[k] += v
 
-    def sell(self, units: dict[UnitType, int]) -> None:
+    def sell(self, units: dict[UnitType[DcsUnitType], int]) -> None:
         for k, v in units.items():
             self.units[k] -= v
 
@@ -46,27 +48,27 @@ class PendingUnitDeliveries:
         self.units = defaultdict(int)
 
     def refund_ground_units(self, game: Game) -> None:
-        ground_units: dict[UnitType[Any], int] = {
+        ground_units: dict[UnitType[DcsUnitType], int] = {
             u: self.units[u] for u in self.units.keys() if isinstance(u, GroundUnitType)
         }
         self.refund(game, ground_units)
         for gu in ground_units.keys():
             del self.units[gu]
 
-    def refund(self, game: Game, units: dict[UnitType, int]) -> None:
+    def refund(self, game: Game, units: dict[UnitType[DcsUnitType], int]) -> None:
         for unit_type, count in units.items():
             logging.info(f"Refunding {count} {unit_type} at {self.destination.name}")
             game.adjust_budget(
                 unit_type.price * count, player=self.destination.captured
             )
 
-    def pending_orders(self, unit_type: UnitType) -> int:
+    def pending_orders(self, unit_type: UnitType[DcsUnitType]) -> int:
         pending_units = self.units.get(unit_type)
         if pending_units is None:
             pending_units = 0
         return pending_units
 
-    def available_next_turn(self, unit_type: UnitType) -> int:
+    def available_next_turn(self, unit_type: UnitType[DcsUnitType]) -> int:
         current_units = self.destination.base.total_units_of_type(unit_type)
         return self.pending_orders(unit_type) + current_units
 
@@ -79,9 +81,9 @@ class PendingUnitDeliveries:
             )
             self.refund_ground_units(game)
 
-        bought_units: dict[UnitType, int] = {}
+        bought_units: dict[UnitType[DcsUnitType], int] = {}
         units_needing_transfer: dict[GroundUnitType, int] = {}
-        sold_units: dict[UnitType, int] = {}
+        sold_units: dict[UnitType[DcsUnitType], int] = {}
         for unit_type, count in self.units.items():
             coalition = "Ally" if self.destination.captured else "Enemy"
             d: dict[Any, int]