Get user info from the db

src/data_storage.py

@@ -42,7 +42,7 @@ def add_new_user(email: str, name: str):
         cur = con.cursor()
         cur.execute(
             f"""INSERT INTO users (email, name )
-                    VALUES('{email}', '{name}')
+                    VALUES('{email.lower()}', '{name}')
                     """
         )
 
@@ -56,3 +56,14 @@ def get_user(user_id: int):
         "email": user_data[1],
         "name": user_data[2],
     }
+
+
+def get_user_id_if_email_exists(email: str) -> int | None:
+    with get_db_connection() as con:
+        cur = con.cursor()
+        query_result = cur.execute(f"select user_id from users where email = '{email.lower()}'").fetchone()
+        if query_result:
+            user_id = query_result[0]
+        else:
+            user_id = None
+    return user_id

src/login.py

@@ -5,6 +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
 
 
 class HFFriendlyGSheetsConnection(GSheetsConnection):
@@ -114,7 +115,7 @@ def new_user_submitted():
         st.warning("No name entered.  Please enter your name for display.")
         return
 
-    if email_exists(email):
+    if get_user_id_if_email_exists(email):
         st.warning(
             "User with that email already exists.  If you would like a new login url, please click the New Login URL button."
         )
@@ -151,27 +152,12 @@ def login_by_token(token: str):
     return False
 
 
-def email_exists(email: str):
-    df = conn.read(
-        worksheet="users",
-        usecols=[1],
-        ttl=1,
-    )
-    return bool(email in df.email.tolist())
-
-
 def get_logged_in_user_name_email() -> tuple[str | None, str | None]:
     if not (user_id := st.session_state.get("logged_in_user")):
         # if not logged
         return (None, None)
 
-    df = conn.read(
-        worksheet="users",
-        usecols=[0, 1, 2],
-        ttl=1,
-    )
-
-    user_info_series = df[df["id"] == user_id].squeeze()
-    email = user_info_series["email"]
-    name = user_info_series["name"]
+    user_info_map = get_user(user_id)
+    email = user_info_map["email"]
+    name = user_info_map["name"]
     return email, name