vikingsClasses.py

import random
import pygame

# Soldier


class Soldier:
    def __init__(self, health, strength):
        # your code here
        self.health = health
        self.strength = strength
    
    def attack(self):
        # your code here
        return self.strength

    def receiveDamage(self, damage):
        # your code here
        self.health -= damage
    

# Viking

class Viking(Soldier):
    def __init__(self, name, health, strength):
        # your code here
        super().__init__(health, strength)
        self.name = name

    def battleCry(self):
        # your code here
        return "Odin Owns You All!"

    def receiveDamage(self, damage):
        # your code here
        self.health -= damage

        if self.health > 0:
            return f"{self.name} has received {damage} points of damage"
        else:
            return f"{self.name} has died in act of combat"

# Saxon

class Saxon(Soldier):
    def __init__(self, health, strength):
        # your code here
        super().__init__(health, strength)

    def receiveDamage(self, damage):
        # your code here
        self.health -= damage

        if self.health > 0:
            return f"A Saxon has received {damage} points of damage"
        else:
            return f"A Saxon has died in combat"

# Davicente

class War():
    def __init__(self):
        # your code here
        self.vikingArmy = []
        self.saxonArmy = []
 
    def addViking(self, viking):
        # your code here
        self.vikingArmy.append(viking)
    
    def addSaxon(self, saxon):
        # your code here
        self.saxonArmy.append(saxon)
    
    def vikingAttack(self):
        # your code here

        return self.attackAction(self.vikingArmy, self.saxonArmy)

    
    def saxonAttack(self):
        #your code here

        return self.attackAction(self.saxonArmy,self.vikingArmy)


    def attackAction(self, attackingArmy, defendingArmy):

        if not self.vikingArmy or not self.saxonArmy:
            return "Battle is over"

        attacker = random.choice(attackingArmy)
        defender = random.choice(defendingArmy)

        result = defender.receiveDamage(attacker.strength)

        if defender.health <= 0:
            defendingArmy.remove(defender)
        return result

    def showStatus(self):
        # your code here
        if len(self.saxonArmy) <= 0:
            return "Vikings have won the war of the century!"

        elif len(self.vikingArmy) <= 0:
            return "Saxons have fought for their lives and survive another day..."
        
        elif len(self.saxonArmy) != 0 and len(self.vikingArmy) != 0:
            return "Vikings and Saxons are still in the thick of battle."
        


    pass