Implement ability to save roster

src/login.py

@@ -1,5 +1,6 @@
 import hmac
 import json
+import pandas as pd
 import streamlit as st
 from streamlit_gsheets import GSheetsConnection
 from streamlit.runtime.secrets import AttrDict, secrets_singleton
@@ -40,6 +41,7 @@ def get_user_id_from_email_if_exists(email: str) -> int:
     df_email = conn.read(
         worksheet="users",
         usecols=[0, 1],
+        ttl=3600,
     )
     df_email = df_email[df_email.email.notna()]
     for row in df_email.itertuples():
@@ -51,8 +53,9 @@ def get_user_id_from_email_if_exists(email: str) -> int:
 def get_password_from_user_id(user_id: int) -> str | None:
     try:
         df_pass = conn.read(
-            worksheet=f"user-{user_id}",
+            worksheet=f"user-{user_id}-password",
             usecols=[0],
+            ttl=0,
         )
         return df_pass.password[0]
     except Exception:
@@ -95,3 +98,22 @@ def check_password():
     if st.session_state.get("password_correct", True) is False:
         st.error("😕 User not known or password incorrect")
     return False
+
+
+def get_user_team():
+    user_id = st.session_state.get("logged_in_user")
+    if not user_id:
+        return {}
+    df_team = conn.read(
+        worksheet=f"user-{user_id}-roster",
+        ttl=0,
+    )
+    return df_team.loc[0].to_dict()
+
+
+def save_user_team(team_selections):
+    user_id = st.session_state.get("logged_in_user")
+    conn.update(
+        worksheet=f"user-{user_id}-roster",
+        data=pd.DataFrame(team_selections, index=[0]),
+    )

src/pages/10_Set_Your_Lineup.py

@@ -1,11 +1,13 @@
 from dataclasses import dataclass
+import numpy as np
 import streamlit as st
 
 from config import DEFAULT_ICON
 from shared_page import common_page_config
 
 from queries.nflverse.github_data import get_weekly_rosters
-from login import check_password
+from login import check_password, get_user_team, save_user_team
+from domain.playoffs import PLAYOFF_WEEK_TO_NAME
 
 
 @dataclass
@@ -45,7 +47,7 @@ def player_options_from_df(df_options, position_filter: str) -> list[PlayerOptio
     )
 
 
-# @st.cache_data(ttl=60 * 60 * 24)
+@st.cache_data(ttl=60 * 60 * 24)
 def load_options():
     # filter active only
     df_rosters = get_weekly_rosters()
@@ -58,7 +60,7 @@ def load_options():
     rb_options = player_options_from_df(df_rosters, "RB")
     te_options = player_options_from_df(df_rosters, "TE")
     k_options = player_options_from_df(df_rosters, "K")
-    return qb_options, wr_options, rb_options, te_options, k_options, df_rosters
+    return qb_options, wr_options, rb_options, te_options, k_options
 
 
 def format_player_option(player_opt: PlayerOption) -> str:
@@ -71,6 +73,32 @@ def display_player(player_opt: PlayerOption | None):
             st.image(player_opt.headshot_url, caption=player_opt.full_name)
 
 
+def position_cell(pos_label: str, options_list: list[PlayerOption], existing_selection_map):
+    if selected_player := existing_selection_map.get(pos_label) != np.nan:
+        try:
+            selected_option_idx, _ = next((i, v) for i, v in enumerate(options_list) if selected_player == v.gsis_id)
+        except StopIteration:
+            selected_option_idx = 0
+    else:
+        selected_option_idx = 0
+
+    selected_player_from_box = st.selectbox(
+        pos_label,
+        options=options_list,
+        format_func=format_player_option,
+        index=selected_option_idx,
+    )
+    if selected_player_from_box:
+        if selected_player_from_box.gsis_id and selected_player_from_box.gsis_id != selected_player:
+            update_and_save_selection(pos_label, selected_player_from_box.gsis_id, existing_selection_map),
+    display_player(selected_player_from_box)
+
+
+def update_and_save_selection(pos_label: str, selection_id: str, existing_selection_map):
+    existing_selection_map[pos_label] = selection_id
+    save_user_team(existing_selection_map)
+
+
 def get_page():
     page_title = "Select Your Team"
     st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
@@ -80,34 +108,31 @@ def get_page():
         st.write("Sorry, you must be logged in first to play")
         st.stop()
 
-    qb_options, wr_options, rb_options, te_options, k_options, df_rosters = load_options()
-
-    selection_cols = st.columns(8)
-
-    with selection_cols[0]:
-        qb_1 = st.selectbox("QB", options=qb_options, format_func=format_player_option)
-        display_player(qb_1)
-    with selection_cols[1]:
-        rb_1 = st.selectbox("RB1", options=rb_options, format_func=format_player_option)
-        display_player(rb_1)
-    with selection_cols[2]:
-        rb_2 = st.selectbox("RB2", options=rb_options, format_func=format_player_option)
-        display_player(rb_2)
-    with selection_cols[3]:
-        wr_1 = st.selectbox("WR1", options=wr_options, format_func=format_player_option)
-        display_player(wr_1)
-    with selection_cols[4]:
-        wr_2 = st.selectbox("WR2", options=wr_options, format_func=format_player_option)
-        display_player(wr_2)
-    with selection_cols[5]:
-        te_1 = st.selectbox("TE", options=te_options, format_func=format_player_option)
-        display_player(te_1)
-    with selection_cols[6]:
-        k_1 = st.selectbox("K", options=k_options, format_func=format_player_option)
-        display_player(k_1)
-    with selection_cols[7]:
-        def_1 = st.selectbox("DEF", options=[], format_func=format_player_option)
-        display_player(def_1)
+    existing_selections = get_user_team()
+    st.write(get_user_team())
+
+    qb_options, wr_options, rb_options, te_options, k_options = load_options()
+
+    for week in range(1, 5):
+        st.header(PLAYOFF_WEEK_TO_NAME[week])
+        selection_cols = st.columns(8)
+
+        with selection_cols[0]:
+            position_cell(f"{week}-QB-1", qb_options, existing_selections)
+        with selection_cols[1]:
+            position_cell(f"{week}-RB-1", rb_options, existing_selections)
+        with selection_cols[2]:
+            position_cell(f"{week}-RB-2", rb_options, existing_selections)
+        with selection_cols[3]:
+            position_cell(f"{week}-WR-1", wr_options, existing_selections)
+        with selection_cols[4]:
+            position_cell(f"{week}-WR-2", wr_options, existing_selections)
+        with selection_cols[5]:
+            position_cell(f"{week}-TE-1", te_options, existing_selections)
+        with selection_cols[6]:
+            position_cell(f"{week}-K-1", k_options, existing_selections)
+        with selection_cols[7]:
+            position_cell(f"{week}-DEF-1", [], existing_selections)
 
 
 if __name__ == "__main__":