Implement defensive live stats

src/stats.py

@@ -192,7 +192,7 @@ def get_live_stats() -> dict[int, dict[str, dict[str, float]]]:
     return get_yahoo_stats()
 
 
-YAHOO_TO_STAT_MAP = {
+YAHOO_TO_STAT_MAP: dict[str, dict[str, str]] = {
     "PASSING": {
         "PASSING_YARDS": PASS_YD.key,
         "PASSING_TOUCHDOWNS": PASS_TD.key,
@@ -218,6 +218,14 @@ YAHOO_TO_STAT_MAP = {
         "FIELD_GOALS_MADE_50_PLUS": FG_50_.key,
         "EXTRA_POINTS_MADE": XP.key,
     },
+    "DEFENSE": {
+        "SACKS": SACK.key,
+        "INTERCEPTIONS_FORCED": DEF_INT.key,
+        "INTERCEPTION_RETURN_TOUCHDOWNS": DEF_TD.key,
+        "FORCED_FUMBLES": FUM_REC.key,
+        "FUMBLE_RETURN_TOUCHDOWNS": DEF_TD.key,
+        "SAFETIES": SAFETY.key,
+    },
 }
 
 
@@ -249,25 +257,34 @@ def add_yahoo_stat_type_to_stat_map(
         if week not in stat_map:
             stat_map[week] = {}
 
+        # only used for defense summary
+        short_team_names_to_player_id = {}
         if yahoo_stat_type == "KICKING":
             week_leaders = week_dict["POSTSEASON"][""]["FIELD_GOALS_MADE"]["leagues"][0]["leagueWeeks"][0]["leaders"]
+        elif yahoo_stat_type == "DEFENSE":
+            week_leaders = week_dict["POSTSEASON"][""]["TOTAL_TACKLES"]["leagues"][0]["leagueWeeks"][0]["leaders"]
+            short_team_names_to_player_id = {t.rosters_short_name: p for t, p in PLAYOFF_TEAM_DEF_PLAYER}
         else:
             week_leaders = week_dict["POSTSEASON"][""][f"{yahoo_stat_type}_YARDS"]["leagues"][0]["leagueWeeks"][0][
                 "leaders"
             ]
 
         for player in week_leaders:
-            raw_player_id = player["player"]["playerId"].split(".")[-1]
-            player_id = YAHOO_PLAYER_ID_MAP.get(raw_player_id, raw_player_id)
+            if yahoo_stat_type == "DEFENSE":
+                player_id = short_team_names_to_player_id[player["player"]["team"]["abbreviation"]]
+            else:
+                raw_player_id = player["player"]["playerId"].split(".")[-1]
+                player_id = YAHOO_PLAYER_ID_MAP.get(raw_player_id, raw_player_id)
+
             if player_id not in stat_map[week]:
                 stat_map[week][player_id] = {}
             stats = player["stats"]
             for stat in stats:
                 if stat_key := YAHOO_TO_STAT_MAP[yahoo_stat_type].get(stat["statId"]):
                     if stat_key in stat_map[week][player_id]:
-                        stat_map[week][player_id][stat_key] += float(stat["value"])
+                        stat_map[week][player_id][stat_key] += float(stat["value"] or 0.0)
                     else:
-                        stat_map[week][player_id][stat_key] = float(stat["value"])
+                        stat_map[week][player_id][stat_key] = float(stat["value"] or 0.0)
                 # else:
                 #     # remove after mapping all intended
                 #     stat_map[week][player_id][stat["statId"]] = stat["value"]
@@ -302,7 +319,7 @@ def get_yahoo_stats() -> dict[int, dict[str, dict[str, float]]]:
     )
     add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballKicking"]["nfl"]["200"]["2023"], "KICKING", stat_map)
     return_stats = stats_json["weeklyStatsFootballReturns"]["nfl"]["200"]["2023"]
-    defense_stats = stats_json["weeklyStatsFootballDefense"]["nfl"]["200"]["2023"]
+    add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballDefense"]["nfl"]["200"]["2023"], "DEFENSE", stat_map)
 
     return stat_map