Draichi commited on
Commit
9ba81d1
·
unverified ·
1 Parent(s): 14d15a4

wip(formula1_databases): insert Event table

Browse files
Files changed (1) hide show
  1. notebooks/formula1_databases.py +17 -17
notebooks/formula1_databases.py CHANGED
@@ -42,15 +42,15 @@ class FastF1ToSQL:
42
  event_date DATE,
43
  event_name TEXT,
44
  session_1_date_utc DATETIME,
45
- session_1_name TEXT CHECK(session_1_name IN ('practice', 'qualify', 'race')),
46
  session_2_date_utc DATETIME,
47
- session_2_name TEXT CHECK(session_2_name IN ('practice', 'qualify', 'race')),
48
  session_3_date_utc DATETIME,
49
- session_3_name TEXT CHECK(session_3_name IN ('practice', 'qualify', 'race')),
50
  session_4_date_utc DATETIME,
51
- session_4_name TEXT CHECK(session_4_name IN ('practice', 'qualify', 'race')),
52
  session_5_date_utc DATETIME,
53
- session_5_name TEXT CHECK(session_5_name IN ('practice', 'qualify', 'race'))
54
  );
55
 
56
  CREATE TABLE IF NOT EXISTS Sessions (
@@ -163,17 +163,17 @@ class FastF1ToSQL:
163
  'round_number': session.event.RoundNumber,
164
  'country': session.event.Country,
165
  'location': session.event.Location,
166
- 'event_date': session.event.EventDate.date(),
167
  'event_name': session.event.EventName,
168
- 'session_1_date_utc': session.event.Session1DateUtc,
169
  'session_1_name': session.event.Session1.lower(),
170
- 'session_2_date_utc': session.event.Session2DateUtc,
171
  'session_2_name': session.event.Session2.lower(),
172
- 'session_3_date_utc': session.event.Session3DateUtc,
173
  'session_3_name': session.event.Session3.lower(),
174
- 'session_4_date_utc': session.event.Session4DateUtc,
175
  'session_4_name': session.event.Session4.lower(),
176
- 'session_5_date_utc': session.event.Session5DateUtc,
177
  'session_5_name': session.event.Session5.lower(),
178
  }
179
 
@@ -194,7 +194,7 @@ class FastF1ToSQL:
194
  'event_id': self.cursor.lastrowid,
195
  'track_id': self.get_or_create_track(session.event.Location, session.event.Country),
196
  'session_type': session.name,
197
- 'date': session.date,
198
  }
199
  columns = ', '.join(session_data.keys())
200
  placeholders = ':' + ', :'.join(session_data.keys())
@@ -252,9 +252,9 @@ class FastF1ToSQL:
252
  'tyre_life_in_laps': lap['TyreLife'],
253
  'is_fresh_tyre': lap['FreshTyre'],
254
  'position': lap['Position'],
255
- 'lap_start_time_in_datetime': lap['lap_start_time_in_datetime'],
256
- 'pin_in_time_in_datetime': lap['pin_in_time_in_datetime'],
257
- 'pin_out_time_in_datetime': lap['pin_out_time_in_datetime'],
258
  }
259
  columns = ', '.join(lap_data.keys())
260
  placeholders = ':' + ', :'.join(lap_data.keys())
@@ -290,7 +290,7 @@ class FastF1ToSQL:
290
  'y_position': sample['Y'],
291
  'z_position': sample['Z'],
292
  'is_off_track': sample['Status'] == 'OffTrack',
293
- 'datetime': sample.name,
294
  }
295
  columns = ', '.join(telemetry_data.keys())
296
  placeholders = ':' + ', :'.join(telemetry_data.keys())
@@ -318,7 +318,7 @@ class FastF1ToSQL:
318
  'relative_air_humidity_in_percentage': sample['Humidity'],
319
  'air_pressure_in_mbar': sample['Pressure'],
320
  'is_raining': sample['Rainfall'],
321
- 'datetime': sample.name,
322
  }
323
  columns = ', '.join(weather_sample.keys())
324
  placeholders = ':' + ', :'.join(weather_sample.keys())
 
42
  event_date DATE,
43
  event_name TEXT,
44
  session_1_date_utc DATETIME,
45
+ session_1_name TEXT,
46
  session_2_date_utc DATETIME,
47
+ session_2_name TEXT,
48
  session_3_date_utc DATETIME,
49
+ session_3_name TEXT,
50
  session_4_date_utc DATETIME,
51
+ session_4_name TEXT,
52
  session_5_date_utc DATETIME,
53
+ session_5_name TEXT
54
  );
55
 
56
  CREATE TABLE IF NOT EXISTS Sessions (
 
163
  'round_number': session.event.RoundNumber,
164
  'country': session.event.Country,
165
  'location': session.event.Location,
166
+ 'event_date': str(session.event.EventDate.date()),
167
  'event_name': session.event.EventName,
168
+ 'session_1_date_utc': str(session.event.Session1DateUtc),
169
  'session_1_name': session.event.Session1.lower(),
170
+ 'session_2_date_utc': str(session.event.Session2DateUtc),
171
  'session_2_name': session.event.Session2.lower(),
172
+ 'session_3_date_utc': str(session.event.Session3DateUtc),
173
  'session_3_name': session.event.Session3.lower(),
174
+ 'session_4_date_utc': str(session.event.Session4DateUtc),
175
  'session_4_name': session.event.Session4.lower(),
176
+ 'session_5_date_utc': str(session.event.Session5DateUtc),
177
  'session_5_name': session.event.Session5.lower(),
178
  }
179
 
 
194
  'event_id': self.cursor.lastrowid,
195
  'track_id': self.get_or_create_track(session.event.Location, session.event.Country),
196
  'session_type': session.name,
197
+ 'date': str(session.date),
198
  }
199
  columns = ', '.join(session_data.keys())
200
  placeholders = ':' + ', :'.join(session_data.keys())
 
252
  'tyre_life_in_laps': lap['TyreLife'],
253
  'is_fresh_tyre': lap['FreshTyre'],
254
  'position': lap['Position'],
255
+ 'lap_start_time_in_datetime': str(lap['lap_start_time_in_datetime']),
256
+ 'pin_in_time_in_datetime': str(lap['pin_in_time_in_datetime']),
257
+ 'pin_out_time_in_datetime': str(lap['pin_out_time_in_datetime']),
258
  }
259
  columns = ', '.join(lap_data.keys())
260
  placeholders = ':' + ', :'.join(lap_data.keys())
 
290
  'y_position': sample['Y'],
291
  'z_position': sample['Z'],
292
  'is_off_track': sample['Status'] == 'OffTrack',
293
+ 'datetime': str(sample.name),
294
  }
295
  columns = ', '.join(telemetry_data.keys())
296
  placeholders = ':' + ', :'.join(telemetry_data.keys())
 
318
  'relative_air_humidity_in_percentage': sample['Humidity'],
319
  'air_pressure_in_mbar': sample['Pressure'],
320
  'is_raining': sample['Rainfall'],
321
+ 'datetime': str(sample.name),
322
  }
323
  columns = ', '.join(weather_sample.keys())
324
  placeholders = ':' + ', :'.join(weather_sample.keys())