Анализ игры крестики-нолики

line = input('Enter cells: ')
print("---------")
for i in range(1, 8, 3):
    print("| {} {} {} |".format(*line[i:i + 3]))
print("---------")
list_x = []
list_o = []
s_x = s_o = 0
for index, elem in enumerate(line):
    if elem == 'X':
        s_x += 1
        list_x.append(index)
    elif elem == 'O':
        s_o += 1
        list_o.append(index)
list_win = [[1, 2, 3], [4, 5, 6], [7, 8, 9], [1, 4, 7], [3, 6, 9], [1, 5, 9], [3, 5, 7]]
if list_x in list_win:
    print('X wins')
elif list_o in list_win:
    print('O wins')
elif abs(s_x - s_o) >= 2:
    print('Impossible')
elif (list_o and list_x) not in list_win:
    print('Draw')
elif s_x + s_o != 9:
    print('Game not finished')

def check_win(c, gf):
    r1, r2, r3, r4 = False, False, True, True
    for i in range(3):
        x = True
        for j in range(3):
            x = x and gf[i][j] == c
        r1 = r1 or x
        x = True
        for j in range(3):
            x = x and gf[j][i] == c
        r2 = r2 or x
        r3 = r3 and gf[i][i] == c
        r4 = r4 and gf[2 - i][i] == c
    return r1 or r2 or r3 or r4
 
 
def check_impossible(gf):
    if check_win('O', gf) and check_win('X', gf):
        return True
    x, y = 0, 0
    for i in range(3):
        for j in range(3):
            if gf[i][j] == 'X':
                x += 1
            if gf[i][j] == 'O':
                y += 1
    if abs(x - y) > 1:
        return True
    return False
 
 
def check_draw(gf):
    for i in range(3):
        for j in range(3):
            if gf[i][j] == ' ':
                return False
    return True
 
 
cells = input()
gf = [[cells[i] for i in range(r * 3, (r + 1) * 3)] for r in range(3)]
if check_impossible(gf):
    print("Impossible")
elif check_win('X', gf):
    print("X wins")
elif check_win('O', gf):
    print("O wins")
elif check_draw(gf):
    print("Draw")
else:
    print("Game not finished")