73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
import random
|
|
from exceptions import WordleGameOver, WordleInvalidGuess, WordleUnknownWord
|
|
|
|
CORRECT = 0
|
|
IN_WORD = 1
|
|
NOT_IN_WORD = 2
|
|
|
|
with open("words.txt", "r") as words_file:
|
|
words = words_file.readline().strip().split(",")
|
|
guesses = words_file.readline().strip().split(",")
|
|
|
|
class Wordle:
|
|
|
|
def __init__(self, max_guesses = 6):
|
|
self.max_guesses = max_guesses
|
|
self.guesses = []
|
|
self._word = random.choice(words).lower()
|
|
self._ended = False
|
|
self._won = False
|
|
|
|
def has_ended(self):
|
|
return self._ended
|
|
|
|
def has_won(self):
|
|
return self._won
|
|
|
|
def get_characters(self):
|
|
characters = { c : None for c in "abcdefghijklmnopqrstuvwxyz"}
|
|
for guess, result in self.guesses:
|
|
for c, r in zip(guess, result):
|
|
characters[c] = r
|
|
return characters
|
|
|
|
def guess(self, word):
|
|
# Check if guess is valid
|
|
if len(word) != 5:
|
|
raise WordleInvalidGuess("Guess must have length 5")
|
|
word = word.lower()
|
|
if word not in words and word not in guesses:
|
|
raise WordleUnknownWord(f"Unknown word '{word}'")
|
|
|
|
if self.has_ended():
|
|
raise WordleGameOver("Game is already over.")
|
|
|
|
result = []
|
|
correct = True
|
|
|
|
for index, c in enumerate(word):
|
|
if c not in self._word:
|
|
result.append(NOT_IN_WORD)
|
|
correct = False
|
|
elif word[index] == self._word[index]:
|
|
result.append(CORRECT)
|
|
correct = correct and True
|
|
else:
|
|
result.append(IN_WORD)
|
|
correct = False
|
|
|
|
self.guesses.append((word, result))
|
|
if correct or len(self.guesses) == self.max_guesses:
|
|
self._ended = True
|
|
self._won = correct
|
|
|
|
return result
|
|
|
|
def reset(self):
|
|
self.guesses = []
|
|
self._ended = False
|
|
self._won = False
|
|
|
|
def reroll(self):
|
|
self._word = random.choice(words)
|
|
self.reset() |