Filter to id map to players on teams in playoffs

src/domain/teams.py

@@ -12,6 +12,7 @@ class NFLTeam:
     city: str
     division: divisions.NFLDivision
     footballguys_short_name: str = ""
+    id_map_short_name: str = ""
     rosters_short_name: str = ""
     url: str = ""
     injury_report_suffix: str = "/team/injury-report/"
@@ -20,6 +21,7 @@ class NFLTeam:
         self.footballguys_short_name = (
             self.team_short_name if self.footballguys_short_name == "" else self.footballguys_short_name
         )
+        self.id_map_short_name = self.team_short_name if self.id_map_short_name == "" else self.id_map_short_name
         self.rosters_short_name = self.team_short_name if self.rosters_short_name == "" else self.rosters_short_name
         self.conference: NFLConference = self.division.conference
         self.injury_report_url = urljoin(self.url, self.injury_report_suffix)
@@ -125,6 +127,7 @@ green_bay_packers = NFLTeam(
     city="Green Bay",
     division=divisions.NFCNorth,
     url="https://www.packers.com/",
+    id_map_short_name="GBP",
 )
 
 houston_texans = NFLTeam(
@@ -149,6 +152,7 @@ jacksonville_jaguars = NFLTeam(
     team_short_name="JAX",
     division=divisions.AFCSouth,
     url="https://www.jaguars.com/",
+    id_map_short_name="JAC",
 )
 
 kansas_city_chiefs = NFLTeam(
@@ -157,6 +161,7 @@ kansas_city_chiefs = NFLTeam(
     team_short_name="KC",
     division=divisions.AFCWest,
     url="https://www.chiefs.com/",
+    id_map_short_name="KCC",
 )
 
 las_vegas_raiders = NFLTeam(
@@ -165,6 +170,7 @@ las_vegas_raiders = NFLTeam(
     team_short_name="LV",
     division=divisions.AFCWest,
     url="https://www.raiders.com/",
+    id_map_short_name="LVR",
 )
 
 los_angeles_chargers = NFLTeam(
@@ -206,6 +212,7 @@ new_england_patriots = NFLTeam(
     team_short_name="NE",
     division=divisions.AFCEast,
     url="https://www.patriots.com/",
+    id_map_short_name="NEP",
 )
 
 new_orleans_saints = NFLTeam(
@@ -214,6 +221,7 @@ new_orleans_saints = NFLTeam(
     team_short_name="NO",
     division=divisions.NFCSouth,
     url="https://www.neworleanssaints.com/",
+    id_map_short_name="NOS",
 )
 
 new_york_giants = NFLTeam(
@@ -254,6 +262,7 @@ san_francisco_49ers = NFLTeam(
     team_short_name="SF",
     division=divisions.NFCWest,
     url="https://www.49ers.com/",
+    id_map_short_name="SFO",
 )
 
 seattle_seahawks = NFLTeam(
@@ -270,6 +279,7 @@ tampa_bay_buccaneers = NFLTeam(
     team_short_name="TB",
     division=divisions.NFCSouth,
     url="https://www.buccaneers.com/",
+    id_map_short_name="TBB",
 )
 
 tennessee_titans = NFLTeam(

src/stats.py

@@ -232,8 +232,9 @@ YAHOO_TO_STAT_MAP: dict[str, dict[str, str]] = {
 # cache id map for 24 hours
 @st.cache_data(ttl=60 * 60 * 24)
 def get_yahoo_id_map() -> dict[str, str]:
+    teams_included = [x.id_map_short_name for x, _ in PLAYOFF_TEAM_DEF_PLAYER]
     df = pd.read_csv(r"https://raw.githubusercontent.com/dynastyprocess/data/master/files/db_playerids.csv")
-    df = df[(df["yahoo_id"].notna() & df["gsis_id"].notna())]
+    df = df[(df["yahoo_id"].notna() & df["gsis_id"].notna() & df["team"].isin(teams_included))]
     df["yahoo_id"] = df["yahoo_id"].astype(int).astype(str)
     return df.set_index("yahoo_id")["gsis_id"].to_dict()
 
@@ -269,7 +270,10 @@ def add_yahoo_stat_type_to_stat_map(
                 player_id = short_team_names_to_player_id[player["player"]["team"]["abbreviation"]]
             else:
                 raw_player_id = player["player"]["playerId"].split(".")[-1]
-                player_id = get_yahoo_id_map().get(raw_player_id, raw_player_id)
+                player_id = get_yahoo_id_map().get(raw_player_id)
+
+            if not player_id:
+                continue
 
             if player_id not in stat_map[week]:
                 stat_map[week][player_id] = {}