Skip to content

Commit c5963bb

Browse files
authored
fix import for application-site objects and threat and https rules (#82)
* fix import for threat and https rules * fix import for application-site objects
1 parent 56c8fef commit c5963bb

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

importing/import_objects.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -966,7 +966,7 @@ def add_suffix_to_objects(payload, api_type, objects_suffix):
966966
return
967967

968968
fields_to_change = ["name", "source", "destination", "service", "members", "inline-layer", "networks", "host",
969-
"protected-scope", "protection-or-site", "exception-group-name", "rule-name", "applied-threat-rules"]
969+
"protected-scope", "protection-or-site", "action", "site-category", "exception-group-name", "rule-name", "applied-threat-rules"]
970970
if api_type == "threat-exception" and "exception-group-name" in payload and "name" not in payload:
971971
payload.update({"name": ""})
972972
for field in fields_to_change:
@@ -976,11 +976,11 @@ def add_suffix_to_objects(payload, api_type, objects_suffix):
976976
newName = oldName + objects_suffix
977977
payload[field] = newName
978978
changed_object_names_map[oldName] = newName
979-
elif field in ["source", "destination", "service", "members", "protected-scope", "protection-or-site"]:
979+
elif field in ["source", "destination", "service", "members", "protected-scope", "protection-or-site", "site-category"]:
980980
for i in range(len(payload[field])):
981981
if payload[field][i] in changed_object_names_map and payload[field][i] != "IPS":
982982
payload[field][i] = changed_object_names_map[payload[field][i]]
983-
elif field in ["inline-layer", "host", "exception-group-name", "rule-name"]:
983+
elif field in ["inline-layer", "host", "exception-group-name", "rule-name", "action"]:
984984
if payload[field] in changed_object_names_map:
985985
payload[field] = changed_object_names_map[payload[field]]
986986
elif field == "networks":

lists_and_dictionaries.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -715,6 +715,9 @@
715715
"service-group": 2,
716716
"time-group": 2,
717717
"application-group": 2,
718+
"application-site-category": 1,
719+
"application-site": 2,
720+
"application-site-group": 3,
718721
}
719722

720723
generic_objects_for_rule_fields = {

0 commit comments

Comments
 (0)