Skip to content

Commit 8b6c71f

Browse files
committed
feat: Support variable assign save input output value
1 parent ab492d8 commit 8b6c71f

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

apps/application/flow/step_node/variable_assign/impl/base_variable_assign_node.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,21 +8,30 @@
88
class 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

Comments
 (0)