From 90076090f05a1f03e2fbff307418ee18d7d046ef Mon Sep 17 00:00:00 2001 From: Kelly Brazil Date: Wed, 21 Apr 2021 09:09:09 -0700 Subject: [PATCH] fix int conversions if 'rules' key does not exist --- jc/parsers/ufw.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/jc/parsers/ufw.py b/jc/parsers/ufw.py index 6762ccd2..d3cae5cf 100644 --- a/jc/parsers/ufw.py +++ b/jc/parsers/ufw.py @@ -40,7 +40,7 @@ Schema: "from_transport": string, # null if from_service is set "from_start_port": integer, # null if from_service is set "from_end_port": integer, # null if from_service is set - "from_service": string, # null if any above are set + "from_service": string # null if any above are set } ] } @@ -304,10 +304,12 @@ def _process(proc_data): """ int_list = ['index', 'to_subnet', 'to_start_port', 'to_end_port', 'from_subnet', 'from_start_port', 'from_end_port'] - for i, item in enumerate(proc_data['rules']): - for key in item: - if key in int_list: - proc_data['rules'][i][key] = jc.utils.convert_to_int(proc_data['rules'][i][key]) + + if 'rules' in proc_data: + for i, item in enumerate(proc_data['rules']): + for key in item: + if key in int_list: + proc_data['rules'][i][key] = jc.utils.convert_to_int(proc_data['rules'][i][key]) return proc_data