added Gazelle as CAS aircraft + minor fixes

This commit is contained in:
Vasyl Horbachenko
2018-10-13 23:38:26 +03:00
parent 270820de0b
commit 05c968edc2
7 changed files with 17 additions and 22 deletions

View File

@@ -1,5 +0,0 @@
from .aircraft import *
class HelicopterConflictGenerator(AircraftConflictGenerator):
pass

View File

@@ -16,13 +16,9 @@ class ShipGenerator:
self.m = mission
self.conflict = conflict
def generate_carrier(self, for_units: db.UnitsDict, country: str, at: Point) -> ShipGroup:
def generate_carrier(self, for_units: typing.Collection[UnitType], country: str, at: Point) -> ShipGroup:
type = db.find_unittype(Carriage, country)[0]
print(for_units)
for unit_type, unit_count in for_units.items():
if unit_count == 0:
continue
for unit_type in for_units:
if unit_type in db.CARRIER_TYPE_BY_PLANE:
type = db.CARRIER_TYPE_BY_PLANE[unit_type]
break