Refactor position_cell to only allow current week edits

src/pages/10_Set_Your_Lineup.py

@@ -1,5 +1,4 @@
 from dataclasses import dataclass
-import numpy as np
 import streamlit as st
 
 from config import DEFAULT_ICON
@@ -7,7 +6,7 @@ from shared_page import common_page_config
 
 from queries.nflverse.github_data import get_weekly_rosters
 from login import check_password, get_user_team, save_user_team
-from domain.playoffs import PLAYOFF_WEEK_TO_NAME
+from domain.playoffs import PLAYOFF_WEEK_TO_NAME, CURRENT_PLAYOFF_WEEK
 
 
 @dataclass
@@ -73,23 +72,33 @@ 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
+def position_cell(week: str, pos_str: str, options_list: list[PlayerOption], existing_selection_map):
+    pos_label = f"{week}-{pos_str}"
+    selected_id = existing_selection_map.get(pos_label)
+    if isinstance(selected_id, str):
+        selected_option_idx, selected_player = next(
+            (i, v) for i, v in enumerate(options_list) if str(selected_id) == str(v.gsis_id)
+        )
     else:
+        selected_player = PlayerOption.empty_player()
+        selected_option_idx = 0
+    if int(week) > CURRENT_PLAYOFF_WEEK:
+        options = []
         selected_option_idx = 0
+    elif int(week) < CURRENT_PLAYOFF_WEEK:
+        options = [selected_player]
+        selected_option_idx = 0
+    else:
+        options = options_list
 
     selected_player_from_box = st.selectbox(
         pos_label,
-        options=options_list,
+        options=options,
         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:
+    if selected_player_from_box and int(week) == CURRENT_PLAYOFF_WEEK:
+        if selected_player_from_box.gsis_id and selected_player_from_box.gsis_id != selected_id:
             update_and_save_selection(pos_label, selected_player_from_box.gsis_id, existing_selection_map),
     display_player(selected_player_from_box)
 
@@ -109,7 +118,6 @@ def get_page():
         st.stop()
 
     existing_selections = get_user_team()
-    st.write(get_user_team())
 
     qb_options, wr_options, rb_options, te_options, k_options = load_options()
 
@@ -118,21 +126,21 @@ def get_page():
         selection_cols = st.columns(8)
 
         with selection_cols[0]:
-            position_cell(f"{week}-QB-1", qb_options, existing_selections)
+            position_cell(week, "QB-1", qb_options, existing_selections)
         with selection_cols[1]:
-            position_cell(f"{week}-RB-1", rb_options, existing_selections)
+            position_cell(week, "RB-1", rb_options, existing_selections)
         with selection_cols[2]:
-            position_cell(f"{week}-RB-2", rb_options, existing_selections)
+            position_cell(week, "RB-2", rb_options, existing_selections)
         with selection_cols[3]:
-            position_cell(f"{week}-WR-1", wr_options, existing_selections)
+            position_cell(week, "WR-1", wr_options, existing_selections)
         with selection_cols[4]:
-            position_cell(f"{week}-WR-2", wr_options, existing_selections)
+            position_cell(week, "WR-2", wr_options, existing_selections)
         with selection_cols[5]:
-            position_cell(f"{week}-TE-1", te_options, existing_selections)
+            position_cell(week, "TE-1", te_options, existing_selections)
         with selection_cols[6]:
-            position_cell(f"{week}-K-1", k_options, existing_selections)
+            position_cell(week, "K-1", k_options, existing_selections)
         with selection_cols[7]:
-            position_cell(f"{week}-DEF-1", [], existing_selections)
+            position_cell(week, "DEF-1", [], existing_selections)
 
 
 if __name__ == "__main__":