fix(userinfo): adds user info in prediction that lets user know about…

… the user
hotosm · Sep 26, 2024 · d8f9bdf · d8f9bdf
1 parent 3d628d1
commit d8f9bdf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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