def main(cursor):
battle_id = int(common.get_val("battle", -1))
unit_id = int(common.get_val("unit", -1))
army_id = int(common.get_val("army", -1))
amount = int(common.get_val("amount", 0))
if battle_id < 1:
raise Exception("No battle supplied")
if army_id < 1:
raise Exception("No army supplied")
if unit_id < 1:
raise Exception("No unit supplied")
if amount == 0:
return ""
raise Exception("No amount supplied")
if amount < 0:
return refund_losses(cursor, battle_id, unit_id, army_id, amount)
# Get all handles and instances
the_battle = battle_q.get_one_battle(cursor, battle_id)
battle_squads = the_battle.get_squads(cursor)
squad_dict = squad_q.get_squads_from_army_and_unit(cursor, army_id, unit_id)
squad_list = [s for s in battle_squads if s in squad_dict]
# return str(squad_list)
squad_f.apply_losses_to_squads(cursor, amount, squad_list, battle_id)
return ""
def main(cursor):
battle_id = int(common.get_val("battle", -1))
squad_id = int(common.get_val("squad", -1))
amount = int(common.get_val("amount", 0))
if battle_id < 1:
raise Exception("No battle supplied")
if squad_id < 1:
raise Exception("No team/unit supplied")
if amount == 0:
return ""
raise Exception("No amount supplied")
if amount < 0:
return refund_losses(cursor, battle_id, squad_id, -amount)
# return str(squad_list)
squad_f.apply_losses_to_squads(cursor, amount, [squad_id], battle_id)
return ""
