game.py

def init():
    return (1, ((0,0,0,0,0,0,0),
                (0,0,0,0,0,0,0),
                (0,0,0,0,0,0,0),
                (0,0,0,0,0,0,0),
                (0,0,0,0,0,0,0),
                (0,0,0,0,0,0,0),), (0,0) )

def small_init():
    return (1, ((3,3,3,0,0,0,0),
                (3,3,3,0,0,0,0),
                (3,3,3,0,0,0,0),
                (3,3,3,0,0,0,0),
                (3,3,3,3,3,3,3),
                (3,3,3,3,3,3,3),), (13,13) )

def coups_possibles(etat):
    grille = etat[1]
    res =[]
    for i in range(7):
        if grille[0][i] == 0 :
            res.append(i)
    return res

def est_gagnant(etat):
    grille = etat[1]

    #victoire en ligne
    for ligne in range(6):
        for j in range(4):
            if grille[ligne][j] == grille[ligne][j+1] == grille[ligne][j+2] == grille[ligne][j+3] and (grille[ligne][j] == 1 or grille[ligne][j] == 2):
                #print('victoire en ligne joueur ',grille[ligne][j])
                return grille[ligne][j]
            
    #victoire en colonne
    for colonne in range(7):
        for j in range(3):
            if grille[j][colonne] == grille[j+1][colonne] == grille[j+2][colonne] == grille[j+3][colonne] and (grille[j][colonne] == 1 or grille[j][colonne] == 2):
                #print('victoire en colonne joueur ',grille[j][colonne])
                return grille[j][colonne]
            
    #victoire en diagonal
    for ligne in range(3,6):
        for colonne in range(4):
            if grille[ligne][colonne] == grille[ligne-1][colonne+1] == grille[ligne-2][colonne+2] == grille[ligne-3][colonne+3] and (grille[ligne][colonne] ==1 or grille[ligne][colonne] ==2):
                return grille[ligne][colonne]
    
    #victoire contre-diago
    for ligne in range(3):
        for colonne in range(4):
            if grille[ligne][colonne] == grille[ligne+1][colonne+1] == grille[ligne+2][colonne+2] == grille[ligne+3][colonne+3] and (grille[ligne][colonne] ==1 or grille[ligne][colonne] ==2):
                return grille[ligne][colonne]

    #pas de victoire        
    return 0

def est_terminal(etat):
    return est_gagnant(etat) != 0 or coups_possibles(etat) == []

def jouer_coup(etat,coup):
    (joueur,grille,(nb1,nb2))=etat
    nv_joueur = 3 - joueur 
    def ligne_coup(grille,coup):
        for i in range(6):
            if grille[5-i][coup] == 0:
                return 5-i
    ligne = ligne_coup(grille,coup)
    nv_grille = tuple( 
        (
        tuple ((joueur if i == ligne and j == coup else grille[i][j] for j in range(7)))
        for i in range(6)
        )
    )
    if joueur == 1:
        nv_coups = (nb1+1,nb2)
    else:
        nv_coups = (nb1,nb2+1)
    return (nv_joueur, nv_grille,nv_coups)

def display(etat):
    grille = etat[1]
    remp = ['.','X','O','+']
    print("  0 1 2 3 4 5 6")
    for i in range(6):
        print(i, " ", sep = "", end = "")
        for j in range(7):
            print(remp[grille[i][j]], " ", sep = "", end = "")
        print()
    print()
    print(f'Player {etat[0]} ({remp[etat[0]]}) to play. {etat[2]}')
    print()

def partie_combi(combinaison):
    partie = init()
    for coup_str in combinaison:
        coup = int(coup_str)-1
        nv_partie = jouer_coup(partie,coup)
        partie = nv_partie
    return partie