from click import Choice, prompt
from strategy import RandomStrategy
import random

class TTTHumanPlayer:
    "A human tic tac toe player."

    def __init__(self, name):
        "Sets up the player."
        self.name = name

    def choose_action(self, game):
        "Chooses an action by prompting the player for a choice."
        choices = Choice([str(i) for i in game.get_actions(game.state)])
        move = prompt("> ", type=choices, show_choices=False)
        return int(move)

class TTTComputerPlayer:
    "A computer tic tac toe player"

    def __init__(self, name):
        "Sets up the player."
        self.name = name

    def choose_action(self, game):
        "Chooses a random move from the moves available."
        strategy = RandomStrategy(game)
        action = strategy.choose_action(game.state)
        print(f"{self.name} chooses {action}.")
        return move

    def get_symbol(self, game):
        "Returns this player's symbol in the game."
        if game.players['X'] == self:
            return 'X'
        elif game.players['O'] == self:
            return 'O'
        else:
            raise ValueError(f"Player {self.name} isn't in this game!")