Reduce repeated code in yahoo_stat obj

src/stats.py

@@ -4,6 +4,7 @@ import pandas as pd
 import requests
 import streamlit as st
 
+from domain.constants import SEASON
 from domain.playoffs import PLAYOFF_TEAM_DEF_PLAYER
 from login import get_stat_overrides
 from queries.nflverse.github_data import get_player_kicking_stats, get_player_stats, get_team_defense_stats
@@ -251,7 +252,9 @@ def add_yahoo_stat_type_to_stat_map(
 ):
     assert yahoo_stat_type in YAHOO_TO_STAT_MAP
 
-    for raw_week, week_dict in stats_object.items():
+    nfl_object = stats_object["nfl"]["200"][f"{SEASON}"]
+
+    for raw_week, week_dict in nfl_object.items():
         week = YAHOO_WEEK_MAP[int(raw_week)]
         if week not in stat_map:
             stat_map[week] = {}
@@ -371,14 +374,12 @@ def get_yahoo_stats() -> dict[int, dict[str, dict[str, float]]]:
 
     stats_json = dom_json["context"]["dispatcher"]["stores"]["GraphStatsStore"]
 
-    add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballPassing"]["nfl"]["200"]["2023"], "PASSING", stat_map)
-    add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballRushing"]["nfl"]["200"]["2023"], "RUSHING", stat_map)
-    add_yahoo_stat_type_to_stat_map(
-        stats_json["weeklyStatsFootballReceiving"]["nfl"]["200"]["2023"], "RECEIVING", stat_map
-    )
-    add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballKicking"]["nfl"]["200"]["2023"], "KICKING", stat_map)
-    return_stats = stats_json["weeklyStatsFootballReturns"]["nfl"]["200"]["2023"]
-    add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballDefense"]["nfl"]["200"]["2023"], "DEFENSE", stat_map)
+    add_yahoo_stat_type_to_stat_map(stats_json["weeklyStatsFootballPassing"], "PASSING", stat_map)
+    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["weeklyStatsFootballDefense"], "DEFENSE", stat_map)
 
     return stat_map