Spaces:
Sleeping
Sleeping
File size: 3,007 Bytes
44db2f9 aec341d 44db2f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
from rich.prompt import Prompt
from rich.console import Console
from random import choice
from wordle_env.words import target_vocabulary, complete_vocabulary
SQUARES = {
'correct_place': '🟩',
'correct_letter': '🟨',
'incorrect_letter': '⬛'
}
WELCOME_MESSAGE = f'\n[white on blue] WELCOME TO WORDLE [/]\n'
PLAYER_INSTRUCTIONS = "You may start guessing\n"
GUESS_STATEMENT = "\nEnter your guess"
ALLOWED_GUESSES = 6
def correct_place(letter):
return f'[black on green]{letter}[/]'
def correct_letter(letter):
return f'[black on yellow]{letter}[/]'
def incorrect_letter(letter):
return f'[black on white]{letter}[/]'
def check_guess(guess, answer):
guessed = [None, None, None, None, None]
wordle_pattern = []
processed_letters = []
for i, letter in enumerate(guess):
if answer[i] == guess[i]:
guessed[i] = correct_place(letter)
wordle_pattern.append(SQUARES['correct_place'])
processed_letters.append(letter)
for i, letter in enumerate(guess):
if answer[i] != guess[i]:
if letter in answer and answer.count(letter) > processed_letters.count(letter):
guessed[i] = correct_letter(letter)
wordle_pattern.append(SQUARES['correct_letter'])
else:
guessed[i] = incorrect_letter(letter)
wordle_pattern.append(SQUARES['incorrect_letter'])
processed_letters.append(letter)
return ''.join(guessed), ''.join(wordle_pattern)
def game(console, chosen_word):
end_of_game = False
already_guessed = []
full_wordle_pattern = []
all_words_guessed = []
while not end_of_game:
guess = Prompt.ask(GUESS_STATEMENT).upper()
while len(guess) != 5 or guess in already_guessed or guess not in complete_vocabulary:
if guess in already_guessed:
console.print("[red]You've already guessed this word!!\n[/]")
else:
console.print('[red]Please enter a valid 5-letter word!!\n[/]')
guess = Prompt.ask(GUESS_STATEMENT).upper()
already_guessed.append(guess)
guessed, pattern = check_guess(guess, chosen_word)
all_words_guessed.append(guessed)
full_wordle_pattern.append(pattern)
console.print(*all_words_guessed, sep="\n")
if guess == chosen_word or len(already_guessed) == ALLOWED_GUESSES:
end_of_game = True
if len(already_guessed) == ALLOWED_GUESSES and guess != chosen_word:
console.print(f"\n[red]WORDLE X/{ALLOWED_GUESSES}[/]")
console.print(f'\n[green]Correct Word: {chosen_word}[/]')
else:
console.print(f"\n[green]WORDLE {len(already_guessed)}/{ALLOWED_GUESSES}[/]\n")
console.print(*full_wordle_pattern, sep="\n")
if __name__ == '__main__':
console = Console()
chosen_word = choice(target_vocabulary)
console.print(WELCOME_MESSAGE)
console.print(PLAYER_INSTRUCTIONS)
game(console, chosen_word)
|