import random,os
from names import createWarriorTeam,createWeapons,Warrior,Weapons
warriors = createWarriorTeam(os.path.join(os.getcwd(),'WARRIORS.json'))
weapons = createWeapons(os.path.join(os.getcwd(),'WEAPONS.json'))
class Battlefield():
def __init__(self,warriors,weapons):
self.warriors = warriors
self.weapons = weapons
def _getweapons(self):
self.weapons = [Weapons(x,self.weapons[x]) for x in self.weapons]
def _getcomands(self):
self.teams = {}
self.teamscount = {}
for warr in self.warriors:
warr = warr.upper()
self.team = [Warrior(x,warr,self.weapons) for x in self.warriors[warr]]
self.teams[warr] = self.team
self.teamscount[warr] = len(self.team)
print('-' * 100)
for tc in self.teamscount:
print('{0}:{1}'.format(tc,self.teamscount[tc]),end=' ')
print()
print('-'*100)
def _fight(self):
while True:
liveTeams = 0
lteams = list(self.teams)
random.shuffle(lteams)
for lt in lteams:
warrior = random.choice(self.teams[lt])
otherTeams = [x for x in lteams if x!=lt]
enemyTeam = random.choice(otherTeams)
enemywarrior = random.choice(self.teams[enemyTeam])
if enemywarrior.is_alive:
warrior.hit(enemywarrior)
for ltlive in lteams:
if (([False]*self.teamscount[ltlive])==[x.is_alive for x in self.teams[ltlive]]):
liveTeams+=1
if liveTeams==len(self.teams)-1:
break
self._end()
self._averageDamage()
def _end(self):
for x in self.teams:
if not all([not x.is_alive for x in self.teams[x]]):
winner = x
break
print('Победила команда {0}!'.format(winner))
print('-'*100)
for t in self.teams:
for w in self.teams[t]:
print('Имя:{0} Команда:{1} Здоровье:{2} Броня:{3} Оружие:{4}({5})'.format(w.name, w.team, w.health,
w.armor, w.weapon.name,w.weapon.damage))
def _averageDamage(self):
print('-' * 100)
warriors =[]
for t in self.teams:
warriors+=self.teams[t]
for x in warriors:
print('Оружие:{0} Средний урон:{1}({2})'.format(x.weapon.name,
x.weapon.getAverageDamage(),x.weapon.damage))
def start(self):
self._getweapons()
self._getcomands()
self._fight()
a = Battlefield(warriors,weapons)
a.start()