Move playoff team definition to playoffs.py

src/domain/playoffs.py

@@ -1,3 +1,6 @@
+from domain import teams
+
+
 PLAYOFF_WEEK_TO_SCHEDULE_WEEK = {
     1: "WildCard",
     2: "Division",
@@ -26,3 +29,43 @@ PLAYOFF_WEEK_TO_NAME = {
 }
 
 CURRENT_PLAYOFF_WEEK = 1
+
+
+PLAYOFFS_TEAMS = {
+    1: [
+        teams.baltimore_ravens.rosters_short_name,
+        teams.miami_dolphins.rosters_short_name,
+        teams.kansas_city_chiefs.rosters_short_name,
+        teams.houston_texans.rosters_short_name,
+        teams.cleveland_browns.rosters_short_name,
+        teams.san_francisco_49ers.rosters_short_name,
+        teams.dallas_cowboys.rosters_short_name,
+        teams.detroit_lions.rosters_short_name,
+        teams.philadelphia_eagles.rosters_short_name,
+        teams.los_angeles_rams.rosters_short_name,
+        teams.tampa_bay_buccaneers.rosters_short_name,
+        teams.buffalo_bills.rosters_short_name,
+        teams.pittsburgh_steelers.rosters_short_name,
+        teams.green_bay_packers.rosters_short_name,
+    ],
+    2: [],
+    3: [],
+    4: [],
+}
+
+PLAYOFF_TEAM_DEF_PLAYER: list[tuple[teams.NFLTeam, str]] = [
+    (teams.baltimore_ravens, ""),
+    (teams.miami_dolphins, ""),
+    (teams.kansas_city_chiefs, ""),
+    (teams.houston_texans, ""),
+    (teams.cleveland_browns, ""),
+    (teams.san_francisco_49ers, ""),
+    (teams.dallas_cowboys, ""),
+    (teams.detroit_lions, ""),
+    (teams.philadelphia_eagles, ""),
+    (teams.los_angeles_rams, ""),
+    (teams.tampa_bay_buccaneers, ""),
+    (teams.buffalo_bills, ""),
+    (teams.pittsburgh_steelers, ""),
+    (teams.green_bay_packers, ""),
+]

src/domain/teams.py

@@ -289,27 +289,4 @@ washington_football_team = NFLTeam(
 )
 
 
-PLAYOFFS_TEAMS = {
-    1: [
-        baltimore_ravens.rosters_short_name,
-        miami_dolphins.rosters_short_name,
-        kansas_city_chiefs.rosters_short_name,
-        houston_texans.rosters_short_name,
-        cleveland_browns.rosters_short_name,
-        san_francisco_49ers.rosters_short_name,
-        dallas_cowboys.rosters_short_name,
-        detroit_lions.rosters_short_name,
-        philadelphia_eagles.rosters_short_name,
-        los_angeles_rams.rosters_short_name,
-        tampa_bay_buccaneers.rosters_short_name,
-        buffalo_bills.rosters_short_name,
-        pittsburgh_steelers.rosters_short_name,
-        green_bay_packers.rosters_short_name,
-    ],
-    2: [],
-    3: [],
-    4: [],
-}
-
-
 SCHEDULE_NAME_TO_PFR_NAME_MAP = {t.rosters_short_name: t.team_full_name for t in ALL_TEAMS}

src/pages/10_Set_Your_Lineup.py

@@ -7,8 +7,8 @@ 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, ROSTER_WEEK_TO_PLAYOFF_WEEK
-from domain.teams import SCHEDULE_NAME_TO_PFR_NAME_MAP, PLAYOFFS_TEAMS
+from domain.playoffs import PLAYOFF_WEEK_TO_NAME, CURRENT_PLAYOFF_WEEK, ROSTER_WEEK_TO_PLAYOFF_WEEK, PLAYOFFS_TEAMS
+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
@@ -89,7 +89,7 @@ def player_options_from_df(df_options) -> dict[str, dict[int, list[PlayerOption]
     return options_map
 
 
-@st.cache_data(ttl=60 * 60 * 24)
+# @st.cache_data(ttl=60 * 60 * 24)
 def load_options():
     df_rosters = get_weekly_rosters()
 
@@ -117,6 +117,8 @@ def load_options():
     df_rosters["in_playoffs"] = df_rosters.apply(lambda x: x.team in PLAYOFFS_TEAMS[x.week], axis=1)
 
     df_rosters = df_rosters[df_rosters.in_playoffs]
+    # st.dataframe(df_rosters)
+    st.write(df_rosters.position.unique())
     player_options = player_options_from_df(df_rosters)
     return player_options