Initial commit

wordle.py

@@ -0,0 +1,73 @@
+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()