fix: Simplify return in insert_data_json_feed

wazuh · Sep 20, 2021 · ccb1aa9 · ccb1aa9
1 parent db25b81
commit ccb1aa9
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/deps/wazuh_testing/wazuh_testing/vulnerability_detector.py b/deps/wazuh_testing/wazuh_testing/vulnerability_detector.py
@@ -798,15 +798,10 @@ def insert_data_json_feed(data, field_name, field_value, append_data, brackets=T
  raw_data = json.dumps(data, indent=4, ensure_ascii=False).replace('"replace_me"', f"{field_name}")
 
  if append_data:
- if brackets:
- return f"[\n{raw_data},\n{json.dumps(append_data, indent=4, ensure_ascii=False)}\n]"
- else:
- return f"\n{raw_data},\n{json.dumps(append_data, indent=4, ensure_ascii=False)}\n"
+ response = f"\n{raw_data},\n{json.dumps(append_data, indent=4, ensure_ascii=False)}\n"
+ return f"[{response}]" if brackets else response
  else:
- if brackets:
- return f"[\n{raw_data}]"
- else:
- return f"\n{raw_data}"
+ return f"[\n{raw_data}]" if brackets else f"\n{raw_data}"
 
 
 def check_if_modulesd_is_running():