Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
encapsulation solution in Clear category for The Lancers by Calen
class Warrior:
def __init__(self):
self.health = 50
self.attack = 5
self.is_alive = True
self.vampirism = 0
self.pending_dmg_to_next_unit = 0
self.defense = 0
def take_hit(self, hit, inflicter):
dmg = hit - self.defense if hit - self.defense > 0 else 0
self.health -= dmg
self.is_alive = True if self.health > 0 else False
if isinstance(inflicter, Lancer):
self.pending_dmg_to_next_unit = int(dmg / 2)
return dmg
def special_skill(self, dmg):
pass
def engage(self, opponent):
while True:
self.special_skill(opponent.take_hit(self.attack, self))
if not opponent.is_alive:
return True
opponent.special_skill(self.take_hit(opponent.attack, opponent))
if not self.is_alive:
return False
class Lancer(Warrior):
def __init__(self):
super().__init__()
self.attack = 6
class Vampire(Warrior):
def __init__(self):
super().__init__()
self.health = 40
self.attack = 4
self.vampirism = 50
def special_skill(self, dmg):
self.health += int(self.vampirism / 100 * dmg)
class Defender(Warrior):
def __init__(self):
super().__init__()
self.health = 60
self.attack = 3
self.defense = 2
class Knight(Warrior):
def __init__(self):
super().__init__()
self.attack = 7
def fight(unit_1, unit_2):
return unit_1.engage(unit_2)
class Army:
def __init__(self):
self.ranks = []
def add_units(self, unit, n: int):
for _ in range(n):
self.ranks.append(unit())
def frontline(self):
for i, v in enumerate(self.ranks):
if v.is_alive:
return v
return None
def inflict_pending_dmg(self):
for i in range(len(self.ranks)-1):
if self.ranks[i].pending_dmg_to_next_unit > 0:
self.ranks[i+1].health -= self.ranks[i].pending_dmg_to_next_unit
self.ranks[i].pending_dmg_to_next_unit = 0
def charge(self, enemy):
while True:
self.frontline().engage(enemy.frontline())
self.inflict_pending_dmg()
enemy.inflict_pending_dmg()
if not sum([x.is_alive for x in enemy.ranks]):
return True
if not sum([x.is_alive for x in self.ranks]):
return False
class Battle:
@staticmethod
def fight(offender: Army, defender: Army):
return offender.charge(defender)
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()
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
#battle tests
my_army = Army()
my_army.add_units(Defender, 2)
my_army.add_units(Vampire, 2)
my_army.add_units(Lancer, 4)
my_army.add_units(Warrior, 1)
enemy_army = Army()
enemy_army.add_units(Warrior, 2)
enemy_army.add_units(Lancer, 2)
enemy_army.add_units(Defender, 2)
enemy_army.add_units(Vampire, 3)
army_3 = Army()
army_3.add_units(Warrior, 1)
army_3.add_units(Lancer, 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(Lancer, 2)
battle = Battle()
assert battle.fight(my_army, enemy_army) == True
assert battle.fight(army_3, army_4) == False
print("Coding complete? Let's try tests!")
Feb. 22, 2021