88class BaseVariableAssignNode (IVariableAssignNode ):
99 def save_context (self , details , workflow_manage ):
1010 self .context ['variable_list' ] = details .get ('variable_list' )
11+ self .context ['result_list' ] = details .get ('result_list' )
1112
1213 def execute (self , variable_list , stream , ** kwargs ) -> NodeResult :
1314 #
15+ result_list = []
1416 for variable in variable_list :
1517 if 'fields' not in variable :
1618 continue
1719 if 'global' == variable ['fields' ][0 ]:
20+ result = {
21+ 'name' : variable ['fields' ][1 ],
22+ 'input_value' : self .get_reference_content (variable ['fields' ]),
23+ }
1824 if variable ['source' ] == 'custom' :
1925 self .workflow_manage .context [variable ['fields' ][1 ]] = variable ['value' ]
26+ result ['output_value' ] = variable ['value' ]
2027 else :
2128 reference = self .get_reference_content (variable ['reference' ])
2229 self .workflow_manage .context [variable ['fields' ][1 ]] = reference
23- # print('variable_list:', variable_list)
30+ result ['output_value' ] = reference
31+ result_list .append (result )
32+ print (result_list )
2433
25- return NodeResult ({'variable_list' : variable_list }, {})
34+ return NodeResult ({'variable_list' : variable_list , 'result_list' : result_list }, {})
2635
2736 def get_reference_content (self , fields : List [str ]):
2837 return str (self .workflow_manage .get_reference_field (
@@ -36,6 +45,7 @@ def get_details(self, index: int, **kwargs):
3645 'run_time' : self .context .get ('run_time' ),
3746 'type' : self .node .type ,
3847 'variable_list' : self .context .get ('variable_list' ),
48+ 'result_list' : self .context .get ('result_list' ),
3949 'status' : self .status ,
4050 'err_message' : self .err_message
4151 }
0 commit comments