Add automatic login to rs wordle to the selenium player

Now the player can login to rs wordle using its google login with selenium
Fixed a bug in the number of guesses made by the player

rs_wordle_player/firebase_connector.py

@@ -20,7 +20,7 @@ class FirebaseConnector():
         return db
 
     def get_credentials_path(self):
-        credentials_path = os.getenv('FIREBASE_CREDENTIALS_PATH')
+        credentials_path = os.getenv('RS_FIREBASE_CREDENTIALS_PATH')
         return credentials_path
 
     def get_user(self):

rs_wordle_player/rs_wordle_player.py

@@ -34,13 +34,13 @@ def play_game(player, fb_connector, env, model_path):
     states = []
     finished = False
     attempts = 0
-    while not finished or attempts == 6:
+    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'
-    return states, finished
+    return words, finished
 
 
 def play():
@@ -50,12 +50,12 @@ def play():
     env = get_env()
     goal_word = fb.today_word()
     if goal_word and len(goal_word) == 5:
-        states, won = play_game(player, fb, env, model_path)
+        words, won = play_game(player, fb, env, model_path)
     else:
         print("Can't play, today's word is not 5 characters long")
-        states, won = None, None
+        words, won = None, None
     player.finish()
-    return states, won
+    return words, won
 
 
 if __name__ == '__main__':

rs_wordle_player/selenium_player.py

@@ -1,25 +1,62 @@
 import os
-from selenium import webdriver
+import time
 from dotenv import load_dotenv
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
 
 
 class SeleniumPlayer():
 
     def __init__(self):
+        self.wordle_url = self.get_wordle_url()
         self.driver = self.get_driver()
+        self.log_in()
 
     def get_driver(self):
-        driver = webdriver.Chrome()
-        driver.get(self.get_wordle_url())
+        chrome_options = Options()
+        chrome_options.add_experimental_option("detach", True)
+        driver = webdriver.Chrome(options=chrome_options)
+        driver.get(self.wordle_url)
         return driver
 
     def get_wordle_url(self):
         load_dotenv()
         return os.getenv('RS_WORDLE_URL')
 
+    def get_credentials(self):
+        load_dotenv()
+        username = os.getenv('RS_WORDLE_USER')
+        password = os.getenv('RS_WORDLE_PASSWORD')
+        return username, password
+
+    def logged_in(self):
+        return self.driver.current_url != self.wordle_url + '/login'
+
+    def log_in(self):
+        if not self.logged_in():
+            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]
+            login_btn.click()
+            time.sleep(10)
+            wordle_window = self.driver.window_handles[0]
+            login_window = self.driver.window_handles[1]
+            self.driver.switch_to.window(login_window)
+            username, password = self.get_credentials()
+            element = self.driver.find_element(By.ID, 'identifierId')
+            element.send_keys(username)
+            element.send_keys(Keys.ENTER)
+            time.sleep(10)
+            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)
+
     def play_word(self, word):
-        element = self.driver.find_element_by_name('html')
+        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