Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for The Healers by tom-tom
from contextlib import suppress
class Warrior:
health = 50
attack = 5
def hit_by(self, enemy_attack):
self.health -= enemy_attack
return enemy_attack
def strike(self, enemy):
enemy.hit_by(self.attack)
@property
def is_alive(self):
return self.health > 0
@property
def max_health(self):
return type(self).health
class Knight(Warrior):
attack = 7
class Defender(Warrior):
health = 60
attack = 3
defense = 2
def hit_by(self, enemy_attack):
damage = max(enemy_attack - self.defense, 0)
self.health -= damage
return damage
class Vampire(Warrior):
health = 40
attack = 4
vampirism = 0.5
def strike(self, enemy):
damage = enemy.hit_by(self.attack)
self.health += int(damage * self.vampirism)
class Lancer(Warrior):
health = 50
attack = 6
pierce = 0.5
def strike(self, enemy):
damage = enemy.hit_by(self.attack)
with suppress(AttributeError):
enemy.next_unit.hit_by(int(damage * self.pierce))
class Healer(Warrior):
health = 60
attack = 0
healing = 2
def heal(self, ally):
ally.health = min(ally.health + self.healing, ally.max_health)
def fight(unit_1, unit_2):
attacker, defender = unit_1, unit_2
while True:
attacker.strike(defender)
with suppress(AttributeError):
attacker.next_unit.heal(attacker)
if not defender.is_alive:
return attacker is unit_1
attacker, defender = defender, attacker
class Army(list):
def add_units(self, unit_cls, n):
self.extend(unit_cls() for _ in range(n))
def first_alive(self):
alive = (x for x in self if x.is_alive)
try:
unit = next(alive)
unit.next_unit = next(alive, None)
return unit
except StopIteration:
return None
class Battle:
def fight(self, army1, army2):
while True:
unit1 = army1.first_alive()
if unit1 is None:
return False
unit2 = army2.first_alive()
if unit2 is None:
return True
fight(unit1, unit2)
if __name__ == '__main__':
#These "asserts" using only for self-checking and not necessary for auto-testing
#fight tests
chuck = Warrior()
bruce = Warrior()
carl = Knight()
dave = Warrior()
mark = Warrior()
bob = Defender()
mike = Knight()
rog = Warrior()
lancelot = Defender()
eric = Vampire()
adam = Vampire()
richard = Defender()
ogre = Warrior()
freelancer = Lancer()
vampire = Vampire()
priest = Healer()
assert fight(chuck, bruce) == True
assert fight(dave, carl) == False
assert chuck.is_alive == True
assert bruce.is_alive == False
assert carl.is_alive == True
assert dave.is_alive == False
assert fight(carl, mark) == False
assert carl.is_alive == False
assert fight(bob, mike) == False
assert fight(lancelot, rog) == True
assert fight(eric, richard) == False
assert fight(ogre, adam) == True
assert fight(freelancer, vampire) == True
assert freelancer.is_alive == True
assert freelancer.health == 14
priest.heal(freelancer)
assert freelancer.health == 16
#battle tests
my_army = Army()
my_army.add_units(Defender, 2)
my_army.add_units(Healer, 1)
my_army.add_units(Vampire, 2)
my_army.add_units(Lancer, 2)
my_army.add_units(Healer, 1)
my_army.add_units(Warrior, 1)
enemy_army = Army()
enemy_army.add_units(Warrior, 2)
enemy_army.add_units(Lancer, 4)
enemy_army.add_units(Healer, 1)
enemy_army.add_units(Defender, 2)
enemy_army.add_units(Vampire, 3)
enemy_army.add_units(Healer, 1)
army_3 = Army()
army_3.add_units(Warrior, 1)
army_3.add_units(Lancer, 1)
army_3.add_units(Healer, 1)
army_3.add_units(Defender, 2)
army_4 = Army()
army_4.add_units(Vampire, 3)
army_4.add_units(Warrior, 1)
army_4.add_units(Healer, 1)
army_4.add_units(Lancer, 2)
battle = Battle()
assert battle.fight(my_army, enemy_army) == False
assert battle.fight(army_3, army_4) == True
print("Coding complete? Let's try tests!")
June 28, 2019
