from random import randint
from battle import Battle
from models.army import Army
if __name__ == '__main__':
armies = []
try:
calc_army = int(
input('Enter how many armies are playing (min = 2): \n'))
if calc_army < 2:
print('You entered less then 2 armies')
else:
for i in range(calc_army):
calc_squad = int(
input('Enter how many squads for Army_{0} (min = 1):\n'.
format(i)))
army = Army(randint(2, calc_squad), i)
army.create_army()
army.damage_rank()
armies.append(army)
Game = Battle(armies)
Game.battle_armies()
print(Game)
except ValueError:
print('Wrong input, you must input only number')
