tracinginsights commited on
Commit
f6a40e3
·
1 Parent(s): 3f5181c

faster drivers

Browse files
Files changed (1) hide show
  1. main.py +11 -10
main.py CHANGED
@@ -71,16 +71,17 @@ def sessions_available(year: int, event: str | int) -> any:
71
  def session_drivers(year: int, event: str | int, session: str) -> any:
72
  # get drivers available for a given year, event and session
73
  f1session = fastf1.get_session(year, event, session)
74
- f1session.load(telemetry=False, weather=False, messages=False)
75
- laps = f1session.laps
76
- team_colors = available_data.team_colors(year)
77
- # add team_colors dict to laps on Team column
78
-
79
- drivers = laps.Driver.unique()
80
- # for each driver in drivers, get the Team column from laps and get the color from team_colors dict
81
- drivers = [{"color": team_colors[laps[laps.Driver ==
82
- driver].Team.iloc[0]], "label": driver, "value": driver} for driver in drivers]
83
-
 
84
  return {"drivers": drivers}
85
 
86
 
 
71
  def session_drivers(year: int, event: str | int, session: str) -> any:
72
  # get drivers available for a given year, event and session
73
  f1session = fastf1.get_session(year, event, session)
74
+ api_path = f1session.api_path
75
+ drivers_raw = fastf1.api.driver_info(api_path)
76
+ drivers = []
77
+ for driver in drivers_raw.items():
78
+ drivers.append({
79
+ "color": available_data.team_colors(year)[driver[1]['TeamName']],
80
+
81
+ "label": driver[1]['Tla'],
82
+ "value": driver[1]['Tla']})
83
+
84
+
85
  return {"drivers": drivers}
86
 
87