def apply_losses_to_squads(cursor, amount, squad_list, battle_id): """Send a list of squad IDs, losses will be applied to them""" squad_dict = squad_q.get_squads_from_list(cursor, squad_list) amount_left = amount if amount < 0: amount_per_squad = int(amount/float(len(squad_list))) amount_left_over = amount % len(squad_list) amount_per_squad = -amount_per_squad amount_left_over = -amount_left_over for s in squad_list: # print (amount_per_squad - amount_left_over), "<br />" battle_f.remove_loss(cursor, amount_per_squad - amount_left_over, s, battle_id) amount_left_over = 0 else: for s in squad_list: if amount_left <= 0: return 0 # If the squad has more in it than there is left to remove, remove all of it if squad_dict[s].amount > amount_left: battle_f.add_loss(cursor, amount_left, s, battle_id) amount_left = 0 else: amount_left -= squad_dict[s].amount battle_f.add_loss(cursor, squad_dict[s].amount, s, battle_id) return amount_left
def main(cursor, battle_id=-1):
battle_id = int(common.get_val('battle', battle_id))
ships = int(common.get_val('ships', True))
if battle_id < 1:
return "No battle selected"
the_battle = battle_q.get_one_battle(cursor, battle_id)
# Get some other stuff
team_dict = team_q.get_all_teams(cursor)
the_campaign = campaign_q.get_one_campaign(cursor, the_battle.campaign)
output = ["""
<div style='padding: 5px;'>
<a href="web.py?mode=list_campaigns&turn={turn}" class="block_link" style="display:inline; text-align:left;">Campaigns of this turn</a>
<a href="web.py?mode=setup_campaign&campaign={c.id}" class="block_link" style="display:inline;">Setup campaign</a>
<a href="web.py?mode=list_battles&campaign={c.id}" class="block_link" style="display:inline;">List battles</a>
<a href="web.py?mode=setup_battle&battle={b.id}" class="block_link" style="display:inline;">Setup battle</a>
<a href="web.py?mode=perform_by_army&battle={b.id}" class="block_link" style="display:inline;">By army</a>
<br /><br />
<span class="stitle">{c.name} - {b.name}</span> - <a href="web.py?mode=perform_battle&battle={b.id}&ships=0">No ships</a>
<br />
<a href="#" onclick="$('.show_evo_wedge').show(); $('.evo_wedge').hide(); return false;">Hide evo wedges</a>
-
<a href="#" onclick="$('.show_evo_wedge').hide(); $('.evo_wedge').show(); return false;">Show evo wedges</a>
""".format(
c = the_campaign,
b = the_battle,
turn = common.current_turn(),
)]
# Now for some stuff about the participants
#------------------------
the_campaign.get_sides_basic(cursor)
the_campaign.get_armies_basic(cursor)
the_battle.get_losses(cursor)
the_battle.get_squads(cursor)
team_list = []
for s in range(1, the_campaign.sides+1):
for t in the_campaign.sides_basic[s]:
team_list.append(t)
unit_dict = unit_q.get_units_from_team_list(cursor, team_list, special_units=True)
evolution_dict = evolution_q.get_all_evolutions(cursor)
# Sort squads by team
# Doing this makes it O(2n) rather than O(t*n)
squad_dict = squad_q.get_squads_from_list(cursor, the_battle.squads)
squads_by_team = {}
for squad_id, the_squad in squad_dict.items():
if the_squad.team not in squads_by_team: squads_by_team[the_squad.team] = []
squads_by_team[the_squad.team].append(squad_id)
for s in range(1, the_campaign.sides+1):
teams_on_side = the_campaign.sides_basic[s]
if len(teams_on_side) > 0:
output.append('''<table border="1" cellspacing="0" cellpadding="5" width="100%">
<tr>
<td class="stitle" colspan="2">{s}</td>
</tr>'''.format(s=s))
for i, team_id in enumerate(teams_on_side):
if team_id not in squads_by_team:
squads_by_team[team_id] = []
# Table tags for team cell
if i % 2 == 0: output.append('<tr><td>')
else: output.append('<td>')
the_team = team_dict[team_id]
# Team header
output.append('''
<span class="stitle">{name}</span>
-
<a href="web.py?mode=list_units&team={id}">Military</a>
-
<a href="web.py?mode=list_armies&team={id}">Armies</a>
-
<a href="#" onclick="$(this).parent().hide(); return false;">Hide</a>
'''.format(
name=the_team.name,
id=team_id,
))
# Build dictionaries/lists
team_units = the_team.get_units(cursor)
total_count = {}
unit_count = {}
unit_losses = {}
army_list = the_campaign.get_armies_from_team(cursor, team_id)
army_dict = army_q.get_armies_from_list(cursor, army_list)
squad_q.mass_get_squads(cursor, army_dict)
# armies = the_battle.get_armies()
# squads = the_battle.get_squads()
transport_lost = 0
transport_capacity = 0
army_size = 0
for unit_id in team_units.keys():
unit_count[unit_id] = 0
unit_losses[unit_id] = 0
for squad_id in squads_by_team[team_id]:
unit_count[squad_dict[squad_id].unit] += squad_dict[squad_id].amount
army_size += squad_dict[squad_id].amount
unit_losses[squad_dict[squad_id].unit] += the_battle.losses.get(squad_id, 0)
# print("")
# print(the_battle.squads, "<br />")
# print(unit_count, "<br />")
# print(unit_losses, "<br />")
# exit()
output.append("""
<table border="0" cellspacing="0" cellpadding="5" style="width: 100%">
<tr class="row2">
<th>Unit</th>
<th colspan="4"> </th>
<th>Equipment</th>
<th>Total</th>
<th>Amount</th>
<th colspan="2">Losses</th>
<th> </th>
</tr>
""")
# Build look-ahead
lookahead = {}
last_unit = -1
for unit_id in team_units.keys():
if unit_dict[unit_id].type_cat == unit.categories.index("Ship") and ships == False:
continue
if unit_count[unit_id] == 0 and unit_losses[unit_id] == 0:
continue
lookahead[last_unit] = unit_id
last_unit = unit_id
lookahead[last_unit] = 0
# Unit row
team_losses = 0
team_size = 0
count = -1
for unit_id, amount in team_units.items():
if unit_dict[unit_id].type_cat == unit.categories.index("Ship") and ships == False:
continue
if unit_count[unit_id] == 0 and unit_losses[unit_id] == 0:
continue
count += 1
team_losses += unit_losses[unit_id]
team_size += unit_count[unit_id]
transport_capacity += (unit_dict[unit_id].transport * unit_count[unit_id])
# Form JS
form_js = """$('#ajax_target').load('web.py', {mode: 'add_unit_loss', ajax: 'True', battle: %(battle_id)s, unit: %(unit_id)s, team: %(team_id)s, amount: $('#amount_for_%(unit_id)s_%(team_id)s').attr('value')}, function ()
{
var loss = parseInt($('#amount_for_%(unit_id)s_%(team_id)s').attr('value').replace(',', ''));
var current_loss = parseInt($('#losses_%(unit_id)s_%(team_id)s').html().replace(',', ''));
var current_amount = parseInt($('#amount_%(unit_id)s_%(team_id)s').html().replace(',', ''));
$('#amount_%(unit_id)s_%(team_id)s').html(current_amount - loss);
$('#losses_%(unit_id)s_%(team_id)s').html(current_loss + loss);
$('#amount_for_%(unit_id)s_%(team_id)s').attr('value', '');
$('#amount_for_%(ahead)s_%(team_id)s').focus();
});
return false;""" % {
"battle_id": battle_id,
"team_id": team_id,
"unit_id": unit_id,
"ahead": lookahead[unit_id],
};
# function add_unit_amount_%(unit_id)s ()
# {
# $('#amountFor_team').removeAttr('value');
# $('#spanAddUnitFor_team').load('ajax.php', {mode: 'getUnitsNotInWar', team: $team, war: $warId});
# });
output.append("""
<tr class="row{count}" id="row_{team_id}_{unit_id}">
<td style='font-weight:bold;'>{unit_name}</td>
<td>{weapon_cat}</td>
<td>{armour_cat}</td>
<td>{move_cat}</td>
<td>{training_cat}</td>
<td>{equipment}</td>
<td>{total}</td>
<td id="amount_{unit_id}_{team_id}">{amount}</td>
<td id="losses_{unit_id}_{team_id}">{losses}</td>
<td id="losses_cent_{unit_id}_{team_id}">{losses_cent}%</td>
<td style="padding:1px;">
<form action="exec.pyy" method="post" id="" onsubmit="{form_js}" accept-charset="utf-8">
<input type="hidden" name="mode" id="mode" value="add_unit_loss" />
<input type="hidden" name="battle" id="battle" value="{battle_id}" />
<input type="hidden" name="unit" id="unit" value="{unit_id}" />
<input type="hidden" name="team" id="team" value="{team_id}" />
<input type="text" onfocus="{on_focus}" onblur="{on_blur}" name="amount" id="amount_for_{unit_id}_{team_id}" value="" size="5"/>
</form>
</td>
</tr>
""".format(
count = count%2,
form_js = form_js,
unit_name = unit_dict[unit_id].name,
equipment = unit_dict[unit_id].equipment_string,
total = common.number_format(unit_count[unit_id] + unit_losses[unit_id]),
amount = common.number_format(unit_count[unit_id]),
losses = common.number_format(unit_losses[unit_id]),
losses_cent = round(unit_losses[unit_id]/(unit_count[unit_id]+unit_losses[unit_id])*100),
battle_id = battle_id,
team_id = team_id,
unit_id = unit_id,
on_focus = "$('#row_{0}_{1}').addClass('row3');".format(team_id, unit_id),
on_blur = "$('#row_{0}_{1}').removeClass('row3');".format(team_id, unit_id),
weapon_cat = "",
armour_cat = "",
move_cat = "",
training_cat = "",
# xweapon_cat = short_weapon_categories[unit_dict[unit_id].weapon_cat],
# xarmour_cat = short_armour_categories[unit_dict[unit_id].armour_cat],
# xmove_cat = short_move_categories[unit_dict[unit_id].move_cat],
# xtraining_cat = short_training_categories[unit_dict[unit_id].training_cat],
))
# Size row
if (team_size+team_losses) == 0:
losses_cent = ""
else:
losses_cent = "%d%%" % round(team_losses/(team_size+team_losses)*100)
count += 1
output.append("""
<tr class="row%(count)s">
<td colspan="6">
Military size:
</td>
<td>
%(size)s
</td>
<td>
%(losses)s
</td>
<td>
%(losses_cent)s
</td>
<td colspan="3"> </td>
</tr>
<tr class="row%(count_)s">
<td colspan="6">Transport:</td>
<td>%(transport_capacity)s</td>
<td colspan="5"> </td>
</tr>
""" % {
"count": count%2,
"count_": (count+1)%2,
"size": format(army_size, ','),
"losses": team_losses,
"losses_cent": losses_cent,
"transport_capacity": format(transport_capacity, ','),
})
count += 1
# Evos row
evo_output = []
the_team.get_evolutions(cursor)
for evolution_id in the_team.evolutions:
the_evo = evolution_dict[evolution_id]
if the_team.evolutions[evolution_id] == 0: continue
if not the_evo.combat_relevant: continue
if the_evo.max_level == 1 and the_team.evolutions[evolution_id] == 1:
evo_output.append("""<strong>%(evo_name)s</strong><br />""" % {
"evo_name": the_evo.name,
})
else:
evo_output.append("""<strong>%(level)sx %(evo_name)s</strong><br />""" % {
"level": the_team.evolutions[evolution_id],
"evo_name": the_evo.name,
})
count += 1
output.append("""
<tr class="row%(count)s">
<td colspan="11" style="padding:0px;">
<div id="show_evos_%(team_id)s" style="display:nnone;" class="show_evo_wedge">
<a href="#" class="block_link" onclick="$('#show_evos_%(team_id)s').hide(); $('#evos_%(team_id)s').fadeIn(250);return false;">Evos</a>
</div>
<div id="evos_%(team_id)s" style="display:none;" class="evo_wedge">
<a href="#" class="block_link" onclick="$('#show_evos_%(team_id)s').fadeIn(250); $('#evos_%(team_id)s').hide(); return false;">Hide</a><br />
%(evo_output)s
</div>
</td>
</tr>
""" % {
"count": count%2,
"team_id": team_id,
"evo_output": "".join(evo_output)
})
# End of team units
output.append("</table>")
# Table tags for team cell
if i % 2 == 0: output.append('</td>')
else: output.append('</td></tr>')
# Not sure that we need to adhere to W3C specs for tables...
# if i % 2 == 0: output.append('<td> </td></tr>')
output.append("</table><br />")
else:
output.append('No teams on side {s}'.format(s=s))
# Field for the ids of the armies listed that we need to sort out
# output.append('</table><br />')
output.append("</div>")
page_data['Title'] = "Perform by unit - %s" % the_battle.name
return "".join(output)
