[lab-data-vikings] Fernando Costa

vikingsClases.py

@@ -1,24 +1,105 @@
-
 # Soldier
-
-
 class Soldier:
-    pass
-
+    def __init__(self, health, strength):
+        self.health = health
+        self.strength = strength
+
+    def attack(self):
+        return self.strength
+
+    def receiveDamage(self, damage):
+        self.damage = damage
+        self.health = self.health - damage
+
 # Viking
+class Viking(Soldier):
+
+    def __init__(self, name, health, strength):
+        super().__init__(health, strength) 
+        self.name = name
 
+    def attack(self):
+        super().attack()
+        return self.strength
 
-class Viking:
-    pass
+    def receiveDamage(self, damage):
+        self.damage = damage
+        self.health = self.health - damage
+
+        if self.health > 0:
+            return (str(self.name)) + " has received " + (str(self.damage)) + " points of damage"
+        elif self.health <= 0:
+            return (str(self.name)) + " has died in act of combat"
+
+    def battleCry(self):
+        return "Odin Owns You All!"
 
 # Saxon
+class Saxon(Soldier):
 
+    def __init__(self, health, strength):
+        super().__init__(health, strength) #Tiene las mismas características que un soldado
+        #hay que incluir la parte de damage de soldier aquí
+
+    def attack(self):
+        super().attack()
+        return self.strength
 
-class Saxon:
-    pass
-
-# War
+    def receiveDamage(self, damage):
+        self.damage = damage
+        self.health = self.health - damage
 
+        if self.health > 0:
+            return "A Saxon has received " + (str(self.damage)) + " points of damage"
+        elif self.health <= 0:
+            return  "A Saxon has died in combat"
+
 
+# War
 class War:
-    pass
+
+    def __init__(self):
+        self.vikingArmy = []
+        self.saxonArmy = []
+
+    def addViking(self, Viking):
+        self.Viking = Viking
+        if len(self.vikingArmy) < 10:
+            self.vikingArmy.append(Viking)
+
+    def addSaxon(self, Saxon):
+        self.Saxon = Saxon
+        if len(self.saxonArmy) < 10:
+            self.saxonArmy.append(Saxon)
+
+    def vikingAttack(self):
+        import random
+        viking_attack = random.choice(self.vikingArmy)
+        random_saxon = random.choice(self.saxonArmy)
+
+        strength_viking = viking_attack.attack()
+        saxon_attacked = random_saxon.receiveDamage(strength_viking)
+
+        if saxon_attacked == "A Saxon has died in combat":
+            self.saxonArmy.remove(random_saxon)
+        return saxon_attacked
+
+    def saxonAttack(self):
+        import random
+        saxon_attack = random.choice(self.saxonArmy)
+        strength_saxon = saxon_attack.attack()
+
+        viking_attack = random.choice(self.vikingArmy)
+        viking_attacked = viking_attack.receiveDamage(strength_saxon)
+
+        if viking_attack.health <= 0:
+            self.vikingArmy.remove(viking_attack)
+        return viking_attacked
+
+    def showStatus(self):
+        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..."
+        else:
+            return "Vikings and Saxons are still in the thick of battle."