Draichi commited on
Commit
4a2f537
·
unverified ·
1 Parent(s): 3591e00

fix(formula1_databases.py): update Laps table to not have NaT values

Browse files
Files changed (1) hide show
  1. notebooks/formula1_databases.py +20 -0
notebooks/formula1_databases.py CHANGED
@@ -425,6 +425,26 @@ class FastF1ToSQL:
425
  (self._session_id, driver_name, lap['LapNumber']))
426
  return self.cursor.fetchone()[0]
427
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
428
  def __create_data_analysis_views(self) -> None:
429
  """Create data analysis views in the database."""
430
  console.print('> Creating data analysis views...')
 
425
  (self._session_id, driver_name, lap['LapNumber']))
426
  return self.cursor.fetchone()[0]
427
 
428
+ def __update_laps(self) -> None:
429
+ """Update the laps table with the new data."""
430
+ console.print('> Updating laps table...')
431
+ self.cursor.execute('''
432
+ UPDATE Laps
433
+ SET
434
+ pin_in_time_in_datetime = CASE
435
+ WHEN pin_in_time_in_datetime = 'NaT' THEN NULL
436
+ ELSE pin_in_time_in_datetime
437
+ END,
438
+ pin_out_time_in_datetime = CASE
439
+ WHEN pin_out_time_in_datetime = 'NaT' THEN NULL
440
+ ELSE pin_out_time_in_datetime
441
+ END
442
+ WHERE
443
+ pin_in_time_in_datetime = 'NaT'
444
+ OR pin_out_time_in_datetime = 'NaT';
445
+
446
+ ''')
447
+
448
  def __create_data_analysis_views(self) -> None:
449
  """Create data analysis views in the database."""
450
  console.print('> Creating data analysis views...')