main.py

import fastf1
import pandas as pd
from fastapi import FastAPI
import os





app = FastAPI()


@app.get("/")
async def root():
    return {"message": "Hello World"}

@app.get("/{year}/{race}/{session}", response_model=None)
async def get_data(year: int, race: int, session: str) -> any:

    if not os.path.exists('Cache'):  
        os.makedirs('Cache')

    

    fastf1.Cache.enable_cache('Cache')
    
    f1session = fastf1.get_session(year, race, session)
    f1session.load(telemetry=False, weather=False, messages=False)
    # Load all laps with telemetry
    laps = f1session.laps
    laps['Sector1Time'] = laps['Sector1Time'].dt.total_seconds()
    laps['Sector2Time'] = laps['Sector2Time'].dt.total_seconds()
    laps['Sector3Time'] = laps['Sector3Time'].dt.total_seconds()
    laps['LapTime_in_seconds'] = laps['LapTime'].dt.total_seconds()
    laps['laptime_sum_sectortimes'] = laps.Sector1Time + laps.Sector2Time + laps.Sector3Time      
    
    return {"laps": laps.to_json(orient='records')}