add waypoints to database

1 files changed, 20 insertions, 1 deletions

diff --git a/fietsboek/convert.py b/fietsboek/convert.py
index 5115dd7..04c7e73 100644
--- a/fietsboek/convert.py
+++ b/fietsboek/convert.py
@@ -6,7 +6,7 @@ import fitparse
 import gpxpy
 
 from . import geo, util
-from .models import Track
+from .models import Track, Waypoint
 
 FIT_RECORD_FIELDS = ["position_lat", "position_long", "altitude", "timestamp"]
 
@@ -103,6 +103,25 @@ def from_gpx(data: bytes) -> Track:
     track.title = track_name
     track.description = track_desc
     track.date = date
+
+    for waypoint in gpx.waypoints:
+        desc = None
+        # GPX waypoints can have both description and comment. It seems like
+        # comment is what is usually used (GPXViewer only shows the comment),
+        # so we'll prioritize that.
+        if waypoint.comment:
+            desc = waypoint.comment
+        if not desc and waypoint.description:
+            desc = waypoint.description
+        wpt = Waypoint(
+            longitude=waypoint.longitude,
+            latitude=waypoint.latitude,
+            elevation=waypoint.elevation,
+            name=waypoint.name,
+            description=desc,
+        )
+        track.waypoints.append(wpt)
+
     return track