filter words

Hangman-bot.py

@@ -3,25 +3,62 @@
 import operator
 import os
 
-words = open('words.txt')
+# Start Functions
+
 def clear():
+    '''
+    Clear console
+    '''
     os.system('cls' if os.name=='nt' else 'clear')
 
+
+def print_words(pos=False):
+    '''
+    Print words resolves
+    Arguments:
+    pos - View numbers
+    '''
+    print_pos = ' '
+    print_pos_bar = ' '
+    print_pos_letters = ' '
+    for i in range(num_letters):
+        if pos:
+            print_pos = print_pos + str(i + 1) + ' '
+            print_pos_bar = print_pos_bar + '| '
+        print_pos_letters = print_pos_letters + word_resolve[i] + ' '
+    print(print_pos)
+    print(print_pos_bar)
+    print(print_pos_letters)
+    print('')
+    print('')
+
 def best_letter(word_resolve):
+    '''
+    Search best letter
+    Arguments:
+    word_resolve - list with word resolve
+    '''
     # Read all words and count letters
     letters = dict()
     ignore = ('\xb1', '\xc3', '\xb6', '.', '\xae')
+    words = open('words.txt')
     # Read alls words for dictionary
     for word in words.readlines():
         # Just consider which have the same length
-        if len(word_resolve) == len(word):
+        if len(word_resolve) == len(word.strip()):
             l_word = list(word.strip().lower())
+            # Check word_resolve is same letters
+            fit = True
+            for i in range(len(l_word)):
+                if word_resolve[i] != l_word[i] and word_resolve[i] != '_':
+                    fit = False
             # Count the letters
+            if fit:
                 for letter in l_word:
                     if letter.lower() not in ignore:
                         if letter in letters:
                             letters[letter] = letters[letter] + 1
-                    else:
+                        elif not letter in letters_used:
                             letters[letter] = 1
     # Sort
     sorted_letters = sorted(
@@ -31,6 +68,9 @@ def best_letter(word_resolve):
     # Best letter 
     return sorted_letters[0][0]
 
+# End Functions
+
+# Start 
 clear()
 print('Hangman Bot 1.0v')
 print(' _________     ')
@@ -44,6 +84,7 @@ print('|              ')
 #  Get num letters and make list resolve
 num_letters = input('Number of letters: ')
 clear()
+play = True
 word_resolve = list()
 letters_used = list()
 for pos in range(num_letters):
@@ -52,13 +93,14 @@ for pos in range(num_letters):
 print('')
 print('Okay, come on!')
 print('')
-print('')
+
 # Logic
-while True:
+while play:
     # Get best letter
     best_option = best_letter(word_resolve)
     # The guard not to give it back
     letters_used.append(best_option)
+    print_words()
     # Print best letter
     print('Test with the letter> {letter}'.format(
         letter=best_option.upper()
@@ -68,20 +110,22 @@ while True:
     question_success = raw_input('I successful? (yes o no): ').lower()
     clear()
     if question_success == 'no':
+        clear()
         print('')
         print('Ups!')
-        print('')
-        print('')
     elif question_success == 'yes':
-        print_pos = ' '
-        print_pos_bar = ' '
-        print_pos_letters = ' '
-        for i in range(num_letters):
-            print_pos = print_pos + str(i + 1) + ' '
-            print_pos_bar = print_pos_bar + '| '
-            print_pos_letters = print_pos_letters + word_resolve[i] + ' '
-        print(print_pos)
-        print(print_pos_bar)
-        print(print_pos_letters)
         print('')
-        good_pos = raw_input('Tell me that positions (Example> 2 4 7): ')
+        print_words(True)
+        good_pos = raw_input('Tell me that positions (Example> 2 4 7): ').split(' ')
+        clear()
+        for pos in good_pos:
+            word_resolve[int(pos) - 1] = best_option
+
+    # Game over
+    end = True
+    for letter in word_resolve:
+        if letter != '_':
+            end = False
+    if end:
+        play = False
+        print('Game over :)')