Implement ST TD for defenses

src/stats.py

@@ -137,6 +137,33 @@ def add_stats_from_team_def_df_to_stat_map(df: pd.DataFrame, stat_map):
             stat_map[week][player_id].update(player_stats)
 
 
+def add_st_stats_to_defense(df: pd.DataFrame, stat_map):
+    short_team_names_to_player_id = {t.rosters_short_name: p for t, p in PLAYOFF_TEAM_DEF_PLAYER}
+    df_playoffs = df[
+        (df.week.isin(NFLVERSE_STAT_WEEK_TO_PLAYOFF_WEEK.keys()) & df.team.isin(short_team_names_to_player_id.keys()))
+    ]
+    df_playoffs.week = df_playoffs.week.apply(lambda x: NFLVERSE_STAT_WEEK_TO_PLAYOFF_WEEK[x])
+
+    for week_team_tuple, row in df_playoffs.set_index(["week", "team"]).iterrows():
+        if isinstance(week_team_tuple, tuple):
+            week, team = week_team_tuple
+        else:
+            # this won't happen but makes mypy happy
+            continue
+        player_id = short_team_names_to_player_id[team]
+        player_stats: dict[str, float] = stat_map[week].get(player_id, {})
+
+        # special teams td update only
+        for k, v in row.to_dict().items():
+            if k == "special_teams_tds":
+                if (mapped_k := NFLVERSE_STAT_COL_TO_ID[k]) in player_stats:
+                    player_stats[mapped_k] += v
+                else:
+                    player_stats[mapped_k] = v
+
+        stat_map[week][player_id] = player_stats
+
+
 # 24 hour cache
 @st.cache_data(ttl=60 * 60 * 24)
 def assemble_nflverse_stats() -> dict[int, dict[str, dict[str, float]]]:
@@ -150,6 +177,7 @@ def assemble_nflverse_stats() -> dict[int, dict[str, dict[str, float]]]:
     add_stats_from_player_df_to_stat_map(df_player_stats, stat_map)
     add_stats_from_player_df_to_stat_map(df_kicking_stats, stat_map)
     add_stats_from_team_def_df_to_stat_map(df_def_stats, stat_map)
+    add_st_stats_to_defense(df_player_stats, stat_map)
 
     return stat_map