add allowed_dataset_choice in rerun valid api

umccr · Dec 11, 2024 · c4c58ff · c4c58ff
1 parent 4bf33f1
commit c4c58ff
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/...oad/stateless/stacks/workflow-manager/workflow_manager/serializers/workflow_run_action.py b/...oad/stateless/stacks/workflow-manager/workflow_manager/serializers/workflow_run_action.py
@@ -9,6 +9,7 @@ class AllowedRerunWorkflow(StrEnum):
 
 class AllowedRerunWorkflowSerializer(serializers.Serializer):
     is_valid = serializers.BooleanField()
+    allowed_dataset_choice = serializers.ListField(child=serializers.CharField())
     valid_workflows = serializers.ListField(child=serializers.CharField())
 
 class BaseRerunInputSerializer(serializers.Serializer):

diff --git a/...rkload/stateless/stacks/workflow-manager/workflow_manager/viewsets/workflow_run_action.py b/...rkload/stateless/stacks/workflow-manager/workflow_manager/viewsets/workflow_run_action.py
@@ -32,9 +32,18 @@ class WorkflowRunActionViewSet(ViewSet):
     def validate_rerun_workflows(self, request, *args, **kwargs):
         wfl_run = get_object_or_404(self.queryset, pk=kwargs.get('pk'))
         is_valid = wfl_run.workflow.workflow_name in AllowedRerunWorkflow
+
+        # Get allowed dataset choice for the workflow
+        wfl_name = wfl_run.workflow.workflow_name
+        allowed_dataset_choice = []
+        if wfl_name == AllowedRerunWorkflow.RNASUM.value:
+            allowed_dataset_choice = RERUN_INPUT_SERIALIZERS[wfl_name].allowed_dataset_choice
+
         reponse = {
             'is_valid': is_valid,
-            'valid_workflows': AllowedRerunWorkflow
+            'allowed_dataset_choice': allowed_dataset_choice,
+            'valid_workflows': AllowedRerunWorkflow,
+
         }
         return Response(reponse, status=status.HTTP_200_OK)