Implement game time in options

src/pages/10_Set_Your_Lineup.py

@@ -1,12 +1,16 @@
 from dataclasses import dataclass
+import pandas as pd
 import streamlit as st
 
 from config import DEFAULT_ICON
 from shared_page import common_page_config
 
+from domain.constants import SEASON
+from domain.playoffs import PLAYOFF_WEEK_TO_NAME, CURRENT_PLAYOFF_WEEK, PLAYOFF_WEEK_TO_SCHEDULE_WEEK
+from domain.teams import SCHEDULE_NAME_TO_PFR_NAME_MAP
 from queries.nflverse.github_data import get_weekly_rosters
+from queries.pfr.league_schedule import get_season_time_map
 from login import check_password, get_user_team, save_user_team
-from domain.playoffs import PLAYOFF_WEEK_TO_NAME, CURRENT_PLAYOFF_WEEK
 
 
 @dataclass
@@ -16,6 +20,7 @@ class PlayerOption:
     headshot_url: str
     position: str
     team: str
+    gametime: pd.Timestamp | None
 
     @classmethod
     def from_series(cls, input_series):
@@ -25,6 +30,7 @@ class PlayerOption:
             headshot_url=input_series.headshot_url,
             position=input_series.position,
             team=input_series.team,
+            gametime=input_series.gametime,
         )
 
     @classmethod
@@ -35,6 +41,7 @@ class PlayerOption:
             headshot_url="",
             position="",
             team="",
+            gametime=None,
         )
 
 
@@ -51,9 +58,20 @@ def load_options():
     # filter active only
     df_rosters = get_weekly_rosters()
     df_rosters = df_rosters[df_rosters.status == "ACT"]
+
+    # get current week games
+    schedule_week = PLAYOFF_WEEK_TO_SCHEDULE_WEEK[CURRENT_PLAYOFF_WEEK]
+    current_week_game_times = get_season_time_map(SEASON)[schedule_week]
+    latest_game_time = max(current_week_game_times.values())
+
     # 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")
+    # set gametime
+    df_rosters["gametime"] = df_rosters.apply(
+        lambda x: current_week_game_times.get(SCHEDULE_NAME_TO_PFR_NAME_MAP[x.team], latest_game_time), axis=1
+    )
+
     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")