Wordle_Solver.py

# Class to solve wordles
class WordleBot:
    # Initialize a wordle bot with a list of 5 letter words and a known letter dictionary:
    # Additionally, add a ranking system for each letter in the alphabet:
    def __init__(self, letters = 5):
        # Constant number of letters in a word
        WordleBot.letters = letters

        # List of english letters & followup dictionary to rank those letters
        self.letterList = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z']
        self.letterRank = {}

        # List of usable 5 letter words
        self.wordList = []
        # List of letters that do not show
        self.blackLetters = []
        # List of letters who's position is not known
        self.yellowLetters = []
        # Dictionary of letters who's position has been correctly identified
        self.greenLetters = {}

        # Access the words document
        wordsAddress = open("Words.txt")

        # For each line in the word document:
        for line in wordsAddress:
            # If the word is 5 letters long excluding the \n:
            if len(line) == letters + 1:
                # Append it to the word list, removing the \n
                self.wordList.append(line[:-1])

        print("Wordle Solver Initialized")
        print("Ranking Letters...\n")
        # For each letter in the letter list:
        for letter in self.letterList:
            # Instantiate a count for how many times that letter is used and total letters
            rankLetterCount = 0
            allLetterCount = 0
            # For each word in the word list:
            for word in self.wordList:
                # For each letter in the word:
                for wLet in word:
                    # If the letter is equal to the current ranked letter; add to the counter
                    if letter == wLet:
                        rankLetterCount += 1
                    # Count the letter
                    allLetterCount += 1

            # Assign that letter to the dictionary with it's percentage ranking
            self.letterRank[letter] = rankLetterCount / allLetterCount

    # Print statement setup:
    def __repr__(self):
        return f'{len(self.wordList)} words remaining.'

    # Method to obtain a guess:
    def getGuesses(self):
        # Instantiate a grading system
        grades = {"" : -99999}

        # Instantiate a counter for items deleted
        deleted = 0

        # For each of the guesses:
        for guess in self.wordList:
            # Keep a new score
            score = 0
            # For each letter of the guess:
            for letter in guess:
                # If the guess contains a letter not used; remove the letter from guess list, and word list
                if letter in self.blackLetters:
                    # To avoid error; check to see if the word is still in the word list
                    if guess in self.wordList:
                        self.wordList.remove(guess)
                        deleted += 1

                        if deleted % 1000 == 0:
                            print(f'{deleted} items removed...')
                    continue
                
                # If the guess contains a yellow letter; add 2 to it's score
                elif letter in self.yellowLetters:
                    score += 2 * self.letterRank[letter]

                # If the guess contains a green letter; add 1 to it's score
                elif letter in self.greenLetters:
                    if guess.index(letter) == self.greenLetters[letter]:
                        score += 1 * self.letterRank[letter]

                # If it is an unused letter, add a special calculation to its score
                else:
                    score += (WordleBot.letters - 2 - (1 / WordleBot.letters * len(self.yellowLetters)) - len(self.greenLetters) // 1) * self.letterRank[letter]

                # If there is more than one instance of this letter; subtract 1 point from it's score
                if guess.count(letter) > 1:
                    score -= 1

            # Add the guess, and it's score to the grading system
            grades[guess] = score

        # Instantiate the selected word
        selectedWord = ""

        # For each graded word:
        for word in grades.keys():
            # If this graded word is better than the next; select this word
            if grades[selectedWord] < grades[word]:
                selectedWord = word

        # Return the selected word
        return selectedWord
    
    # Method to set a Black Letter
    def blackLet(self, char, index):
        self.blackLetters.append(char)
        # Remove all words that have that letter
        changed = True
        while changed:
            changed = False
            for word in self.wordList:
                if char in word:
                    self.wordList.remove(word)
                    changed = True

    # Method to set a Yellow Letter
    def yellowLet(self, char, index):
        # Obtain the index of the yellow letter
        self.yellowLetters.append(char)
        # Remove all words that have that letter in that index
        changed = True
        while changed:
            changed = False
            for word in self.wordList:
                if word[index] == char:
                    self.wordList.remove(word)
                    changed = True

    # Method to set a Green Letter
    def greenLet(self, char, index):
        # Remove any instances of this letter in the yellow letters
        if char in self.yellowLetters:
            self.yellowLetters.remove(char)

        self.greenLetters[char] = index

        # Remove all words that have that letter, but not at the right index
        changed = True
        while changed:
            changed = False
            for word in self.wordList:
                if char in word and word[index] != char:
                    self.wordList.remove(word)
                    changed = True
    
    # Method to remove a guess
    def removeGuess(self, word):
        self.wordList.remove(word)