nim.py

def main():
    print("Bem-vindo ao jogo do NIM! Escolha:")
    print("1 - para jogar uma partida isolada")
    print("2 - para jogar um campeonato")
    o = input()

    if o == "1":
        print("Você escolheu partida isolada!")
        partida()
    elif o == "2":
        print("Você escolheu um campeonato!")
        campeonato()
    else:
        print("Por favor selecione uma das opções.")
        main()

def partida():
    n = int(input("Quantas peças? "))
    m = int(input("Limite de peças por jogada? "))

    computador = True
    if n % (m+1) == 0:
        print("Você começa!")        
        computador = False
    else:
        print("Computador começa!")

    while (n > 0):
        if computador:
            x = computador_escolhe_jogada(n, m)
            computador = False
            print("O computador tirou",x,"peças.")
        else:
            x = usuario_escolhe_jogada(n, m)
            computador = True
            print("Você tirou", x, "peças.")
        
        n = n - x
        if n > 0:
            print("Restam apenas", n,"peças em jogo.")

    if computador:
        print("Fim do jogo! Você ganhou!")
        return True
    else:
        print("Fim do jogo! O computador ganhou!")
        return False

def campeonato():
    usuario = 0
    computador = 0
    partidas = 3

    for x in range(1, partidas+1):
        print("**** Rodada",x,"****")
        vencedor = partida()

        if vencedor:
            usuario = usuario + 1
        else:
            computador = computador + 1
    
    print("Placar final: Você", usuario,"x",computador,"Computador")

def computador_escolhe_jogada(n, m):
    if n <= m:
        return n
    else:
        x = n % (m+1)
        if (x > 0):
            return x
        else:
            return m

def usuario_escolhe_jogada(n, m):
    p = 0
    while(p == 0):
        p = int(input("Quantas peças você vai tirar? "))
        if p <= 0 or p > n or p > m:
            p = 0

    return p

main()