Map to playoff week in the league schedule

src/queries/pfr/league_schedule.py

@@ -1,5 +1,7 @@
 import pandas as pd
 
+from domain.playoffs import SCHEDULE_WEEK_TO_PLAYOFF_WEEK
+
 
 def get_full_schedule(season_int: str | int) -> pd.DataFrame:
     url = f"https://www.pro-football-reference.com/years/{season_int}/games.htm#games"
@@ -11,13 +13,12 @@ def get_full_schedule(season_int: str | int) -> pd.DataFrame:
 
 
 def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str, pd.Timestamp]]:
-    min_week = 1
-    max_week = 23
-    week_team_time_map: dict[int, dict[str, pd.Timestamp]] = {k: {} for k in range(min_week, max_week + 1)}
+    week_team_time_map: dict[int, dict[str, pd.Timestamp]] = {k: {} for k in SCHEDULE_WEEK_TO_PLAYOFF_WEEK.values()}
     for _, row in df_schedule.iterrows():
-        game_time = pd.to_datetime(row.Date + " " + row.Time, yearfirst=True)
-        week_team_time_map[int(row.Week)][row["Winner/tie"]] = game_time
-        week_team_time_map[int(row.Week)][row["Loser/tie"]] = game_time
+        if mapped_week := SCHEDULE_WEEK_TO_PLAYOFF_WEEK.get(row.Week):
+            game_time = pd.to_datetime(row.Date + " " + row.Time, yearfirst=True)
+            week_team_time_map[mapped_week][row["Winner/tie"]] = game_time
+            week_team_time_map[mapped_week][row["Loser/tie"]] = game_time
     return week_team_time_map