Update pages/Pit_Stops.py

pages/Pit_Stops.py

@@ -7,51 +7,37 @@ YEAR_SELECTED = st.selectbox(
     'Select Year',
     utils.get_years())
 
-
-
-season_events, events_list = PitStops.get_season_events(YEAR_SELECTED)
-
 RACE_SELECTED =  st.selectbox(
     'Select Race',
-    events_list)
+    utils.get_events(YEAR_SELECTED))
 
-event_id = PitStops.get_event_id(season_events, RACE_SELECTED)
-df = PitStops.get_pitstops(event_id) #dhl pitstops
+df = PitStops.get_pitstops(RACE_SELECTED, YEAR_SELECTED) #dhl pitstops
 
 try:
     PitStops.plot_event_pitstops(df, RACE_SELECTED)
 except:
     pass
 
-
 try:    
     PitStops.plot_full_season_median(YEAR_SELECTED)
 except:
     pass
-try:
-
-    race_names_df,  pit_stops_df, drivers_df =  PitStops.load_data()
     
-    event_date = PitStops.get_event_date(season_events, RACE_SELECTED)
+try:
+    race_names_df,  pit_stops_df, drivers_df =  PitStops.load_data() 
     
-    ergast_pitstops, grandprix = PitStops.get_pitstops_by_date(pit_stops_df,drivers_df,race_names_df,event_date)
+    ergast_pitstops, grandprix = PitStops.get_ergast_pitstops(pit_stops_df, drivers_df, race_names_df, YEAR_SELECTED, RACE_SELECTED)
     
     df_agg = PitStops.combine_dfs(ergast_pitstops, df)
 except:
     pass
 
-
-
 try:    
     PitStops.plot_event_ratings(df_agg, grandprix)
 except:
     pass
 
 
-try:
-    PitStops.plot_event_ratings(df_agg, grandprix)
-except:
-    pass