diff --git a/backend/core/models.py b/backend/core/models.py index ba5cf22c..7ad9afd6 100644 --- a/backend/core/models.py +++ b/backend/core/models.py @@ -145,3 +145,6 @@ class ApprovedPredictions(models.Model): srid=4326 ) ## Making this geometry field to support point/line prediction later on approved_at = models.DateTimeField(auto_now_add=True) + approved_by = models.ForeignKey( + OsmUser, to_field="osm_id", on_delete=models.CASCADE + ) diff --git a/backend/core/views.py b/backend/core/views.py index 5ee2217d..7a95fc88 100644 --- a/backend/core/views.py +++ b/backend/core/views.py @@ -307,6 +307,7 @@ class ApprovedPredictionsViewSet(viewsets.ModelViewSet): def create(self, request, *args, **kwargs): training_id = request.data.get("training") geom = request.data.get("geom") + request.data["approved_by"] = self.request.user existing_approved_feature = ApprovedPredictions.objects.filter( training=training_id, geom=geom