ps4b.py

from ps4a import *
import time


#
#
# Problem #6: Computer chooses a word
#
#
def compChooseWord(hand, wordList, n):
    """
    Given a hand and a wordList, find the word that gives 
    the maximum value score, and return it.

    This word should be calculated by considering all the words
    in the wordList.

    If no words in the wordList can be made from the hand, return None.

    hand: dictionary (string -> int)
    wordList: list (string)
    n: integer (HAND_SIZE; i.e., hand size required for additional points)

    returns: string or None
    """
    # Create a new variable to store the maximum score seen so far (initially 0)
    maxScore = 0
    # Create a new variable to store the best word seen so far (initially None)  
    bestWord = None
    # For each word in the wordList
    for word in wordList:
        # If you can construct the word from your hand
        # (hint: you can use isValidWord, or - since you don't really need to test if the word is in the wordList - you can make a similar function that omits that test)
        if isValidWord(word, hand, wordList):
            # Find out how much making that word is worth
            currentScore = getWordScore(word, n)
            # If the score for that word is higher than your best score
            if currentScore > maxScore:
                maxScore = currentScore
                bestWord= word
                # Update your best score, and best word accordingly


    # return the best word you found.
    return bestWord

#
# Problem #7: Computer plays a hand
#
def compPlayHand(hand, wordList, n):
    """
    Allows the computer to play the given hand, following the same procedure
    as playHand, except instead of the user choosing a word, the computer 
    chooses it.

    1) The hand is displayed.
    2) The computer chooses a word.
    3) After every valid word: the word and the score for that word is 
    displayed, the remaining letters in the hand are displayed, and the 
    computer chooses another word.
    4)  The sum of the word scores is displayed when the hand finishes.
    5)  The hand finishes when the computer has exhausted its possible
    choices (i.e. compChooseWord returns None).
 
    hand: dictionary (string -> int)
    wordList: list (string)
    n: integer (HAND_SIZE; i.e., hand size required for additional points)
    """
    # Keep track of the total score
    totalScore = 0
    # As long as there are still letters left in the hand:

    while 1:
        # Display the hand
        if calculateHandlen(hand) != 0:
            print 'Current Hand: ',
            displayHand(hand)
            
        bestWord = compChooseWord(hand, wordList, n)
        if bestWord == None:
            break
        totalScore += getWordScore(bestWord, n)
        print '"'+str(bestWord)+'"'+ 'earned '+str(getWordScore(bestWord, n))+' points. Total: '+str(totalScore)+' points'
        print 
        hand = updateHand(hand, bestWord) 
        
    # Game is over, so tell user the total score
    print 'Total score: '+str(totalScore)+' points.'
    return None
    
            
#
# Problem #8: Playing a game
#
#
def playGame(wordList):
    """
    Allow the user to play an arbitrary number of hands.
 
    1) Asks the user to input 'n' or 'r' or 'e'.
        * If the user inputs 'e', immediately exit the game.
        * If the user inputs anything that's not 'n', 'r', or 'e', keep asking them again.

    2) Asks the user to input a 'u' or a 'c'.
        * If the user inputs anything that's not 'c' or 'u', keep asking them again.

    3) Switch functionality based on the above choices:
        * If the user inputted 'n', play a new (random) hand.
        * Else, if the user inputted 'r', play the last hand again.
      
        * If the user inputted 'u', let the user play the game
          with the selected hand, using playHand.
        * If the user inputted 'c', let the computer play the 
          game with the selected hand, using compPlayHand.

    4) After the computer or user has played the hand, repeat from step 1

    wordList: list (string)
    """
    lastHand_flag = False
    n = HAND_SIZE
    while 1:
        print 'Enter n to deal a new hand, r to replay the last hand, or e to end game: ',
        userInput = raw_input()
        print
        if userInput=='n':
            
            while 1:
                print 'Enter u to have yourself play, c to have the computer play: ',
                inputYourselfOrComputer = raw_input()
                print

                if inputYourselfOrComputer=='u':                    
                    hand = dealHand(n)
                    lastHand = hand.copy()
                    lastHand_flag = True
                    playHand(hand, wordList, n)
                    print
                    break
                elif inputYourselfOrComputer=='c':
                    hand = dealHand(n)
                    lastHand = hand.copy()
                    lastHand_flag = True
                    compPlayHand(hand, wordList, n)
                    print 
                    break
                else:
                    print 'Invalid command.'
                    print

        elif userInput=='r':
            if lastHand_flag==False:
                print 'You have not played a hand yet. Please play a new hand first!'
                continue
            else:
                while 1: 
                    print 'Enter u to have yourself play, c to have the computer play: ',
                    inputYouselfOrComputer = raw_input()
                    if inputYouselfOrComputer=='u':                    
                        playHand(lastHand, wordList, n)
                        print
                        break
                    elif inputYouselfOrComputer=='c':
                        compPlayHand(lastHand, wordList, n)
                        print 
                        break
                    else:
                        print 'Invalid command.'
                        print
                    
                    
        elif userInput=='e':
            break
        else:
            print 'Invalid command.'
    
    

        
#
# Build data structures used for entire session and play game
#
if __name__ == '__main__':
    wordList = loadWords()
    playGame(wordList)