workshop_Atanas

class Player:
    def __init__(self, name, sign):
        self.name = name
        self.sign = sign


def read_player_name(helper_text):
    while True:
        player_name = input(helper_text).strip()
        if player_name:
            return player_name


def read_players():
    first_player_name = read_player_name('Player one name: ')
    while True:
        second_player_name = read_player_name(f'Player two name: ')
        if second_player_name != first_player_name:
            break

    while True:
        first_player_sign = input(f'{first_player_name} would you like to play with "X" or "O"? ').upper()
        if first_player_sign == 'X' or first_player_sign == 'O':
            break

    second_player_sign = 'O' if first_player_sign == 'X' else 'X'

    return Player(first_player_name, first_player_sign), Player(second_player_name, second_player_sign)


def print_board_numeration():
    print('This is the numeration of the board:')
    print('|  1  |  2  |  3  |')
    print('|  4  |  5  |  6  |')
    print('|  7  |  8  |  9  |')


def init_board():
    size = 3
    result = []
    for _ in range(size):
        result.append([None] * size)
    return result


def is_valid_position(board, board_mapper, position):
    if position < 1 or position > 9:
        return False
    row, col = board_mapper[position]
    return board[row][col] is None


def print_board(board):
    for row in board:
        print('|  ', end='')
        print('  |  '.join([x if x is not None else ' ' for x in row]), end='')
        print('  |')


def has_won(board, sign):
    # rows
    for row in board:
        # ['X', None, 'O']
        # [True, False, False]
        # ['X', 'X', 'X']
        # [True, True, True]
        if all([x == sign for x in row]):
            return True

    # cols
    for col in range(len(board)):
        col_win = True
        for row in range(len(board)):
            if board[row][col] != sign:
                col_win = False
                break
        if col_win:
            return True

    # diagonals
    left_diagonal_win = True
    right_diagonal_win = True
    for idx in range(len(board)):
        if board[idx][idx] != sign:
            left_diagonal_win = False
        if board[idx][len(board) - 1 - idx] != sign:
            right_diagonal_win = False
    return left_diagonal_win or right_diagonal_win


def is_draw(board):
    for row in board:
        if not all([x is not None for x in row]):
            return False
    return True


first_player, second_player = read_players()

print_board_numeration()

print(f'{first_player.name} starts first!')

board = init_board()
board_mapper = {
    1: [0, 0],
    2: [0, 1],
    3: [0, 2],
    4: [1, 0],
    5: [1, 1],
    6: [1, 2],
    7: [2, 0],
    8: [2, 1],
    9: [2, 2],
}

turn = 1

while True:
    current_player = first_player if turn % 2 != 0 else second_player
    position = int(input(f'{current_player.name} choose a free position [1-9]: '))

    if not is_valid_position(board, board_mapper, position):
        continue

    row, col = board_mapper[position]
    board[row][col] = current_player.sign

    print_board(board)

    if has_won(board, current_player.sign):
        print(f'{current_player.name} won!')
        break

    if is_draw(board):
        print('draw!')
        break

    turn += 1