Add conditional in get_config_block_regex

SUNET · Oct 21, 2024 · 89d85da · 89d85da
1 parent 45271a5
commit 89d85da
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/cnaas_nms/tools/jinja_filters.py b/src/cnaas_nms/tools/jinja_filters.py
@@ -233,8 +233,9 @@ def get_config_block_regex(config:str, section: str = "firewall", parser: Option
     config_parser = parser(config)
     config_relationship = config_parser.build_config_relationship()
     children = config_parser.find_all_children(section, match_type="regex")
-
+    
     if len(children) > 1:
-        return "\n".join(children) + "\n}"
+        collect = "\n".join(children)
+        return collect + "\n}" if isinstance(config_parser, JunosConfigParser) else collect
 
     return []