def army_selection(self):
player_game = PlayerGame.get_by_name('Me')
army_list = player_game.armies
if army_list != None and len(army_list) > 0:
return_value = {'existing': True, 'choices': [{'id': army.id, 'name': army.name} for army in army_list]}
else:
return_value = {'existing': False}
return return_value
def do_army_creation(self):
army_position = ArmyPosition.get_by_id(1)
army = Army(army_position)
army.name = self.request.POST['army_name']
player_game = PlayerGame.get_by_name('Me')
if player_game.armies == None:
player_game.armies = [army]
else:
player_game.armies.append(army)
army.save(explicit=True)
url = self.request.route_url('army_edition', id=army.id)
return HTTPFound(location=url)
