[fix] Create dict value if it doesn't exist

This commit is contained in:
William Wernert
2021-02-19 09:31:22 -05:00
parent 9183c0a92c
commit 2184c6d59f

View File

@@ -175,6 +175,9 @@ def add_rem_disabled(args: dict):
pillar_dict = read_pillar(args.pillar)
if not sids_key_exists(pillar_dict, 'disabled'):
pillar_dict['idstools']['sids']['disabled'] = None
if args.remove:
temp_pillar_dict = rem_from_sids(pillar_dict, 'disabled', args.sid_pattern)
else:
@@ -237,6 +240,9 @@ def add_rem_enabled(args: dict):
pillar_dict = read_pillar(args.pillar)
if not sids_key_exists(pillar_dict, 'enabled'):
pillar_dict['idstools']['sids']['enabled'] = None
if args.remove:
temp_pillar_dict = rem_from_sids(pillar_dict, 'enabled', args.sid_pattern)
else:
@@ -285,6 +291,9 @@ def add_rem_modify(args: dict):
pillar_dict = read_pillar(args.pillar)
if not sids_key_exists(pillar_dict, 'modify'):
pillar_dict['idstools']['sids']['modify'] = None
if args.remove:
temp_pillar_dict = rem_from_sids(pillar_dict, 'modify', string_val)
else: