feat: DEV-3259: Support updated_by for Annotations (HumanSignal#3244)

* feat: DEV-3259: Support updated_by for Annotations

* added tests for annotation updated_by

* update updated_by field when updating annotation

* fix migrations conflict

Author: hakan458

label_studio/tasks/api.py

@@ -269,6 +269,8 @@ def update(self, request, *args, **kwargs):
         # save user history with annotator_id, time & annotation result
         annotation_id = self.kwargs['pk']
         annotation = get_object_with_check_and_log(request, Annotation, pk=annotation_id)
+        annotation.updated_by = request.user
+        annotation.save(update_fields=['updated_by'])
 
         task = annotation.task
         if self.request.data.get('ground_truth'):
@@ -384,6 +386,7 @@ def perform_create(self, ser):
             # serialize annotation
             extra_args.update({
                 'prediction': prediction_ser,
+                'updated_by': user
             })
 
         if 'was_cancelled' in self.request.GET:

label_studio/tasks/migrations/0031_auto_20221118_2338.py

@@ -0,0 +1,28 @@
+# Generated by Django 3.2.16 on 2022-11-18 23:38
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+def set_updated_by(apps, _):
+    Annotation = apps.get_model('tasks', 'Annotation')
+    for obj in Annotation.objects.all().iterator():
+        obj.updated_by = obj.task.updated_by
+        obj.save()
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('tasks', '0030_auto_20221102_1118'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='annotation',
+            name='updated_by',
+            field=models.ForeignKey(help_text='Last user who updated this annotation', null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='updated_annotations', to=settings.AUTH_USER_MODEL, verbose_name='updated by'),
+        ),
+        migrations.RunPython(set_updated_by)
+    ]

label_studio/tasks/models.py

@@ -362,6 +362,9 @@ class Annotation(AnnotationMixin, models.Model):
                                 help_text='Project ID for this annotation')
     completed_by = models.ForeignKey(settings.AUTH_USER_MODEL, related_name="annotations", on_delete=models.SET_NULL,
                                      null=True, help_text='User ID of the person who created this annotation')
+    updated_by = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='updated_annotations',
+                                   on_delete=models.SET_NULL, null=True, verbose_name=_('updated by'),
+                                   help_text='Last user who updated this annotation')
     was_cancelled = models.BooleanField(_('was cancelled'), default=False, help_text='User skipped the task', db_index=True)
     ground_truth = models.BooleanField(_('ground_truth'), default=False, help_text='This annotation is a Ground Truth (ground_truth)', db_index=True)
     created_at = models.DateTimeField(_('created at'), auto_now_add=True, help_text='Creation time')
@@ -432,6 +435,9 @@ def update_task(self):
         self.task.save(update_fields=update_fields)
 
     def save(self, *args, **kwargs):
+        request = get_current_request()
+        if request:
+            self.updated_by = request.user
         result = super().save(*args, **kwargs)
         self.update_task()
         return result

label_studio/tests/test_annotations.py

@@ -47,9 +47,11 @@ def test_create_annotation(caplog, any_client, configured_project_min_annotation
         # annotator client
         if hasattr(any_client, 'annotator') and any_client.annotator is not None:
             assert annotation.completed_by.id == any_client.user.id
+            assert annotation.updated_by.id == any_client.user.id
         # business client
         else:
             assert annotation.completed_by.id == any_client.business.admin.id
+            assert annotation.updated_by.id == any_client.business.admin.id
 
         if apps.is_installed('businesses'):
             assert annotation.task.accuracy == 1.0
@@ -93,6 +95,9 @@ def test_create_annotation_with_ground_truth(caplog, any_client, configured_proj
         assert m.called
         task = Task.objects.get(id=task.id)
         assert task.annotations.count() == 2
+        annotations = Annotation.objects.filter(task=task)
+        for a in annotations:
+            assert a.updated_by.id == any_client.user.id
 
 
 @pytest.mark.django_db