Bugfixing

src/background_jobs/elevation.py

@@ -8,7 +8,7 @@ from paths import FLIGHT_GPX_TRACK_PATH
 
 async def add_terrain_elevation_to_flight(flight_id: int, gpx_filename: str):
     gpx_parser = GPXParser(f"{FLIGHT_GPX_TRACK_PATH}/{gpx_filename}")
-    coordinates = await gpx_parser.get_coordinates()
+    coordinates = await gpx_parser.get_coordinates_all()
 
     try:
         elevation = await elevation_api.get_elevation_for_points(coordinates)

src/external/gpx_parser.py

@@ -80,10 +80,12 @@ class GPXParser:
 
     @cached()
     async def get_coordinates(self) -> List[Dict[str, float]]:
+        return await self.average_coordinates(await self.get_coordinates_all())
+
+    @cached()
+    async def get_coordinates_all(self) -> List[Dict[str, float]]:
         nodes = self.run_xpath("//gpx:trkpt")
-        return await self.average_coordinates(
-            [{"lat": float(node.attrib["lat"]), "lng": float(node.attrib['lon'])} for node in nodes]
-        )
+        return [{"lat": float(node.attrib["lat"]), "lng": float(node.attrib['lon'])} for node in nodes]
 
     @cached()
     async def get_speed(self) -> List[float]:
@@ -141,7 +143,12 @@ class GPXParser:
                 continue
 
             extensions = target_node.find("./extensions", self.gpx.getroot().nsmap)
-            extensions.append(etree.XML(f"<terrain_elevation>{elevation}</terrain_elevation>"))
+            if extensions is not None:
+                extensions.append(etree.XML(f"<terrain_elevation>{elevation}</terrain_elevation>"))
+            else:
+                target_node.append(
+                    etree.XML(f"<extensions><terrain_elevation>{elevation}</terrain_elevation></extensions>")
+                )
 
         return self.gpx
 

src/scripts/elevation.py

@@ -1,7 +1,6 @@
 import asyncio
 import sys
 from sqlalchemy import select
-
 sys.path.insert(0, "/app/src")
 from paths import FLIGHT_GPX_TRACK_PATH  # noqa
 from database import async_session, models  # noqa
@@ -46,13 +45,18 @@ async def add_elevation_to_tracks():
 
         for flight in flights:
             gpx_file = f"{FLIGHT_GPX_TRACK_PATH}/{flight.gpx_track_filename}"
-            gpx = GPXParser(gpx_file)
+            try:
+                gpx = GPXParser(gpx_file)
+            except OSError:
+                print(f"ERROR, cannot open {gpx_file=}, skipping")
+                continue
 
-            coordinates = await gpx.get_coordinates()
+            coordinates = await gpx.get_coordinates_all()
             elevation = await elevation_api.get_elevation_for_points(coordinates)
+
             gpx_with_elevation = gpx.add_terrain_elevation(elevation)
 
-            output_name = f"terrain_{flight.gpx_track_filename[30:]}"
+            output_name = f"terrain_{flight.gpx_track_filename[-30:]}"
             gpx.write(gpx_with_elevation, output=f"{FLIGHT_GPX_TRACK_PATH}/{output_name}")
             await models.Flight.update(
                 db_session=session, obj=flight, data={