[WIP] Mk/validation api small fixes

cvat/apps/engine/migrations/0087_alter_validationlayout_task_data.py

@@ -0,0 +1,24 @@
+# Generated by Django 4.2.15 on 2024-12-11 17:13
+
+import cvat.apps.engine.models
+from django.db import migrations
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("engine", "0086_profile_has_analytics_access"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="validationlayout",
+            name="task_data",
+            field=cvat.apps.engine.models.NullableOneToOneField(
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="validation_layout",
+                to="engine.data",
+            ),
+        ),
+    ]

cvat/apps/engine/models.py

@@ -16,11 +16,12 @@
 
 from django.conf import settings
 from django.contrib.auth.models import User
-from django.core.exceptions import ValidationError
+from django.core.exceptions import ValidationError, ObjectDoesNotExist
 from django.core.files.storage import FileSystemStorage
 from django.db import IntegrityError, models, transaction
 from django.db.models import Q, TextChoices
 from django.db.models.fields import FloatField
+from django.db.models.fields.related_descriptors import ReverseOneToOneDescriptor
 from django.utils.translation import gettext_lazy as _
 from drf_spectacular.types import OpenApiTypes
 from drf_spectacular.utils import extend_schema_field
@@ -259,11 +260,22 @@ class ValidationFrame(models.Model):
     )
     path = models.CharField(max_length=1024, default='')
 
+class ReverseNullableOneToOneDescriptor(ReverseOneToOneDescriptor):
+    def __get__(self, instance: models.Model, cls=None):
+        try:
+            return super().__get__(instance=instance, cls=cls)
+        except ObjectDoesNotExist:
+            return None
+
+class NullableOneToOneField(models.OneToOneField):
+    related_accessor_class = ReverseNullableOneToOneDescriptor
+
+
 class ValidationLayout(models.Model):
     "Represents validation configuration in a task"
 
-    task_data = models.OneToOneField(
-        'Data', on_delete=models.CASCADE, related_name="validation_layout"
+    task_data = NullableOneToOneField(
+        'Data', on_delete=models.CASCADE, related_name="validation_layout",
     )
 
     mode = models.CharField(max_length=32, choices=ValidationMode.choices())
@@ -298,7 +310,7 @@ class Data(models.Model):
     images: models.manager.RelatedManager[Image]
     video: MaybeUndefined[Video]
     related_files: models.manager.RelatedManager[RelatedFile]
-    validation_layout: MaybeUndefined[ValidationLayout]
+    validation_layout: ValidationLayout | None
 
     client_files: models.manager.RelatedManager[ClientFile]
     server_files: models.manager.RelatedManager[ServerFile]

cvat/apps/engine/serializers.py

@@ -991,10 +991,15 @@ def update(self, instance: models.Job, validated_data: dict[str, Any]) -> models
         db_task = db_segment.task
         db_data = db_task.data
 
-        if not (
-            hasattr(db_job.segment.task.data, 'validation_layout') and
-            db_job.segment.task.data.validation_layout.mode == models.ValidationMode.GT_POOL
-        ):
+        db_task_validation_layout = db_data.validation_layout
+
+        if not db_task_validation_layout:
+            raise serializers.ValidationError(
+                "Task has no validation layout configured. "
+                "Validation must be initialized during a task creation"
+            )
+
+        if db_task_validation_layout.mode != models.ValidationMode.GT_POOL:
             raise serializers.ValidationError(
                 "Honeypots can only be modified if the task "
                 f"validation mode is '{models.ValidationMode.GT_POOL}'"
@@ -1254,7 +1259,7 @@ class JobValidationLayoutReadSerializer(serializers.Serializer):
     )
 
     def to_representation(self, instance: models.Job):
-        validation_layout = getattr(instance.segment.task.data, 'validation_layout', None)
+        validation_layout = instance.segment.task.data.validation_layout
         if not validation_layout:
             return {}
 
@@ -1332,9 +1337,13 @@ def validate(self, attrs):
 
     @transaction.atomic
     def update(self, instance: models.Task, validated_data: dict[str, Any]) -> models.Task:
-        validation_layout = getattr(instance.data, 'validation_layout', None)
+        validation_layout = instance.data.validation_layout
+
         if not validation_layout:
-            raise serializers.ValidationError("Validation is not configured in the task")
+            raise serializers.ValidationError(
+                "Task has no validation layout configured. "
+                "Validation must be initialized during a task creation"
+            )
 
         if 'disabled_frames' in validated_data:
             requested_disabled_frames = validated_data['disabled_frames']

cvat/apps/engine/views.py

@@ -1746,13 +1746,15 @@ def preview(self, request, pk):
         summary="Allows getting current validation configuration",
         responses={
             '200': OpenApiResponse(TaskValidationLayoutReadSerializer),
+            '204': OpenApiResponse(description="No validation layout is associated with a task"),
         })
     @extend_schema(
         methods=["PATCH"],
         summary="Allows updating current validation configuration",
         request=TaskValidationLayoutWriteSerializer,
         responses={
             '200': OpenApiResponse(TaskValidationLayoutReadSerializer),
+            '400': OpenApiResponse(description="Bad request"),
         },
         examples=[
             OpenApiExample("set honeypots to random validation frames", {
@@ -1774,22 +1776,17 @@ def preview(self, request, pk):
     def validation_layout(self, request, pk):
         db_task = self.get_object() # call check_object_permissions as well
 
-        validation_layout = getattr(db_task.data, 'validation_layout', None)
-
         if request.method == "PATCH":
-            if not validation_layout:
-                return ValidationError(
-                    "Task has no validation setup configured. "
-                    "Validation must be initialized during task creation"
-                )
-
+            # TODO: probably it should take validation_layout instead of task
             request_serializer = TaskValidationLayoutWriteSerializer(db_task, data=request.data)
             request_serializer.is_valid(raise_exception=True)
-            validation_layout = request_serializer.save().data.validation_layout
+            updated_task = request_serializer.save()
+            validation_layout = updated_task.data.validation_layout
+        else:
+            validation_layout = db_task.data.validation_layout
 
-        if not validation_layout:
-            response_serializer = TaskValidationLayoutReadSerializer(SimpleNamespace(mode=None))
-            return Response(response_serializer.data, status=status.HTTP_200_OK)
+        if validation_layout is None:
+            return Response(status=status.HTTP_204_NO_CONTENT)
 
         response_serializer = TaskValidationLayoutReadSerializer(validation_layout)
         return Response(response_serializer.data, status=status.HTTP_200_OK)
@@ -2352,13 +2349,15 @@ def preview(self, request, pk):
         summary="Allows getting current validation configuration",
         responses={
             '200': OpenApiResponse(JobValidationLayoutReadSerializer),
+            '204': OpenApiResponse(description="No validation layout is associated with a job"),
         })
     @extend_schema(
         methods=["PATCH"],
         summary="Allows updating current validation configuration",
         request=JobValidationLayoutWriteSerializer,
         responses={
             '200': OpenApiResponse(JobValidationLayoutReadSerializer),
+            '400': OpenApiResponse(description="Bad request"),
         },
         examples=[
             OpenApiExample("set honeypots to random validation frames", {
@@ -2394,7 +2393,12 @@ def validation_layout(self, request, pk):
             db_job = request_serializer.save()
 
         response_serializer = JobValidationLayoutReadSerializer(db_job)
-        return Response(response_serializer.data, status=status.HTTP_200_OK)
+        content = response_serializer.data
+
+        if content:
+            return Response(response_serializer.data, status=status.HTTP_200_OK)
+
+        return Response(status=status.HTTP_204_NO_CONTENT)
 
 @extend_schema(tags=['issues'])
 @extend_schema_view(

cvat/schema.yml

@@ -2664,6 +2664,8 @@ paths:
               schema:
                 $ref: '#/components/schemas/JobValidationLayoutRead'
           description: ''
+        '204':
+          description: No validation layout is associated with a job
     patch:
       operationId: jobs_partial_update_validation_layout
       summary: Allows updating current validation configuration
@@ -2707,6 +2709,8 @@ paths:
               schema:
                 $ref: '#/components/schemas/JobValidationLayoutRead'
           description: ''
+        '400':
+          description: Bad request
   /api/labels:
     get:
       operationId: labels_list
@@ -6147,6 +6151,8 @@ paths:
               schema:
                 $ref: '#/components/schemas/TaskValidationLayoutRead'
           description: ''
+        '204':
+          description: No validation layout is associated with a task
     patch:
       operationId: tasks_partial_update_validation_layout
       summary: Allows updating current validation configuration
@@ -6201,6 +6207,8 @@ paths:
               schema:
                 $ref: '#/components/schemas/TaskValidationLayoutRead'
           description: ''
+        '400':
+          description: Bad request
   /api/tasks/backup/:
     post:
       operationId: tasks_create_backup