Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Uncategorized category for Straight Fight by Orthomonalus
class Warrior:
"""Basic warrior class"""
def __init__(self):
self.health = 50
self.attack = 5
self.defence = 0
@property
def is_alive(self):
return self.health > 0
def take_damage(self, damage, defence):
self.health = self.health - (damage - defence) if damage >= defence else self.health
class Knight(Warrior):
"""Knight is an improved warrior with better attack"""
def __init__(self):
super().__init__()
self.attack = 7
class Defender(Warrior):
"""Defender is an improved warrior with better health, additional defence but lower attack"""
def __init__(self):
super().__init__()
self.health = 60
self.attack = 3
self.defence = 2
class Vampire(Warrior):
"""Vampire is a specific warrior with ability to restore his health points attacking opponent"""
def __init__(self):
super().__init__()
self.health = 40
self.attack = 4
self.vampirism = 50
class Lancer(Warrior):
"""Lancer is a specific warrior with ability to give a damage to opponent and the next opponent"""
def __init__(self):
super().__init__()
self.attack = 6
class Healer(Warrior):
"""Healer is a support unit with ability to heal friendly unit if it stays behind friendly unit during its attack"""
def __init__(self):
super().__init__()
self.attack = 0
self.health = 60
self.healing_power = 2
def heal(self, unit):
"""Heals friendly unit but not more that its max health"""
if unit.is_alive:
unit.health = min(unit.__class__().health, unit.health + self.healing_power)
class Army:
"""Army that holds multiple warriors"""
def __init__(self):
self.units: list = []
def add_units(self, unit_class, quantity):
self.units.extend([unit_class() for _ in range(quantity)])
@property
def has_units(self):
return bool(self.units)
def get_next_unit(self):
"""Return the next alive unit or None if no units are left"""
return self.units[0] if self.has_units else None
def check_units(self):
self.units = [unit for unit in self.units if unit.is_alive]
class Battle:
"""Handles battles between two armies."""
def fight(self, army_1, army_2):
"""Fight mode when army fights with an enemy army"""
while army_1.has_units and army_2.has_units:
unit_1 = army_1.get_next_unit()
unit_2 = army_2.get_next_unit()
unit_after_2 = army_2.units[1] if len(army_2.units) > 1 else None
unit_after_1 = army_1.units[1] if len(army_1.units) > 1 else None
fight(unit_1, unit_2, unit_after_2, unit_after_1)
army_1.check_units()
army_2.check_units()
return army_1.has_units
def straight_fight(self, army_1, army_2):
"""Fight mode when every unit in army fights with opponent unit one by one"""
while army_1.has_units and army_2.has_units:
min_length = min(len(army_1.units), len(army_2.units))
for i in range(min_length):
fight(army_1.units[i], army_2.units[i])
army_1.check_units()
army_2.check_units()
return army_1.has_units
def fight(unit_1, unit_2, unit_after_2=None, unit_after_1=None):
"""Handles battles between two units"""
while unit_1.is_alive and unit_2.is_alive:
# Unit 1 attacks Unit 2
damage = max(0, unit_1.attack - unit_2.defence)
unit_2.take_damage(unit_1.attack, unit_2.defence)
# Vampire restores health
if isinstance(unit_1, Vampire):
heal = damage * unit_1.vampirism // 100
unit_1.health = min(unit_1.__class__().health, unit_1.health + heal) # Limit to max health
# Lancer's second attack
if isinstance(unit_1, Lancer) and unit_after_2 and unit_after_2.is_alive:
secondary_damage = damage // 2
unit_after_2.take_damage(secondary_damage, unit_after_2.defence)
# Healer heals unit_1 after attack
if isinstance(unit_after_1, Healer):
unit_after_1.heal(unit_1)
if unit_2.is_alive:
# Unit 2 attacks Unit 1
damage = max(0, unit_2.attack - unit_1.defence)
unit_1.take_damage(unit_2.attack, unit_1.defence)
# Vampire restores health
if isinstance(unit_2, Vampire):
heal = damage * unit_2.vampirism // 100
unit_2.health = min(unit_2.__class__().health, unit_2.health + heal) # Limit to max health
# Lancer's second attack
if isinstance(unit_2, Lancer) and unit_after_1 and unit_after_1.is_alive:
secondary_damage = damage // 2
unit_after_1.take_damage(secondary_damage, unit_after_1.defence)
# Healer heals unit_2 after attack
if isinstance(unit_after_2, Healer):
unit_after_2.heal(unit_2)
return unit_1.is_alive
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()
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)
battle = Battle()
print(battle.fight(my_army, enemy_army)) # False
print(priest.heal(freelancer))
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)
army_5 = Army()
army_5.add_units(Warrior, 10)
army_6 = Army()
army_6.add_units(Warrior, 6)
army_6.add_units(Lancer, 5)
battle = Battle()
assert battle.fight(my_army, enemy_army) == False
assert battle.fight(army_3, army_4) == True
assert battle.straight_fight(army_5, army_6) == False
print("Coding complete? Let's try tests!")
Feb. 18, 2025
