Get stat overrides from google sheet

src/login.py

@@ -148,3 +148,23 @@ def get_logged_in_user_name_email() -> tuple[str | None, str | None]:
     email = user_info_map.get("email")
     name = user_info_map.get("name")
     return email, name
+
+
+# hack - data overrides for stats allowing for quick manual edits
+def get_stat_overrides() -> dict[int, dict[str, dict[str, float]]]:
+    df = conn.read(
+        worksheet="stats-overrides",
+        ttl="10m",
+        usecols=[0, 1, 2, 3],
+    )
+    stat_overrides_map: dict[int, dict[str, dict[str, float]]] = {w: {} for w in df.week.values}
+
+    for week, df_week in df.groupby("week"):
+        for player_id, df_player in df_week.groupby("player_id"):
+            if player_id not in stat_overrides_map[week]:
+                stat_overrides_map[week][player_id] = {}
+            for row in df_player.itertuples():
+                if row.stat_key and row.stat_value:
+                    stat_overrides_map[week][player_id][row.stat_key] = float(row.stat_value)
+
+    return stat_overrides_map

src/stats.py

@@ -3,6 +3,7 @@ import pandas as pd
 import streamlit as st
 
 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
 
 
@@ -37,13 +38,13 @@ DEF_INT = StatType(key="DEF INT", score=2.0)
 FUM_REC = StatType(key="FUM REC", score=2.0)
 SAFETY = StatType(key="SAFETY", score=2.0)
 SACK = StatType(key="SACK", score=1.0)
-PTS_ALLOW_0 = StatType(key="PTS ALLOW 0", score=10.0)
-PTS_ALLOW_1_6 = StatType(key="PTS ALLOW 1-6", score=7.0)
-PTS_ALLOW_7_13 = StatType(key="PTS ALLOW 7-13", score=4.0)
-PTS_ALLOW_14_20 = StatType(key="PTS ALLOW 14-20", score=1.0)
-PTS_ALLOW_21_27 = StatType(key="PTS ALLOW 21-27", score=0.0)
-PTS_ALLOW_28_34 = StatType(key="PTS ALLOW 28-34", score=-1.0)
-PTS_ALLOW_35_ = StatType(key="PTS ALLOW 35+", score=-4.0)
+PTS_ALLOW_0 = StatType(key="PTS 0", score=10.0)
+PTS_ALLOW_1_6 = StatType(key="PTS 1-6", score=7.0)
+PTS_ALLOW_7_13 = StatType(key="PTS 7-13", score=4.0)
+PTS_ALLOW_14_20 = StatType(key="PTS 14-20", score=1.0)
+PTS_ALLOW_21_27 = StatType(key="PTS 21-27", score=0.0)
+PTS_ALLOW_28_34 = StatType(key="PTS 28-34", score=-1.0)
+PTS_ALLOW_35_ = StatType(key="PTS 35+", score=-4.0)
 TEAM_WIN = StatType(key="TEAM WIN", score=5.0)
 ST_TD = StatType(key="ST TD", score=6.0)
 
@@ -202,6 +203,15 @@ def get_stats_map() -> dict[int, dict[str, dict[str, float]]]:
         for player_id, player_stats in week_stats.items():
             stat_map[week][player_id] = player_stats
 
+    stat_overrides = get_stat_overrides()
+    # for stat overrides, override at the stat level
+    for week, week_stats in stat_overrides.items():
+        for player_id, player_stats in week_stats.items():
+            for stat_key, stat_value in player_stats.items():
+                if player_id not in stat_map[week]:
+                    stat_map[week][player_id] = {}
+                stat_map[week][player_id][stat_key] = stat_value
+
     return stat_map