Add set your team page with player images

src/pages/10_Set_Your_Lineup.py

@@ -0,0 +1,110 @@
+from dataclasses import dataclass
+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
+
+
+@dataclass
+class PlayerOption:
+    full_name: str
+    gsis_id: str
+    headshot_url: str
+    position: str
+    team: str
+
+    @classmethod
+    def from_series(cls, input_series):
+        return cls(
+            full_name=input_series.full_name,
+            gsis_id=input_series.gsis_id,
+            headshot_url=input_series.headshot_url,
+            position=input_series.position,
+            team=input_series.team,
+        )
+
+    @classmethod
+    def empty_player(cls):
+        return cls(
+            full_name="",
+            gsis_id="",
+            headshot_url="",
+            position="",
+            team="",
+        )
+
+
+def player_options_from_df(df_options, position_filter: str) -> list[PlayerOption]:
+    empty_first_option_list = [PlayerOption.empty_player()]
+    return (
+        empty_first_option_list
+        + df_options[df_options.position == position_filter].apply(PlayerOption.from_series, axis=1).tolist()
+    )
+
+
+# @st.cache_data(ttl=60 * 60 * 24)
+def load_options():
+    # filter active only
+    df_rosters = get_weekly_rosters()
+    df_rosters = df_rosters[df_rosters.status == "ACT"]
+    # select latest
+    sort_by_cols = ["position", "week", "fantasy_points"]
+    df_rosters = df_rosters.sort_values(sort_by_cols, ascending=False).drop_duplicates(subset="gsis_id")
+    qb_options = player_options_from_df(df_rosters, "QB")
+    wr_options = player_options_from_df(df_rosters, "WR")
+    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
+
+
+def format_player_option(player_opt: PlayerOption) -> str:
+    return f"{player_opt.team} - {player_opt.full_name}"
+
+
+def display_player(player_opt: PlayerOption | None):
+    if player_opt:
+        if player_opt.headshot_url:
+            st.image(player_opt.headshot_url, caption=player_opt.full_name)
+
+
+def get_page():
+    page_title = "Select Your Team"
+    st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
+    common_page_config()
+    st.title(page_title)
+
+    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)
+
+
+if __name__ == "__main__":
+    get_page()