Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Small fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
PrzeG committed Mar 5, 2024
1 parent a4fc265 commit b29afb8
Showing 1 changed file with 14 additions and 10 deletions.
24 changes: 14 additions & 10 deletions catalystwan/api/templates/feature_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,22 +27,26 @@ def map_fields(cls, values: Union[Any, Dict[str, Union[List[FlattenedTemplateVal

if not isinstance(values, dict):
return values
mapped_values = {}
for field_name, field_info in cls.model_fields.items():
payload_name = get_extra_field(field_info, "vmanage_key") or field_info.alias or field_name
data_path = get_extra_field(field_info, "data_path", [])
if payload_name not in values:
continue
vmanage_key = get_extra_field(field_info, "vmanage_key")
if vmanage_key in values:
payload_name = vmanage_key
elif field_info.alias in values:
payload_name = field_info.alias
elif field_name in values:
payload_name = field_name
else:
value = values[payload_name]
if value and isinstance(value, list) and all([isinstance(v, FlattenedTemplateValue)] for v in value):
continue
data_path = get_extra_field(field_info, "data_path", [])
value = values.pop(payload_name)
if value and isinstance(value, list) and all([isinstance(v, FlattenedTemplateValue) for v in value]):
for template_value in value:
if template_value.data_path == data_path:
mapped_values[field_name] = template_value.value
values[field_name] = template_value.value
break
else:
mapped_values[field_name] = value
return mapped_values
values[field_name] = value
return values


class FeatureTemplate(FeatureTemplateValidator, ABC):
Expand Down

0 comments on commit b29afb8

Please sign in to comment.