Browse Source

fix: enhance file extension condition check for if-else node (#17060)

tags/1.2.0
Arcaner 7 months ago
parent
commit
becd03a4aa
No account linked to committer's email address
1 changed files with 15 additions and 1 deletions
  1. 15
    1
      api/core/workflow/utils/condition/processor.py

+ 15
- 1
api/core/workflow/utils/condition/processor.py View File

@@ -375,11 +375,25 @@ def _process_sub_conditions(
for condition in sub_conditions:
key = FileAttribute(condition.key)
values = [file_manager.get_attr(file=file, attr=key) for file in files]
expected_value = condition.value
if key == FileAttribute.EXTENSION:
if not isinstance(expected_value, str):
raise TypeError("Expected value must be a string when key is FileAttribute.EXTENSION")
if expected_value and not expected_value.startswith("."):
expected_value = "." + expected_value

normalized_values = []
for value in values:
if value and isinstance(value, str):
if not value.startswith("."):
value = "." + value
normalized_values.append(value)
values = normalized_values
sub_group_results = [
_evaluate_condition(
value=value,
operator=condition.comparison_operator,
expected=condition.value,
expected=expected_value,
)
for value in values
]

Loading…
Cancel
Save