Send and retrieve data from website

Author: Adrien227

wordle_solver/wordle_solver.py

@@ -6,15 +6,78 @@
 import string
 from typing import Set
 from selenium import webdriver
+from selenium.webdriver.common.by import By
 from selenium.webdriver.firefox.service import Service as FirefoxService
 from webdriver_manager.firefox import GeckoDriverManager
+from pynput import keyboard
+import time
+
+
+# Check word length
+def check_word_length(word):
+    if len(word) != 6:
+        return False
+    else:
+        return True
+
+
+list_of_words = open("words_alpha.txt", "r").read().splitlines()
+list_of_words = list(filter(check_word_length, list_of_words))
+
+
+def on_release(key):
+    if key == keyboard.Key.esc:
+        return False  # stop listener
+
+
+def get_wordle_evaluation(game_row, browser):
+    tiles = game_row.find_elements(By.CLASS_NAME, "Tile-module_tile__3ayIZ")
+    evaluation = []
+    eval_to_int = {
+        "correct": 1,
+        "present": 0,
+        "absent": -1,
+        "empty": -2,
+        "tbd": -3
+    }
+    for tile in tiles:
+        evaluation.append(eval_to_int[tile.get_attribute("data-state")])
+    print(evaluation)
+    return tuple(evaluation)
 
 
 def main():
     start_button = 'esc'
     browser = webdriver.Firefox(service=FirefoxService(GeckoDriverManager().install()))
     browser.get("https://www.nytimes.com/games/wordle/index.html")
 
+    # Wait for start
+    with keyboard.Listener(on_release=on_release) as listener:
+        listener.join()
+    print("Starting")
+
+    first_string = "moron"
+    keyboard_controller = keyboard.Controller()
+    keyboard_controller.type(first_string)
+    keyboard_controller.tap(keyboard.Key.enter)
+
+    time.sleep(2)
+
+    game_rows = browser.find_elements(By.CLASS_NAME, 'Row-module_row__dEHfN')
+    # print("Game rows: ", len(game_rows))
+    get_wordle_evaluation(game_rows[0], browser)
+
+    time.sleep(1)
+
+    second_string = "tests"
+    keyboard_controller.type(second_string)
+    keyboard_controller.tap(keyboard.Key.enter)
+
+    time.sleep(2)
+
+    get_wordle_evaluation(game_rows[1], browser)
+
+
 #From the basic list of words, return all the words with 5 characters.
 def get_list_of_five():