def _validateVehiclesInfo(vInfos):
if g_currentVehicle.isPresent():
vehicle = g_currentVehicle.item
vInfos = (unit_items.VehicleInfo(vehicle.invID, vehicle.intCD, vehicle.level),)
elif vInfos is None:
vInfos = (unit_items.VehicleInfo(),)
return vInfos
def validateVehicles(self, vInfo, flags):
if g_currentVehicle.isPresent():
vehicle = g_currentVehicle.item
vInfo = unit_items.VehicleInfo(vehicle.invID, vehicle.intCD,
vehicle.level)
elif vInfo is None:
vInfo = unit_items.VehicleInfo()
return super(SquadActionValidator, self).validateVehicles(vInfo, flags)
def _getVehiclesInfo(self):
vInfos = super(SquadVehiclesValidator, self)._getVehiclesInfo()
if not findFirst(lambda v: not v.isEmpty(), vInfos, False):
if g_currentVehicle.isPresent():
vehicle = g_currentVehicle.item
vInfos = (unit_items.VehicleInfo(vehicle.invID, vehicle.intCD,
vehicle.level), )
return vInfos
def _getVehiclesInfo(self):
vInfos = super(SquadVehiclesValidator, self)._getVehiclesInfo()
if not findFirst(lambda v: not v.isEmpty(), vInfos, False):
if g_currentVehicle.isPresent():
vehicle = g_currentVehicle.item
vehClassIdx = VEHICLE_CLASS_INDICES[vehicle.type]
vInfos = (unit_items.VehicleInfo(vehicle.invID, vehicle.intCD,
vehicle.level, vehClassIdx), )
return vInfos
def getVehiclesInfo(self, dbID=None, unitIdx=None):
return (unit_items.VehicleInfo(), )
