demo_bot.py

import random


class SpyBot(object):
    """
    This abstract class represents a bot that acts as a spy master for Codenames and can either be asked to
    give a clue to the guesser or be updated about moves made in the game.
    """

    def __init__(self, vocab, game_board, p_id):
        """
        Initialization for the bot
        Args:
            vocab (List<string>): list of English words that can be used to generate clues. No other clue words allowed!
            game_board (Map<string, Enum:WordType>): a dictionary representing the game board, mapping the words on
                the board with an enum representing whether the word belongs to player 1, player 2, is neutral or a spy word.
                See game_platform.py.
            p_id: (int): integer (0, 1) representing the the id of the bot, player 1 or 2.
        """
        pass

    def update(self, is_my_turn, clue_word, clue_num_guesses, guesses):
        """
        Informs the bot with what happened during a turn.
        Args:
            is_my_turn (bool): True if the turn was the bots or the opponent.
            clue_word (str): String representing the clue word in the round.
            clue_num_guesses (int): int representing the number of words that matched the given clue word.
            guesses: (List<string>): list of words that were guessed in the round.
        """
        pass

    def getClue(self, invalid_words):
        """
        Gives a clue for the turn.
        Args:
            invalid_words (Set<string>): set of words from the original vocab that are not allowed to be clued for this
            turn. Eg. words that share a root with any of the face-up words on the board.
        Returns:
            a tuple (clue, num_guesses) where `clue` is a string clue from the given vocab and but not in the set of
            invalid words, and `num_guesses` is the number of words on the board related to the clue.
        """
        pass


# example of how to write a bot
class RandomBot(SpyBot):

    def __init__(self, vocab, game_board, p_id):
        self.vocab = set(vocab)

    def getClue(self, invalid_words):
        return random.choice(list(self.vocab.difference(invalid_words))), 2


"""TODO[1]: implement your bot here that inherits from SpyBot"""