Jon Solow commited on
Commit
9ac79a9
·
1 Parent(s): 72cd17c

Enable stat loads for regular season

Browse files
Files changed (1) hide show
  1. src/stats.py +7 -4
src/stats.py CHANGED
@@ -262,6 +262,7 @@ def get_yahoo_id_map() -> dict[str, str]:
262
  YAHOO_PLAYER_ID_MAP = get_yahoo_id_map()
263
 
264
  YAHOO_WEEK_MAP = {
 
265
  19: 1,
266
  20: 2,
267
  21: 3,
@@ -278,19 +279,21 @@ def add_yahoo_stat_type_to_stat_map(
278
 
279
  for raw_week, week_dict in nfl_object.items():
280
  week = YAHOO_WEEK_MAP[int(raw_week)]
 
281
  if week not in stat_map:
282
  stat_map[week] = {}
283
 
 
284
  if yahoo_stat_type == "KICKING":
285
- week_leaders = week_dict["POSTSEASON"][""]["FIELD_GOALS_MADE"]["leagues"][0]["leagueWeeks"][0]["leaders"]
286
  elif yahoo_stat_type == "DEFENSE":
287
- week_leaders = week_dict["POSTSEASON"][""]["TOTAL_TACKLES"]["leagues"][0]["leagueWeeks"][0]["leaders"]
288
  elif yahoo_stat_type == "RETURNING":
289
- week_leaders = week_dict["POSTSEASON"][""]["RETURN_YARDS_PER_KICKOFF"]["leagues"][0]["leagueWeeks"][0][
290
  "leaders"
291
  ]
292
  else:
293
- week_leaders = week_dict["POSTSEASON"][""][f"{yahoo_stat_type}_YARDS"]["leagues"][0]["leagueWeeks"][0][
294
  "leaders"
295
  ]
296
 
 
262
  YAHOO_PLAYER_ID_MAP = get_yahoo_id_map()
263
 
264
  YAHOO_WEEK_MAP = {
265
+ 17: 1,
266
  19: 1,
267
  20: 2,
268
  21: 3,
 
279
 
280
  for raw_week, week_dict in nfl_object.items():
281
  week = YAHOO_WEEK_MAP[int(raw_week)]
282
+ print(week_dict)
283
  if week not in stat_map:
284
  stat_map[week] = {}
285
 
286
+ season_type = "REGULAR_SEASON" if int(raw_week) < 18 else "POSTSEASON"
287
  if yahoo_stat_type == "KICKING":
288
+ week_leaders = week_dict[season_type][""]["FIELD_GOALS_MADE"]["leagues"][0]["leagueWeeks"][0]["leaders"]
289
  elif yahoo_stat_type == "DEFENSE":
290
+ week_leaders = week_dict[season_type][""]["TOTAL_TACKLES"]["leagues"][0]["leagueWeeks"][0]["leaders"]
291
  elif yahoo_stat_type == "RETURNING":
292
+ week_leaders = week_dict[season_type][""]["RETURN_YARDS_PER_KICKOFF"]["leagues"][0]["leagueWeeks"][0][
293
  "leaders"
294
  ]
295
  else:
296
+ week_leaders = week_dict[season_type][""][f"{yahoo_stat_type}_YARDS"]["leagues"][0]["leagueWeeks"][0][
297
  "leaders"
298
  ]
299