Connect the user registration and new url request to the db

src/data_storage.py

@@ -104,3 +104,4 @@ def create_new_token_for_user(user_id: int, existing_user: bool = False):
                     VALUES({user_id}, '{token}')
                     """
         )
+    return token

src/login.py

@@ -5,7 +5,7 @@ import re
 import streamlit as st
 from streamlit_gsheets import GSheetsConnection
 from streamlit.runtime.secrets import AttrDict, secrets_singleton
-from data_storage import get_user_id_if_email_exists, get_user, add_new_user
+from data_storage import get_user_id_if_email_exists, get_user, add_new_user, create_new_token_for_user
 
 
 class HFFriendlyGSheetsConnection(GSheetsConnection):
@@ -78,32 +78,35 @@ def generate_new_token_request():
     if not check_email(email := st.session_state["reset_token_user_email"]):
         st.warning("Sorry email is invalid.  Please try a valid email address.")
         return
+    user_id = get_user_id_if_email_exists(email)
+    if not user_id:
+        return
+    token = create_new_token_for_user(user_id, existing_user=True)
+    insert_new_token_to_sheet(email, token)
 
-    conn.update(
-        worksheet="token-requests",
-        data=pd.DataFrame(
-            {
-                "email": email,
-            },
-            index=[0],
-        ),
-    )
     st.info("Request submitted.  Please check your email for a login url with a few minutes.")
     del st.session_state["reset_token_user_email"]
 
 
-def create_new_user_request(email: str, name: str):
+def insert_new_token_to_sheet(email: str, token: str):
     conn.update(
-        worksheet="new-user",
+        worksheet="token-requests",
         data=pd.DataFrame(
             {
                 "email": email,
-                "name": name,
+                "token": token,
             },
             index=[0],
         ),
     )
+
+
+def create_new_user_request(email: str, name: str):
     add_new_user(email, name)
+    user_id = get_user_id_if_email_exists(email)
+    assert user_id
+    token = create_new_token_for_user(user_id)
+    insert_new_token_to_sheet(email, token)
     st.info("New user request submitted.  Please check your email for a login url with a few minutes.")