Implement return td stats mapped both to player and team def

src/stats.py

@@ -227,6 +227,10 @@ YAHOO_TO_STAT_MAP: dict[str, dict[str, str]] = {
         "FUMBLE_RETURN_TOUCHDOWNS": DEF_TD.key,
         "SAFETIES": SAFETY.key,
     },
+    "RETURNING": {
+        "KICKOFF_RETURN_TOUCHDOWNS": ST_TD.key,
+        "PUNT_RETURN_TOUCHDOWNS": ST_TD.key,
+    },
 }
 
 
@@ -266,33 +270,49 @@ def add_yahoo_stat_type_to_stat_map(
         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}
+        elif yahoo_stat_type == "RETURNING":
+            week_leaders = week_dict["POSTSEASON"][""]["RETURN_YARDS_PER_KICKOFF"]["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:
+            def_player_id = ""
+            player_id = ""
             if yahoo_stat_type == "DEFENSE":
-                player_id = short_team_names_to_player_id[player["player"]["team"]["abbreviation"]]
+                def_player_id = short_team_names_to_player_id[player["player"]["team"]["abbreviation"]]
+            if yahoo_stat_type == "RETURNING":
+                raw_player_id = player["player"]["playerId"].split(".")[-1]
+                player_id = YAHOO_PLAYER_ID_MAP.get(raw_player_id, "")
+                def_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, "")
 
-            if not player_id:
-                continue
+            map_stats_to_week_player_id(player_id, week, player, stat_map, yahoo_stat_type)
+            map_stats_to_week_player_id(def_player_id, week, player, stat_map, yahoo_stat_type)
+
 
-            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"] or 0.0)
-                    else:
-                        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"]
+def map_stats_to_week_player_id(player_id: str, week: int, player, stat_map, yahoo_stat_type):
+    if not player_id:
+        return
+
+    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"] or 0.0)
+            else:
+                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"]
 
 
 def get_yahoo_stat_json_obj():
@@ -378,7 +398,7 @@ def get_yahoo_stats() -> dict[int, dict[str, dict[str, float]]]:
     add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballRushing"], "RUSHING", stat_map)
     add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballReceiving"], "RECEIVING", stat_map)
     add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballKicking"], "KICKING", stat_map)
-    # return_stats = stats_json["weeklyStatsFootballReturns"]
+    add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballReturns"], "RETURNING", stat_map)
     add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballDefense"], "DEFENSE", stat_map)
 
     return stat_map