Задача по ООП

from random import randrange, randint
 
class Person:
    def __init__(self, c):
        self.id = randint(0, 100)
        self.command = c
 
class Heroes(Person):
    level = 1
    def level_up():
        level = level + 1
        print('герой с id ' +str(self.id) +' достиг второго уровня')
 
class Soldiers(Person):
    def __init__(self):
        Person.__init__(self, c = 1)
        random_number = randrange(-1, 2, 2)
        if random_number > 0:
            self.command = 'blue'
            print(self)
        elif random_number < 0:
            self.command = 'red'
            print(self)
 
hero_blue = Heroes('blue')
hero_red = Heroes('red')

from random import randrange, randint
 
class Person:
    def __init__(self, c):
        self.id = randint(0, 100)
        self.command = c
 
class Heroes(Person):
    level = 1
    def level_up():
        level = level + 1
        print('герой с id ' +str(self.id) +' достиг второго уровня')
 
 
class Soldiers(Person): #
    def __init__(self,r = [], b = [] ):
        Person.__init__(self, c = 1)
        random_number = randrange(-1, 2, 2)
        red_army = r
        blue_army = b
 
        if random_number > 0:
            self.command = 'blue'
            blue_army.append(self)
        elif random_number < 0:
            self.command = 'red'
            red_army.append(self)
 
        print('blue army soldiers: {0} units, red army soldiers: {1} units'.format(len(blue_army), len(red_army)))
 
        if len(blue_army) > len(red_army):
            print(1)
        elif len(blue_army) < len(red_army):
            print(2)
 
 
    def __str__(self):
        return 'test'
 
hero_blue = Heroes('blue')
hero_red = Heroes('red')
soldier1 = Soldiers()
soldier2 = Soldiers()
soldier3 = Soldiers()
soldier4 = Soldiers()
soldier5 = Soldiers()
soldier6 = Soldiers()
soldier7 = Soldiers()
soldier8 = Soldiers()
soldier9 = Soldiers()

soldiers = []
for e in range(300):
    soldiers.append(Soldier())

import random
 
class Soldiers():
    id_ = 1
    def __init__(self):      
        self.id_ = Soldiers.id_ 
        Soldiers.id_ += 1
 
    def go_hero(self,hero):
        print(f'солдат с id {self.id_} следует за героем {hero.id_}')
        
    def __str__(self):
        return f'{self.id_}'
    
class Heroes(Soldiers):
    
    def __init__(self, team):
        Soldiers.__init__(self)
        self.team = team
        self.level = 0
 
    def level_up(self):
        self.level += 1
        print(f'герой с id {self.id_} достиг {self.level} уровня')
    
heroes = Heroes('red')
heroes_2 = Heroes('blue')
list_heroes = []
list_heroes_2 = []
for _ in range(50):
    if random.choice(['red', 'blue']) == 'red':
        list_heroes.append(Soldiers())
    else:
        list_heroes_2.append(Soldiers())
        
len_heroes = len( list_heroes)
len_heroes_2 = len( list_heroes_2) 
print(f'солдат у героя heroes - {len_heroes}')
print(f'солдат у героя heroes_2 - {len_heroes_2}')
if len_heroes > len_heroes_2:
    heroes.level_up()
else:
    heroes_2.level_up()
 
random.choice(list_heroes).go_hero(heroes)