from random import *

nb_batonnets = 20
joueur = 0
IA = False

s = ''
while s.lower() != 'oui' and s.lower() != 'non': # lower permet de mettre la chaione de charactère en minuscule 
    s = input('Voulez vous jouer à 1 joueur contre l\'ordinateur ? Oui - Non')
if s.lower() == 'oui':
   IA = True
   
while nb_batonnets != 0:
    affichage = ''
    if joueur == 0:
        affichage = 'Au joueur 1 de jouer : Enlever de 1 à 3 batonnet(s).'
    else:
        affichage = 'Au joueur 2 de jouer : Enlever de 1 à 3 batonnet(s).'
    
    coup = 0
    while (joueur==0 or (joueur==1 and not IA)) and coup != 1 and coup != 2 and coup != 3:
        coup = input(affichage)
        if coup.isnumeric():    # Vérification que ma chaine de caractère est convertible en nombre pour éviter un plantage
            coup = int(coup)
            if coup > nb_batonnets and (coup==1 or coup==2 or coup==3):
                coup = 0
                print('Vous ne pouvez pas prendre autant de batonnets')
    
    if IA:
        if nb_batonnets > 3:
            coup = randint(1, 3)
        else:
            coup = randint(1, nb_batonnets)
    
    if IA:
        print('L\'ordinateur a joué et il prend', coup, 'batonnets.')
    nb_batonnets = nb_batonnets - coup
    print('Il reste', nb_batonnets, 'batonnet(s).')
    
    if nb_batonnets != 0:
        joueur = 1 - joueur
if IA and joueur==1:
    print('L\'ordinateur a gagné!')
else:    
    print('Le joueur', joueur+1, 'a gagné!')