Add a filter for nfl teams for player selection

src/pages/10_Set_Your_Lineup.py

@@ -39,11 +39,18 @@ def format_player_option(player_opt: PlayerOption) -> str:
 
 
 def position_cell(
-    week: str, pos_str: str, pos_idx: int, options_map: dict[str, dict[int, list[PlayerOption]]], existing_selection_map
+    week: str,
+    pos_str: str,
+    pos_idx: int,
+    options_map: dict[str, dict[int, list[PlayerOption]]],
+    existing_selection_map,
+    team_filter: list[str],
 ):
     pos_label = f"{week}-{pos_str}-{pos_idx}"
     selected_id = existing_selection_map.get(pos_label)
     options_list = options_map[pos_str][int(week)]
+    if team_filter:
+        options_list = [x for x in options_list if x.team in team_filter or str(selected_id) == str(x.gsis_id)]
     non_locked_options = [x for x in options_list if not x.is_locked()]
     disabled = False
     # get selected player by id from options
@@ -100,25 +107,25 @@ def update_and_save_selection(pos_label: str, selection_id: str):
     st.rerun()
 
 
-def week_selections(week, player_options, existing_selections):
+def week_selections(week, player_options, existing_selections, team_filter: list[str]):
     selection_cols = st.columns(8)
 
     with selection_cols[0]:
-        position_cell(week, "QB", 1, player_options, existing_selections)
+        position_cell(week, "QB", 1, player_options, existing_selections, team_filter)
     with selection_cols[1]:
-        position_cell(week, "RB", 1, player_options, existing_selections)
+        position_cell(week, "RB", 1, player_options, existing_selections, team_filter)
     with selection_cols[2]:
-        position_cell(week, "RB", 2, player_options, existing_selections)
+        position_cell(week, "RB", 2, player_options, existing_selections, team_filter)
     with selection_cols[3]:
-        position_cell(week, "WR", 1, player_options, existing_selections)
+        position_cell(week, "WR", 1, player_options, existing_selections, team_filter)
     with selection_cols[4]:
-        position_cell(week, "WR", 2, player_options, existing_selections)
+        position_cell(week, "WR", 2, player_options, existing_selections, team_filter)
     with selection_cols[5]:
-        position_cell(week, "TE", 1, player_options, existing_selections)
+        position_cell(week, "TE", 1, player_options, existing_selections, team_filter)
     with selection_cols[6]:
-        position_cell(week, "K", 1, player_options, existing_selections)
+        position_cell(week, "K", 1, player_options, existing_selections, team_filter)
     with selection_cols[7]:
-        position_cell(week, "DEF", 1, player_options, existing_selections)
+        position_cell(week, "DEF", 1, player_options, existing_selections, team_filter)
 
 
 def get_page():
@@ -146,15 +153,18 @@ def get_page():
         st.stop()
 
     player_options = load_options()
+    # TODO - make more robust than just picking week 1 defenses but should work for now
+    team_options_for_filter = sorted([x.team for x in player_options["DEF"][1]])
+    team_filter = st.multiselect("Filter for NFL teams", team_options_for_filter)
 
     for week in range(1, 5):
         existing_week_selections = {k: v for k, v in existing_selections.items() if k[0] == str(week)}
         st.header(PLAYOFF_WEEK_TO_NAME[week])
         if week < CURRENT_PLAYOFF_WEEK:
             with st.expander("Show Previous Week"):
-                week_selections(week, player_options, existing_week_selections)
+                week_selections(week, player_options, existing_week_selections, team_filter)
         else:
-            week_selections(week, player_options, existing_week_selections)
+            week_selections(week, player_options, existing_week_selections, team_filter)
 
     set_selectbox_readonly()