tracinginsights commited on
Commit
4cc9fb2
·
1 Parent(s): e742cfd

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +9 -7
main.py CHANGED
@@ -84,23 +84,25 @@ async def root():
84
 
85
  @app.get("/{year}/{race}/{session}", response_model=None)
86
  async def get_data(year: int, race: int, session: str) -> any:
 
87
  f1session = fastf1.get_session(year, race, session)
88
  f1session.load(telemetry=False, weather=False, messages=False)
89
  # Load all laps with telemetry
90
  laps = f1session.laps
91
- # laps['Sector1Time'] = laps['Sector1Time'].dt.total_seconds()
92
- # laps['Sector2Time'] = laps['Sector2Time'].dt.total_seconds()
93
- # laps['Sector3Time'] = laps['Sector3Time'].dt.total_seconds()
94
- # laps['LapTime_in_seconds'] = laps['LapTime'].dt.total_seconds()
95
- # laps['laptime_sum_sectortimes'] = laps.Sector1Time + \
96
- # laps.Sector2Time + laps.Sector3Time
97
 
98
  # # convert laps to csv
99
  # laps.to_csv('laps.csv', index=False)
100
  # return laps
101
 
102
  # in laps replace NaN with ""
103
- laps = laps.fillna("")
 
104
  return {"laps": laps.to_dict(orient='records')}
105
 
106
  # return FileResponse(path='laps.csv', filename='laps.csv')
 
84
 
85
  @app.get("/{year}/{race}/{session}", response_model=None)
86
  async def get_data(year: int, race: int, session: str) -> any:
87
+
88
  f1session = fastf1.get_session(year, race, session)
89
  f1session.load(telemetry=False, weather=False, messages=False)
90
  # Load all laps with telemetry
91
  laps = f1session.laps
92
+ laps['Sector1Time'] = laps['Sector1Time'].dt.total_seconds()
93
+ laps['Sector2Time'] = laps['Sector2Time'].dt.total_seconds()
94
+ laps['Sector3Time'] = laps['Sector3Time'].dt.total_seconds()
95
+ laps['LapTime_in_seconds'] = laps['LapTime'].dt.total_seconds()
96
+ laps['laptime_sum_sectortimes'] = laps.Sector1Time + \
97
+ laps.Sector2Time + laps.Sector3Time
98
 
99
  # # convert laps to csv
100
  # laps.to_csv('laps.csv', index=False)
101
  # return laps
102
 
103
  # in laps replace NaN with ""
104
+ laps = laps.fillna(" ")
105
+
106
  return {"laps": laps.to_dict(orient='records')}
107
 
108
  # return FileResponse(path='laps.csv', filename='laps.csv')