fix sessions

Author: durandom

telemetry/management/commands/maintenance.py

@@ -2,6 +2,7 @@
 import logging
 
 from django.core.management.base import BaseCommand
+from django.db.models import Prefetch, Q
 
 from telemetry.influx import Influx
 from telemetry.models import FastLap, Lap, Session
@@ -71,6 +72,12 @@ def add_arguments(self, parser):
             action="store_true",
         )
 
+        parser.add_argument(
+            "--fix-session-car-track",
+            help="fix sessions missing car or track by using data from first lap",
+            action="store_true",
+        )
+
     def handle(self, *args, **options):
         if options["delete_influx"]:
             self.influx = Influx()
@@ -87,6 +94,8 @@ def handle(self, *args, **options):
             self.dump_session()
         elif options["fix_cars"]:
             self.fix_cars()
+        elif options["fix_session_car_track"]:
+            self.fix_session_car_track()
 
     def fix_cars(self):
         from django.db.models import Count
@@ -133,6 +142,38 @@ def fix_cars(self):
         # else:
         #     print("\nNo cars without associated sessions found.")
 
+    def fix_session_car_track(self):
+        """
+        Find sessions missing car or track and set them from the first lap's data.
+        Process in batches to handle large numbers of sessions efficiently.
+        """
+
+        batch_size = 1000
+        sessions = Session.objects.filter(Q(car__isnull=True) | Q(track__isnull=True)).prefetch_related(Prefetch("laps", queryset=Lap.objects.order_by("number")))
+
+        total = sessions.count()
+        print(f"Found {total} sessions with missing car or track")
+
+        processed = 0
+        for session in sessions.iterator(chunk_size=batch_size):
+            first_lap = session.laps.first()
+            if first_lap:
+                updated = False
+                if not session.car and first_lap.car:
+                    session.car = first_lap.car
+                    updated = True
+                if not session.track and first_lap.track:
+                    session.track = first_lap.track
+                    updated = True
+
+                if updated:
+                    session.save()
+                    processed += 1
+                    if processed % 100 == 0:
+                        print(f"Processed {processed}/{total} sessions")
+
+        print(f"Updated {processed} sessions with car/track data from their laps")
+
     def dump_session(self):
         # TODO: Revisit this method. The Telemetry model is not defined.
         # Uncomment and fix the code below once the Telemetry model is available.

telemetry/pitcrew/session_saver.py

@@ -85,6 +85,8 @@ def save_sessions(self):
                         session_id=session.session_id,
                         session_type=session.session_type,
                         game=session.game,
+                        car=session.car,
+                        track=session.track,
                         defaults={"start": session.start, "end": session.end},
                     )
                     logging.debug(f"{session.session_id}: Saving session {session_id}")
@@ -113,9 +115,7 @@ def save_sessions(self):
                     #     continue
 
                     try:
-                        lap_record, created = session.record.laps.get_or_create(
-                            number=lap.number, car=session.car, track=track
-                        )
+                        lap_record, created = session.record.laps.get_or_create(number=lap.number, car=session.car, track=track)
                         if created:
                             logging.info(f"{session.session_id}: Created lap {lap_record}")
                         else:
@@ -143,10 +143,7 @@ def save_sessions(self):
                     if lap_length > track.length:
                         track.refresh_from_db()
                         if lap_length > track.length:
-                            logging.info(
-                                f"{session.session_id}: updating {track.name} "
-                                + f"length from {track.length} to {lap_length}"
-                            )
+                            logging.info(f"{session.session_id}: updating {track.name} " + f"length from {track.length} to {lap_length}")
                             track.length = lap_length
                             track.save()