Add selenium class to simulate the user in the rs wordle

Change to class firebase conection
Refactor rs_wordle_player

play_rs_wordle.py

@@ -1,4 +0,0 @@
-from rs_wordle_player.rs_wordle_player import play
-
-if __name__ == '__main__':
-    print(play())

requirements.txt

@@ -1,3 +1,6 @@
+dotenv
+firebase-admin
 gym
 matplotlib
+selenium
 torch

rs_wordle_player/firebase_connector.py

@@ -6,55 +6,54 @@ from datetime import datetime
 from dotenv import load_dotenv
 
 
-def init_fb_app():
-    cert_path = get_credentials_path()
-    cred = credentials.Certificate(cert_path)
-    firebase_admin.initialize_app(cred)
-
-
-def connect_to_db():
-    db = firestore.client()
-    return db
-
-
-def get_credentials_path():
-    load_dotenv()
-    credentials_path = os.getenv('FIREBASE_CREDENTIALS_PATH')
-    return credentials_path
-
-
-def get_user():
-    load_dotenv()
-    user = os.getenv('RS_WORDLE_USER')
-    return user
-
-
-def today():
-    return datetime.today().strftime('%Y%m%d')
-
-
-def today_user_results():
-    db = connect_to_db()
-    daily_results_col = 'dailyResults'
-    currentUser = get_user()
-    # Execute the query and get the first result
-    docs = db.collection(daily_results_col).where(
-        'user.email', '==', currentUser).where(
-        'date', '==', today()).limit(1).get()
-    return docs
-
-
-def today_user_attempts():
-    docs = today_user_results()
-    attempted_words = []
-    if len(docs) > 0:
-        doc = docs[0]
-        attempted_words = doc.to_dict().get('attemptedWords')
-    return attempted_words
-
-
-def today_word():
-    words_col = 'words'
-    db = connect_to_db()
-    doc = db.collection(words_col).document(today())
-    return doc.get().get('word')
+class FirebaseConnector():
+
+    def __init__(self):
+        load_dotenv()
+        cert_path = self.get_credentials_path()
+        cred = credentials.Certificate(cert_path)
+        firebase_admin.initialize_app(cred)
+        self.db = self.get_db()
+
+    def get_db(self):
+        db = firestore.client()
+        return db
+
+    def get_credentials_path(self):
+        credentials_path = os.getenv('FIREBASE_CREDENTIALS_PATH')
+        return credentials_path
+
+    def get_user(self):
+        user = os.getenv('RS_WORDLE_USER')
+        return user
+
+    def get_state_from_fb_result(self, firebase_result):
+        result_number_map = {'incorrect': '0',
+                             'misplaced': '1',
+                             'correct': '2'}
+        return ''.join(map(lambda char_res: result_number_map[char_res], firebase_result))
+
+    def today(self):
+        return datetime.today().strftime('%Y%m%d')
+
+    def today_user_results(self):
+        daily_results_col = 'dailyResults'
+        currentUser = self.get_user()
+        # Execute the query and get the first result
+        docs = self.db.collection(daily_results_col).where(
+            'user.email', '==', currentUser).where(
+            'date', '==', self.today()).limit(1).get()
+        return docs
+
+    def today_user_attempts(self):
+        docs = self.today_user_results()
+        attempted_words = []
+        if len(docs) > 0:
+            doc = docs[0]
+            attempted_words = doc.to_dict().get('attemptedWords')
+        return attempted_words
+
+    def today_word(self):
+        words_col = 'words'
+        doc = self.db.collection(words_col).document(self.today())
+        return doc.get().get('word')

rs_wordle_player/rs_wordle_player.py

@@ -1,53 +1,62 @@
 import gym
 import os
 from a3c.play import suggest
-from .firebase_connector import init_fb_app, today_user_attempts, today_word
+from dotenv import load_dotenv
+from .firebase_connector import FirebaseConnector
+from .selenium_player import SeleniumPlayer
 
 
-def get_state_from_fb_result(firebase_result):
-    result_number_map = {'incorrect': '0',
-                         'misplaced': '1',
-                         'correct': '2'}
-    return ''.join(map(lambda char_res: result_number_map[char_res], firebase_result))
+def get_model_path():
+    load_dotenv()
+    model_name = os.getenv('RS_WORDLE_MODEL_NAME')
+    model_checkpoint_dir = os.path.join('checkpoints', 'best_models')
+    return os.path.join(model_checkpoint_dir, model_name)
+
 
+def get_env():
+    env_id = 'WordleEnvFull-v0'
+    return gym.make(env_id)
 
-def get_attempts():
-    attempts = today_user_attempts()
+
+def get_attempts(fb_connector):
+    attempts = fb_connector.today_user_attempts()
     words = []
     results = []
     for attempt in attempts:
         word, result = attempt.values()
         words.append(word)
-        results.append(get_state_from_fb_result(result))
+        results.append(fb_connector.get_state_from_fb_result(result))
     return words, results
 
 
-def play_word(word):
-    # Makes an attemp to rs wordle
-    # TO DO
-    pass
+def play_game(player, fb_connector, env, model_path):
+    words = []
+    states = []
+    finished = False
+    attempts = 0
+    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
 
 
 def play():
-    init_fb_app()
-    pretrained_model_name = 'model_95%_one_hot.pth'
-    env_id = 'WordleEnvFull-v0'
-    env = gym.make(env_id)
-    model_checkpoint_dir = os.path.join('checkpoints', 'best_models')
-    pretrained_model_path = os.path.join(model_checkpoint_dir, pretrained_model_name)
-    goal_word = today_word()
+    fb = FirebaseConnector()
+    player = SeleniumPlayer()
+    model_path = get_model_path()
+    env = get_env()
+    goal_word = fb.today_word()
     if goal_word and len(goal_word) == 5:
-        words = []
-        states = []
-        finished = False
-        attempts = 0
-        while not finished or attempts == 6:
-            attempts += 1
-            new_attempt = suggest(env, words, states, pretrained_model_path)
-            play_word(new_attempt)
-            words, states = get_attempts()
-            finished = states[-1] == '22222' or len(states) == 6
+        states, won = play_game(player, fb, env, model_path)
     else:
         print("Can't play, today's word is not 5 characters long")
-        new_attempt = None
-    return new_attempt
+        states, won = None, None
+    player.finish()
+    return states, won
+
+
+if __name__ == '__main__':
+    print(play())

rs_wordle_player/selenium_player.py

@@ -0,0 +1,29 @@
+import os
+from selenium import webdriver
+from dotenv import load_dotenv
+from selenium.webdriver.common.keys import Keys
+
+
+class SeleniumPlayer():
+
+    def __init__(self):
+        self.driver = self.get_driver()
+
+    def get_driver(self):
+        driver = webdriver.Chrome()
+        driver.get(self.get_wordle_url())
+        return driver
+
+    def get_wordle_url(self):
+        load_dotenv()
+        return os.getenv('RS_WORDLE_URL')
+
+    def play_word(self, word):
+        element = self.driver.find_element_by_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)
+
+    def finish(self):
+        self.driver.quit()