fix: When there is a form node in the loop node, the loop data is lost (#4214)

Author: shaohuzhang1

apps/application/flow/step_node/loop_node/impl/base_loop_node.py

@@ -225,6 +225,9 @@ def handler(self, workflow):
 class BaseLoopNode(ILoopNode):
     def save_context(self, details, workflow_manage):
         self.context['result'] = details.get('result')
+        for key, value in details['context'].items():
+            if key not in self.context:
+                self.context[key] = value
         self.answer_text = str(details.get('result'))
 
     def get_answer_list(self) -> List[Answer] | None:
@@ -262,7 +265,13 @@ def workflow_manage_new_instance(loop_data, global_data, start_node_id=None,
                           _write_context=get_write_context(loop_type, array, number, loop_body, stream),
                           _is_interrupt=_is_interrupt_exec)
 
+    def get_loop_context_data(self):
+        fields = self.node.properties.get('config', []).get('fields', []) or []
+        return {f.get('value'): self.context.get(f.get('value')) for f in fields if
+                self.context.get(f.get('value')) is not None}
+
     def get_details(self, index: int, **kwargs):
+
         return {
             'name': self.node.properties.get('stepName'),
             "index": index,
@@ -276,6 +285,7 @@ def get_details(self, index: int, **kwargs):
             "current_item": self.context.get("item"),
             'loop_type': self.node_params_serializer.data.get('loop_type'),
             'status': self.status,
+            'loop_context_data': self.get_loop_context_data(),
             'loop_node_data': self.context.get("loop_node_data"),
             'loop_answer_data': self.context.get("loop_answer_data"),
             'err_message': self.err_message

apps/application/flow/workflow_manage.py

@@ -202,6 +202,9 @@ def get_node_params(n):
                     {**self.start_node.node_params, 'form_data': start_node_data}, self.start_node.workflow_params)
                 if self.start_node.type == 'loop-node':
                     loop_node_data = node_details.get('loop_node_data', {})
+                    for k, v in node_details.get('loop_context_data').items():
+                        if v is not None:
+                            self.start_node.context[k] = v
                     self.start_node.context['loop_node_data'] = loop_node_data
                     self.start_node.context['current_index'] = node_details.get('current_index')
                     self.start_node.context['current_item'] = node_details.get('current_item')