Load time slots based on separate sheet

src/maximum_roster_strategy/data_loader.py

@@ -6,6 +6,14 @@ MRS_SHEET_ID = os.environ.get("MRS_SHEET_ID")
 
 
 def get_google_sheet_data() -> pd.DataFrame:
-    sheet_url = f"https://docs.google.com/spreadsheet/ccc?key={MRS_SHEET_ID}&output=csv"
+    return get_sheet_data(0)
+
+
+def get_sheet_data(sheet_id: int = 0):
+    sheet_url = f"https://docs.google.com/spreadsheet/ccc?key={MRS_SHEET_ID}&output=csv&gid={sheet_id}"
     df = pd.read_csv(sheet_url)
     return df
+
+
+def get_timeslot_labels() -> pd.DataFrame:
+    return get_sheet_data(1875906423)

src/pages/80_Maximum_Roster_Strategy.py

@@ -24,9 +24,9 @@ POSITION_ABBR_FULL_NAME_MAP = {
 }
 
 
-@st.cache_data(ttl=60 * 60 * 24)
+@st.cache_data(ttl=60 * 60)
 def load_data():
-    return data_loader.get_google_sheet_data()
+    return data_loader.get_google_sheet_data(), data_loader.get_timeslot_labels()
 
 
 def get_player_grid_div(player_series: pd.Series) -> str:
@@ -73,11 +73,10 @@ def get_player_container(df_players: pd.DataFrame, slot_number: int | str) -> st
     return f"""<div class="playerslot{slot_number} playerslot">{player_code_str}</div>"""
 
 
-def get_position_breakdown(df: pd.DataFrame, position_abbr: str, position_full_str: str):
+def get_position_breakdown(df: pd.DataFrame, position_abbr: str, position_full_str: str, time_slots: list[str]):
     with st.container():
         st.header(position_full_str)
         df_pos = df[df["Position"] == position_abbr]
-        time_slots = df.sort_values("WeekTimeSlotIndex")["TimeSlotName"].unique().tolist()
 
         grid_code_str = ""
         grid_code_str += get_time_slot_div(time_slots)
@@ -113,10 +112,13 @@ def get_page():
         position = st.selectbox(label="Position", options=POSITION_OPTIONS, index=0)
     with week_select:
         week = st.selectbox(label="Week", options=list(range(MAXIMUM_WEEK, MINIMUM_WEEK - 1, -1)), index=0)
-    df_mrs = load_data()
+    df_mrs, all_time_slots_df = load_data()
     df_mrs = df_mrs[df_mrs["Week"] == week]
+    current_week_timeslots = (
+        all_time_slots_df[all_time_slots_df["Week"] == week].sort_values("WeekTimeSlotIndex").TimeSlotName.tolist()
+    )
 
-    get_position_breakdown(df_mrs, position, POSITION_ABBR_FULL_NAME_MAP[position])
+    get_position_breakdown(df_mrs, position, POSITION_ABBR_FULL_NAME_MAP[position], current_week_timeslots)
 
 
 if __name__ == "__main__":