New to code

# The game of infection
# I do not know what is "b3"

# example game command line:
# python -c 'import p;p.main(4)'

import random, pdb

class avatar:

    def __init__(self,name,team,offensive_skill,defense_rating):
        if not ((0 < offensive_skill < 1) and (0 < defense_rating < 1)):
            raise ValueError('offensive_skill and defense_rating must be between 0 and 1')
        self.name = name
        self.team = team
        self.offense = offensive_skill
        self.defense = defense_rating
        self.state = 'live'

    def __str__(self):
        return ' '.join(self.team,self.name,self.state)

    def __gt__(self,other):
        try:
            return random.random() < self.offense*(1-other.defense)
        except:
            raise ValueError('undefined operation with '+self.__class__.__name__)

def create(name,team):
    return avatar(name,
                  team,
                  offensive_skill = 0.2 + random.random()*0.2, # baseball like statistics
                  defense_rating = 0.7 + random.random()*0.2,
                  )

def shoot(red,blue):
    # shooting is not simultaneous.
    #   a dead player can be shot.
    #   a dead player cannot shoot.
    players = red+blue
    random.shuffle(players)
    #pdb.set_trace()
    change = 0
    for shooter in players:
        if 'live' != shooter.state:
            continue
        i = 'red' == shooter.team
        receivers = (red,blue,)[i]
        if not(len(receivers)):
            continue
        j = random.randrange(len(receivers))
        receiver = receivers[j]
        if receiver < shooter:
            change = 1
            if i:
                red.append(receiver)
                del blue[j]
            else:
                red[i].state = 'dead'
    return change

def simulation(players):
    red = [create(str(i),'red') for i in range(max(0,min(2,players)))]
    blue = [create(str(i+2),'blue') for i in range(max(0,players-2))]
    print('%d red, %d blue'%(sum('live' == r.state for r in red),len(blue)))
    while red and blue:
        if shoot(red,blue):
            print('%d red, %d blue'%(sum('live' == r.state for r in red),len(blue)))
            red = [r for r in red if 'live' == r.state]
    if red:
        print('The zombie infection overwhelmed the living')
    if blue:
        print('blue wins---zombie threat eliminated')

infection = main = simulation