Add live passing, rushing, and receiving stats

src/stats.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
+import json
 import pandas as pd
+import requests
 import streamlit as st
 
 from domain.playoffs import PLAYOFF_TEAM_DEF_PLAYER
@@ -187,8 +189,107 @@ def assemble_nflverse_stats() -> dict[int, dict[str, dict[str, float]]]:
 
 
 def get_live_stats() -> dict[int, dict[str, dict[str, float]]]:
+    return get_yahoo_stats()
+
+
+YAHOO_TO_STAT_MAP = {
+    "PASSING": {
+        "PASSING_YARDS": PASS_YD.key,
+        "PASSING_TOUCHDOWNS": PASS_TD.key,
+        "PASSING_INTERCEPTIONS": PASS_INT.key,
+        "FUMBLES_LOST": FUM_LOST.key,
+    },
+    "RUSHING": {
+        "RUSHING_TOUCHDOWNS": RUSH_TD.key,
+        "FUMBLES_LOST": FUM_LOST.key,
+        "RUSHING_YARDS": RUSH_YD.key,
+    },
+    "RECEIVING": {
+        "RECEPTIONS": RECEPTION.key,
+        "RECEIVING_YARDS": REC_YD.key,
+        "RECEIVING_TOUCHDOWNS": REC_TD.key,
+        "FUMBLES_LOST": FUM_LOST.key,
+    },
+}
+
+
+# HACK find permanent source of this map
+YAHOO_PLAYER_ID_MAP = {
+    "30123": "00-0033873",  # Patrick Mahomes
+    "28654": "00-0031687",  # Raheem Mostert
+    "29399": "00-0033040",  # Tyreek
+    "31093": "00-0034798",  # Durham Smythe
+    "31199": "00-0034794",  # Jason Sanders
+    "30142": "00-0033885",  # David Njoku
+    "29792": "00-0032726",  # Fairbairn
+    "34207": "00-0037197",  # Pacheco
+    "26686": "00-0030506",  # Kelce
+    "30346": "00-0033303",  # Butker
+}
+
+# happens to be the same
+YAHOO_WEEK_MAP = NFLVERSE_STAT_WEEK_TO_PLAYOFF_WEEK
+
+
+def add_yahoo_stat_type_to_stat_map(
+    stats_object, yahoo_stat_type: str, stat_map: dict[int, dict[str, dict[str, float]]]
+):
+    assert yahoo_stat_type in YAHOO_TO_STAT_MAP
+
+    for raw_week, week_dict in stats_object.items():
+        week = YAHOO_WEEK_MAP[int(raw_week)]
+        if week not in stat_map:
+            stat_map[week] = {}
+
+        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 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"]):
+                    stat_map[week][player_id][stat_key] = float(stat["value"])
+                # else:
+                #     # remove after mapping all intended
+                #     stat_map[week][player_id][stat["statId"]] = stat["value"]
+
+
+def get_yahoo_stat_json_obj():
+    url = "https://sports.yahoo.com/nfl/stats/weekly/?selectedTable=0"
+    request = requests.get(url)
+    request_content_str = request.text
+
+    start_str = """root.App.main = """
+    end_str = """;\n}(this));"""
+
+    start_slice_pos = request_content_str.find(start_str) + len(start_str)
+    first_slice = request_content_str[start_slice_pos:]
+    end_slice_pos = first_slice.find(end_str)
+    dom_str = first_slice[:end_slice_pos]
+    dom_json = json.loads(dom_str)
+    return dom_json
+
+
+def get_yahoo_stats() -> dict[int, dict[str, dict[str, float]]]:
+    dom_json = get_yahoo_stat_json_obj()
     stat_map: dict[int, dict[str, dict[str, float]]] = {w: {} for w in NFLVERSE_STAT_WEEK_TO_PLAYOFF_WEEK.values()}
-    # TODO - implement live stats
+
+    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
+    )
+    kicking_stats = stats_json["weeklyStatsFootballKicking"]["nfl"]["200"]["2023"]
+    return_stats = stats_json["weeklyStatsFootballReturns"]["nfl"]["200"]["2023"]
+    defense_stats = stats_json["weeklyStatsFootballDefense"]["nfl"]["200"]["2023"]
+
     return stat_map