labyrinth-cs-proj/wordle/wordle.py

193 lines
5.7 KiB
Python

#
# Copyright © 2023 adithyagenie
#
# SPDX-License-Identifier: AGPL-3.0-or-later
#
import curses
import random
import time
import maze.menu
import maze.modules.maze as m1
from wordle.dictionary import defnsyn
quitwordle = False
words = open("wordle\\words.txt", "r").read().split("\n")
colorPairBindings = {"c": 2, "w": 3, "n": 7, "u": 6}
completionMessages = [
"",
"Genius!",
"Unbelievable!",
"Splendid!",
"Amazing!",
"Great!",
"Good!",
]
# Draw one row of the board
# Generates new row + colours alphabets in
def writeWord(s, word, remark, y):
s.addstr(y, 0, "│ │ │ │ │ │\n├─┼─┼─┼─┼─┤")
for i, (letter, color) in enumerate(zip(word, remark)):
s.addstr(
y,
i * 2 + 1,
letter.upper(),
curses.color_pair(colorPairBindings[color]),
)
# Score a word
def score(guess, word, alphabet):
res = [" "] * 5
counts = [0] * 26
# First process correct letters
for i, c in enumerate(guess):
if (
c == word[i]
): # checking if guess letter corresponds to letter of the same index in chosen word
charIndex = (
ord(c) - 97
) # 97 corresponds to a - gives alphabet number. (eg. h = 8)
counts[charIndex] += 1
res[i] = "c" # correct spot
alphabet[charIndex] = "c"
# Then handle wrong and nonpresent letters
for i, c in enumerate(guess):
if c != word[i]:
charIndex = ord(c) - 97
counts[charIndex] += 1
if (
c in word and word.count(c) >= counts[charIndex]
): # if freq of letters in guess lesser than freq in word
res[i] = "w" # wrong spot
if alphabet[charIndex] != "c":
alphabet[charIndex] = "w"
else:
res[i] = "n" # not in word
alphabet[charIndex] = "n"
return "".join(res), alphabet
# Render current board
# Updates alphabet use state + renders colours of guesses
def render(s, guesses, alphabet):
s.addstr(0, 0, "=== WORDLE ===", curses.color_pair(2))
for i, c in enumerate(alphabet):
s.addstr(
1 + int(i // 7),
(i % 7) * 2,
chr(65 + i),
curses.color_pair(colorPairBindings[c]),
)
s.addstr(6, 0, "╭─┬─┬─┬─┬─╮")
for i, (w, r) in enumerate(guesses):
writeWord(s, w, r, i * 2 + 7)
s.addstr(len(guesses) * 2 + 10, 0, " ")
# Accept word from user input
def getWord(s, y):
word = ""
while True:
writeWord(s, word, "u" * len(word), y) # u = default blue colour
k = s.getch()
try:
if k == 8: # backspace
word = word[:-1]
elif k == 27: # esc
global quitwordle
quitwordle = True
return "hello"
elif chr(k) == "\n" and len(word) == 5:
return word
elif chr(k).isalpha() and len(word) < 5:
word += chr(k)
except:
pass
# Run one game of Wordle
def run(s):
s.clear()
word = random.choice(words) # chosen word
defn, synonyms = defnsyn(word)
guesses = [] # stores each guess and its result
alphabet = ["u"] * 26 # current status of each letter whether used or not
# c = correct positon, w = correct letter but not position, n = wrong letter, u = not used
# "ccccc" means all letters are in correct spot
while not (len(guesses)) or (guesses[-1][1] != "ccccc" and len(guesses) < 6):
if quitwordle:
return
render(s, guesses, alphabet) # Update current state of board from start
guess = getWord(s, len(guesses) * 2 + 7).lower()
if not (guess in words): # Check if given word is valid
s.addstr(len(guesses) * 2 + 10, 0, "INVALID WORD", curses.color_pair(1))
s.refresh()
time.sleep(1)
continue
res, alphabet = score(guess, word, alphabet)
guesses.append([guess, res])
render(s, guesses, alphabet) # Renders final board
# Ending spiel
s.addstr(len(guesses) * 2 + 6, 0, "╰─┴─┴─┴─┴─╯")
if guesses[-1][1] != "ccccc":
s.addstr(
len(guesses) * 2 + 8, 0, "No more tries - the word was " + word.upper()
)
else:
s.addstr(len(guesses) * 2 + 8, 0, completionMessages[len(guesses)])
if defn and synonyms:
s.addstr(len(guesses) * 2 + 9, 0, word + ": ", curses.color_pair(2))
synonyms = ", ".join(synonyms)
s.addstr(len(guesses) * 2 + 9, 8, defn)
s.addstr(len(guesses) * 2 + 10, 0, "Some synonyms: ", curses.color_pair(2))
s.addstr(len(guesses) * 2 + 10, 16, synonyms)
if len(guesses) == 6 and guesses[-1][1] != "ccccc":
guesses.append("Bruh")
finalscore = allocatescore(guesses)
return finalscore
def allocatescore(guesses):
finalscore = 0
if len(guesses) <= 3:
finalscore = 50
elif len(guesses) == 4:
finalscore = 40
elif len(guesses) == 5:
finalscore = 30
elif len(guesses) == 6:
finalscore = 20
return finalscore
# Main function
def main(s):
# Initialize colors
global quitwordle
quitwordle = False
for p in [
(1, curses.COLOR_RED),
(2, curses.COLOR_GREEN),
(3, curses.COLOR_YELLOW),
(7, curses.COLOR_WHITE),
(6, curses.COLOR_CYAN),
]:
curses.init_pair(p[0], p[1], curses.COLOR_BLACK)
# Run game
finalscore = run(s)
if not finalscore:
finalscore = 0
else:
while s.getch() == -1:
pass
m1.play(s, executeguest=True, outerscore=finalscore, outergame="wordle")
maze.menu.menu(s)
return