TMSS-234: TaskBlueprint Copy To Task Draft using mixin approach

SAS/TMSS/src/tmss/tmssapp/viewsets/specification.py

@@ -494,6 +494,27 @@ class SchedulingUnitBlueprintCopyToSchedulingUnitDraftViewSet(LOFARCopyViewSet):
         else:
             content = {'Error': 'scheduling_unit_draft_id is missing'}
             return Response(content, status=status.HTTP_404_NOT_FOUND)
+   
+    
+class TaskBlueprintCopyToTaskDraftViewSet(LOFARCopyViewSet):
+    queryset = models.SchedulingUnitBlueprint.objects.all()
+    serializer_class = serializers.SchedulingUnitBlueprintCopyToSchedulingUnitDraftSerializer
+    
+    @swagger_auto_schema(responses={201: "The TaskDraft created from this TaskBlueprint",
+                                    403: 'forbidden'},
+                         operation_description="Copy this TaskBlueprint to a new TaskDraft.")
+    def create(self, request, *args, **kwargs):
+        if 'task_blueprint_id' in kwargs:
+            task_blueprint = get_object_or_404(models.TaskBlueprint, pk=kwargs['task_blueprint_id'])
+            task_draft = copy_task_blueprint_to_task_draft(task_blueprint)
+
+            # return a response with the new serialized scheduling_unit_blueprint (with references to the created task_blueprint(s) and (scheduled) subtasks)
+            return Response(serializers.TaskDraftSerializer(task_draft, context={'request':request}).data,
+                        status=status.HTTP_201_CREATED)
+
+        else:
+            content = {'Error': 'task_blueprint_id is missing'}
+            return Response(content, status=status.HTTP_404_NOT_FOUND)
 
    
 class SchedulingUnitBlueprintViewSet(LOFARViewSet):
@@ -724,19 +745,6 @@ class TaskBlueprintViewSet(LOFARViewSet):
         return Response(serializer.data)
  
 
-    @swagger_auto_schema(responses={201: "The TaskDraft created from this TaskBlueprint",
-                                    403: 'forbidden'},
-                         operation_description="Copy this TaskBlueprint to a new TaskDraft.")
-    @action(methods=['get'], detail=True, url_name="copy_task_blueprint_to_task_draft", name="Copy to a new TaskDraft")
-    def copy_task_blueprint_to_task_draft(self, request, pk=None):
-   
-        task_blueprint = get_object_or_404(models.TaskBlueprint, pk=pk)
-        task_draft = copy_task_blueprint_to_task_draft(task_blueprint)
-
-        # return a response with the new serialized scheduling_unit_blueprint (with references to the created task_blueprint(s) and (scheduled) subtasks)
-        return Response(serializers.TaskDraftSerializer(task_draft, context={'request':request}).data,
-                        status=status.HTTP_201_CREATED)
-
 class TaskBlueprintNestedViewSet(LOFARNestedViewSet):
     queryset = models.TaskBlueprint.objects.all()
     serializer_class = serializers.TaskBlueprintSerializer

SAS/TMSS/src/tmss/urls.py

@@ -144,6 +144,7 @@ router.register(r'task_blueprint/(?P<task_blueprint_id>\d+)/subtask', viewsets.S
 router.register(r'scheduling_set/(?P<scheduling_set_id>\d+)/copy_scheduling_unit_draft', viewsets.SchedulingUnitDraftCopyFromSchedulingSetViewSet)
 router.register(r'scheduling_unit_draft/(?P<scheduling_unit_draft_id>\d+)/copy', viewsets.SchedulingUnitDraftCopyViewSet)
 router.register(r'scheduling_unit_blueprint/(?P<scheduling_unit_blueprint_id>\d+)/copy_to_scheduling_unit_draft', viewsets.SchedulingUnitBlueprintCopyToSchedulingUnitDraftViewSet)
+router.register(r'task_blueprint/(?P<task_blueprint_id>\d+)/copy_to_task_draft', viewsets.TaskBlueprintCopyToTaskDraftViewSet)
 
 
 # SCHEDULING