Add different sleep so the simulated player dont get errors and plays smoothly

rs_wordle_player/rs_wordle_player.py

@@ -29,17 +29,22 @@ def get_attempts(fb_connector):
     return words, results
 
 
+def is_game_finished(states):
+    if states:
+        return states[-1] == '22222' or len(states) == 6
+    return False
+
+
 def play_game(player, fb_connector, env, model_path):
-    words = []
-    states = []
-    finished = False
     attempts = 0
+    words, states = get_attempts(fb_connector)
+    finished = is_game_finished(states)
     while not finished or attempts < 6:
         attempts += 1
         new_attempt = suggest(env, words, states, model_path)
         player.play_word(new_attempt)
         words, states = get_attempts(fb_connector)
-        finished = states[-1] == '22222'
+        finished = is_game_finished(states)
     return words, finished
 
 

rs_wordle_player/selenium_player.py

@@ -2,6 +2,7 @@ import os
 import time
 from dotenv import load_dotenv
 from selenium import webdriver
+from selenium.common.exceptions import UnexpectedAlertPresentException
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
@@ -36,6 +37,7 @@ class SeleniumPlayer():
 
     def log_in(self):
         if not self.logged_in():
+            time.sleep(2)
             login_div = self.driver.find_element(By.CLASS_NAME, 'login-button')
             login_btns = login_div.find_elements(By.TAG_NAME, 'button')
             login_btn = login_btns[0]
@@ -52,15 +54,23 @@ class SeleniumPlayer():
             element = self.driver.find_element(By.NAME, 'password')
             element.send_keys(password)
             element.send_keys(Keys.ENTER)
-            time.sleep(10)
             self.driver.switch_to.window(wordle_window)
+            time.sleep(5)
+            onboard_div = self.driver.find_element(By.CLASS_NAME, 'onboarding-modal-container')
+            onboard_btn = onboard_div.find_elements(By.TAG_NAME, 'button')
+            onboard_btn[-1].click()
+
 
     def play_word(self, word):
-        element = self.driver.find_element(By.TAG_NAME, 'html')
-        # simulate typing the letters in the word into the input field
-        element.send_keys(word)
-        # simulate pressing the Enter key
-        element.send_keys(Keys.ENTER)
+        try:
+            element = self.driver.find_element(By.TAG_NAME, 'html')
+            # simulate typing the letters in the word into the input field
+            element.send_keys(word)
+            # simulate pressing the Enter key
+            element.send_keys(Keys.ENTER)
+            time.sleep(5)
+        except UnexpectedAlertPresentException:
+            print('Won game alert on screen')
 
     def finish(self):
         self.driver.quit()