一个Python编写的简单hangman游戏代码

一个Python编写的简单hangman游戏代码

#!/usr/bin/env python
import random 
import cPickle 
class Hangman(object):
    '''A simple hangman game that tries to improve your vocabulary a bit '''
    def __init__(self):
        # the variables used, this is not necessary
        self.dumpfile = ''       #the dictionary file
        self.dictionary = {}     #the pickled dict
        self.words = []          #list of words used
        self.secret_word = ''    #the 'key'
        self.length = 0          #length of the 'key'
        self.keys = []           #inputs that match the 'key'
        self.used_keys = []      #keys that are already used
        self.guess = ''          #player's guess
        self.mistakes = 0        #number of incorrect inputs
        
        return self.load_dict()
        
    #insert some random hints for the player
    def insert_random(self, length):
        randint = random.randint
        
        # 3 hints
        if length >= 7: hint = 3
        else: hint = 1
        for x in xrange(hint):
                a = randint(1, length - 1)
                self.keys[a-1] = self.secret_word[a-1]
    def test_input(self):
        #if the guessed letter matches
        if self.guess in self.secret_word:
            indexes = [i for i, item in enumerate(self.secret_word) if item == self.guess]
            for index in indexes:
                self.keys[index] = self.guess
                self.used_keys.append(self.guess)
                print "used letters ",set(self.used_keys),'\n'
       
        #if the guessed letter didn't match
        else:
            self.used_keys.append(self.guess)
            self.mistakes += 1
            print "used letters ",set(self.used_keys),'\n'
    
    # load the pickled word dictionary and unpickle them    
    def load_dict(self):
        try :
            self.dumpfile = open("~/python/hangman/wordsdict.pkl", "r")
        except IOError:
            print "Couldn't find the file 'wordsdict.pkl'"
            quit()
        self.dictionary = cPickle.load(self.dumpfile)
        self.words = self.dictionary.keys()
        self.dumpfile.close()
        return self.prepare_word()
    
    #randomly choose a word for the challenge
    def prepare_word(self):
        
        self.secret_word = random.choice(self.words)
        #don't count trailing spaces
        self.length = len(self.secret_word.rstrip())
        self.keys = ['_' for x in xrange(self.length)]
        self.insert_random(self.length)
        return self.ask()
    #display the challenge
    def ask(self):
        print ' '.join(self.keys), ":", self.dictionary[self.secret_word] 
        print 
        return self.input_loop()
    #take input from the player
    def input_loop(self):
        #four self.mistakes are allowed
        chances = len(set(self.secret_word)) + 4          
        while chances != 0 and self.mistakes < 5: try: self.guess = raw_input("> ")
                
            except EOFError:
                exit(1)
            self.test_input()
            print ' '.join(self.keys)
            if '_' not in self.keys:
                print 'well done!'
                break
            chances -= 1
       
        if self.mistakes > 4: print 'the word was', ''.join(self.secret_word).upper()
        return self.quit_message()
    
    def quit_message(self):
        print "\n"
        print "Press 'c' to continue, or any other key to quit the game. "
        print "You can always quit the game by pressing 'Ctrl+D'"
        try:
            command = raw_input('> ')
            if command == 'c': return self.__init__() #loopback
            else : exit(0)
        except EOFError: exit(1)
            
        
        
        
if __name__ == '__main__':
    game = Hangman()
    game.__init__()