Untitled

import random


class Rytir:
    
    def __init__(self, jmeno, HP, ATK, DEF, zbran):
        self.jmeno = jmeno
        self.HP = HP
        self.ATK = ATK
        self.DEF = DEF
        self.zbran = zbran
    
    def zautoc(self, protivnik):
        return self.ATK + self.zbran.ATK - protivnik.DEF
    
    def __str__(self):
        return f"Jmeno: {self.jmeno}, HP: {self.HP}"


class Zbran:
    
    def __init__(self, jmeno, ATK):
        self.jmeno = jmeno
        self.ATK = ATK


class Turnaj:
    
    def __init__(self):
        self.seznam_rytiru = []
    
    def registrace(self, rytir):
        self.seznam_rytiru.append(rytir)
    
    def duel(self):
        r1 = random.choice(self.seznam_rytiru)
        r2 = random.choice(self.seznam_rytiru)
        while r1.HP >= 0 and r2.HP >= 0:
            r2.HP -= r1.zautoc(r2)
            r1.HP -= r2.zautoc(r1)
        if r1.HP == 0:
            self.seznam_rytiru.remove(r1)
        else:
            self.seznam_rytiru.remove(r2)

turnaj = Turnaj()
dragon_slayer = Zbran("Dragonslayer", 50)
excalibur = Zbran("Excalibur", 30)
guts = Rytir("Guts", HP=200, ATK=30, DEF=10, zbran=dragon_slayer)
griffith = Rytir("Griffith", HP=180, ATK=25, DEF=25, zbran=excalibur)
turnaj.registrace(guts)
turnaj.registrace(griffith)
turnaj.duel()
print(turnaj.seznam_rytiru[0])