bot.py

"""A Bot to solve the game wordle"""
from string import ascii_lowercase
from random import randrange


class Bot:
  """A bot to solve the game wordle"""

  def __init__(self, file):
    """initializes a bot, takes file name(string), and print condition(bool)"""
    self.words = self.get_words(file)

  def get_words(self, fname):
    file = open(fname)
    words = file.read().split("\n")
    return words

  def check_move(self, letter, index, prev_guess):
    cleaned = prev_guess[:index]+prev_guess[index+1:]
    return letter in cleaned

  def check_doubles(self, pattern, prev_guess, index):
    for ind, val in enumerate(prev_guess):
      if index != ind:
        if val == prev_guess[index] and (pattern[ind] == 'x' or pattern[ind] == 'o'):
          return True
    return False

  def check_doubles_move(self, pattern, prev_guess, new_guess, letter):
    count = 0
    for ind, pat in enumerate(pattern):
      if pat in 'xo' and prev_guess[ind] == letter:
        count += 1
    return new_guess.count(letter) >= count

  def check(self, prev_guess, pattern, new_guess):
    for index, pat in enumerate(pattern):

      # check for exact
      if pat == "x":
        if new_guess[index] != prev_guess[index]:
          return False

      # check for move
      elif pat == "o":
        # check for difference
        if new_guess[index] == prev_guess[index]:
          return False
        if not self.check_doubles_move(pattern, prev_guess, new_guess, prev_guess[index]):
          return False
        if not self.check_move(prev_guess[index], index, new_guess):
          return False

      # check for not
      elif pat == "_":
        if self.check_doubles(pattern, prev_guess, index):
          letter = prev_guess[index]
          if new_guess.count(letter) >= prev_guess.count(letter):
            return False
        elif prev_guess[index] in new_guess:
          return False

    return True

  def cull_list(self, prev_guess, pattern):
    return [i for i in self.words if self.check(prev_guess, pattern, i)]

  def prompt(self):
    x = input("enter word:")
    y = input(
        "enter pattern ('_' = not in word, 'o' = wrong spot, 'x'=correct:")
    return x, y

  def check_input(self, guess, pattern):
    if (len(guess) != 5):
      return False
    if (len(pattern) != 5):
      return False
    for i in guess.lower():
      if i not in ascii_lowercase:
        return False
    for i in pattern.lower():
      if i not in ["_", "o", "x"]:
        return False
    return True

  def get_input(self):
    cond = False
    while not cond:
      guess, pattern = self.prompt()
      cond = self.check_input(guess, pattern)
    return guess, pattern

  def get_word(self):
    if len(self.words) < 1:
      raise ValueError("WORD LIST EMPTY!!!!")
    return self.words[randrange(0, len(self.words))]

  def run(self, print_d=False):
    for _ in range(6):
      word_choie = self.get_word()
      if print_d:
        print(f"word space:{len(self.words)}")
        print(word_choie)
      if len(self.words) == 1 and print_d:
        print("FINAL ANSWER")
        return
      guess, pattern = self.get_input()
      if pattern == "xxxxx" and print_d:
        print("congratulations!")
        return
      self.words = self.cull_list(guess, pattern)

    if print_d:
      print("failed game")