Untitled

import random

class Soldier:
    def __init__(self, name, strength):
        self.name = name
        self.strength = strength

    def attack(self):
        return random.randint(1, self.strength)

class Team:
    def __init__(self, name):
        self.name = name
        self.soldiers = []

    def add_soldier(self, soldier):
        self.soldiers.append(soldier)

    def total_strength(self):
        return sum(soldier.strength for soldier in self.soldiers)

    def attack(self):
        return sum(soldier.attack() for soldier in self.soldiers)

def battle(team1, team2):
    print(f"Battle starts between {team1.name} and {team2.name}!")
    team1_power = team1.attack()
    team2_power = team2.attack()
    
    print(f"{team1.name} attacks with a power of {team1_power}.")
    print(f"{team2.name} attacks with a power of {team2_power}.")
    
    if team1_power > team2_power:
        print(f"{team1.name} wins!")
    elif team2_power > team1_power:
        print(f"{team2.name} wins!")
    else:
        print("It's a draw!")

# Create teams
team_a = Team("Team A")
team_b = Team("Team B")

# Add soldiers to Team A
team_a.add_soldier(Soldier("Soldier A1", strength=10))
team_a.add_soldier(Soldier("Soldier A2", strength=12))
team_a.add_soldier(Soldier("Soldier A3", strength=8))

# Add soldiers to Team B
team_b.add_soldier(Soldier("Soldier B1", strength=11))
team_b.add_soldier(Soldier("Soldier B2", strength=9))
team_b.add_soldier(Soldier("Soldier B3", strength=10))

# Simulate battle
battle(team_a, team_b)