Add connection to firebase and method to get user attempts from firestore

create play method to make a model play rs version alone
Play word method still not done

.gitignore

@@ -122,3 +122,6 @@ GitHub.sublime-settings
 *.png
 *.jpg
 *.jpeg
+
+# Json files (used for firebase certs and others)
+*.json

play_rs_wordle.py

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

rs_wordle_player/firebase_connector.py

@@ -0,0 +1,60 @@
+import os
+import firebase_admin
+from firebase_admin import credentials
+from firebase_admin import firestore
+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')

rs_wordle_player/rs_wordle_player.py

@@ -0,0 +1,53 @@
+import gym
+import os
+from a3c.play import suggest
+from .firebase_connector import init_fb_app, today_user_attempts, today_word
+
+
+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_attempts():
+    attempts = today_user_attempts()
+    words = []
+    results = []
+    for attempt in attempts:
+        word, result = attempt.values()
+        words.append(word)
+        results.append(get_state_from_fb_result(result))
+    return words, results
+
+
+def play_word(word):
+    # Makes an attemp to rs wordle
+    # TO DO
+    pass
+
+
+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()
+    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
+    else:
+        print("Can't play, today's word is not 5 characters long")
+        new_attempt = None
+    return new_attempt