Implement points against and team win stats

src/domain/playoffs.py

@@ -74,4 +74,6 @@ PLAYOFF_TEAM_DEF_PLAYER: list[tuple[teams.NFLTeam, str]] = [
 ]
 
 SHORT_TEAM_NAMES_TO_DEFENSE_PLAYER_ID = {t.team_short_name: p for t, p in PLAYOFF_TEAM_DEF_PLAYER}
+
 ROSTER_TEAM_NAMES_TO_DEFENSE_PLAYER_ID = {t.rosters_short_name: p for t, p in PLAYOFF_TEAM_DEF_PLAYER}
+DEFENSE_PLAYER_ID_TO_ROSTER_TEAM_NAMES = {v: k for k, v in ROSTER_TEAM_NAMES_TO_DEFENSE_PLAYER_ID.items()}

src/stats.py

@@ -7,6 +7,7 @@ import streamlit as st
 from domain.constants import SEASON
 from domain.playoffs import (
     SHORT_TEAM_NAMES_TO_DEFENSE_PLAYER_ID,
+    DEFENSE_PLAYER_ID_TO_ROSTER_TEAM_NAMES,
     ROSTER_TEAM_NAMES_TO_DEFENSE_PLAYER_ID,
     PLAYOFF_TEAM_DEF_PLAYER,
 )
@@ -435,8 +436,47 @@ def get_schedule_with_live() -> dict[int, dict[str, dict[str, str | int | pd.Tim
     return schedule
 
 
-def add_points_against_stat(stat_map):
-    pass
+def add_points_against_team_win_stat(stat_map: dict[int, dict[str, dict[str, float]]]):
+    schedule = get_schedule_with_live()
+
+    for week, week_map in stat_map.items():
+        for player_id in week_map.keys():
+            if team_short_name := DEFENSE_PLAYER_ID_TO_ROSTER_TEAM_NAMES.get(player_id):
+                team_game = schedule[week][team_short_name]
+                opponent_team = str(team_game["opponent"])
+                opponent_player_id = ROSTER_TEAM_NAMES_TO_DEFENSE_PLAYER_ID[opponent_team]
+                opponent_def_stats = week_map[opponent_player_id]
+
+                if isinstance((opponent_score_str := team_game["opponent_score"]), pd.Timestamp):
+                    # another case not possible but makes mypy happy
+                    continue
+                opponent_score = float(opponent_score_str)  # noqa
+
+                opponent_def_points_scored = (
+                    opponent_def_stats.get(TWO_PT.key, 0.0) * 2.0
+                    + opponent_def_stats.get(DEF_TD.key, 0.0) * 6.0
+                    + opponent_def_stats.get(ST_TD.key, 0.0) * 6.0
+                )
+                points_allowed = opponent_score - opponent_def_points_scored
+
+                if points_allowed == 0:
+                    stat_map[week][player_id].update({PTS_ALLOW_0.key: 1})
+                elif points_allowed < 7:
+                    stat_map[week][player_id].update({PTS_ALLOW_1_6.key: 1})
+                elif points_allowed < 14:
+                    stat_map[week][player_id].update({PTS_ALLOW_7_13.key: 1})
+                elif points_allowed < 21:
+                    stat_map[week][player_id].update({PTS_ALLOW_14_20.key: 1})
+                elif points_allowed < 28:
+                    stat_map[week][player_id].update({PTS_ALLOW_21_27.key: 1})
+                elif points_allowed < 35:
+                    stat_map[week][player_id].update({PTS_ALLOW_28_34.key: 1})
+                else:
+                    stat_map[week][player_id].update({PTS_ALLOW_35_.key: 1})
+
+                # check for win
+                if team_game["status"] == "Win":
+                    stat_map[week][player_id].update({TEAM_WIN.key: 1})
 
 
 @st.cache_data(ttl=STAT_CACHE_SECONDS)
@@ -452,7 +492,7 @@ def get_stats_map() -> dict[int, dict[str, dict[str, float]]]:
         for player_id, player_stats in week_stats.items():
             stat_map[week][player_id] = player_stats
 
-    add_points_against_stat(stat_map)
+    add_points_against_team_win_stat(stat_map)
     stat_overrides = get_stat_overrides()
     # for stat overrides, override at the stat level
     for week, week_stats in stat_overrides.items():