Skip to content

Commit 4b14774

Browse files
committed
proxmox_firewall: simplify conditions
1 parent 6327901 commit 4b14774

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

plugins/modules/proxmox_firewall.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -465,16 +465,14 @@ def __init__(self, module):
465465

466466
def validate_params(self):
467467
if self.params.get('state') in ['present', 'update']:
468-
if ((self.params.get('group_conf') and self.params.get('rules') is None) or
469-
(not self.params.get('group_conf') and self.params.get('rules') is not None)):
468+
if self.params.get('group_conf') != bool(self.params.get('rules')):
470469
return True
471470
else:
472471
self.module.fail_json(
473472
msg="When state is present either group_conf should be true or rules must be present but not both"
474473
)
475474
elif self.params.get('state') == 'absent':
476-
if ((self.params.get('group_conf') and self.params.get('pos') is None) or
477-
(not self.params.get('group_conf') and self.params.get('pos') is not None)):
475+
if self.params.get('group_conf') != bool(self.params.get('pos')):
478476
return True
479477
else:
480478
self.module.fail_json(
@@ -490,6 +488,8 @@ def run(self):
490488
force = self.params.get("force")
491489
level = self.params.get("level")
492490
rules = self.params.get("rules")
491+
group = self.params.get("group")
492+
group_conf = self.params.get("group_conf")
493493

494494
if level == "vm":
495495
vm = self.get_vm(vmid=self.params.get('vmid'))
@@ -510,27 +510,27 @@ def run(self):
510510
rules_obj = firewall_obj().rules
511511

512512
elif level == "group":
513-
rules_obj = getattr(self.proxmox_api.cluster().firewall().groups(), self.params.get('group'))
513+
rules_obj = getattr(self.proxmox_api.cluster().firewall().groups(), group)
514514

515515
else:
516516
firewall_obj = self.proxmox_api.cluster().firewall
517517
rules_obj = firewall_obj().rules
518518

519519
if state == "present":
520-
if self.params.get('group_conf'):
521-
self.create_group(group=self.params.get('group'), comment=self.params.get('comment'))
520+
if group_conf:
521+
self.create_group(group=group, comment=self.params.get('comment'))
522522
if rules is not None:
523523
self.create_fw_rules(rules_obj=rules_obj, rules=rules, force=force)
524524
elif state == "update":
525-
if self.params.get('group_conf'):
526-
self.create_group(group=self.params.get('group'), comment=self.params.get('comment'))
525+
if group_conf:
526+
self.create_group(group=group, comment=self.params.get('comment'))
527527
if rules is not None:
528528
self.update_fw_rules(rules_obj=rules_obj, rules=rules, force=force)
529529
elif state == "absent":
530530
if self.params.get('pos'):
531531
self.delete_fw_rule(rules_obj=rules_obj, pos=self.params.get('pos'))
532-
if self.params.get('group_conf'):
533-
self.delete_group(group_name=self.params.get('group'))
532+
if group_conf:
533+
self.delete_group(group_name=group)
534534
else:
535535
rules = self.get_fw_rules(rules_obj, pos=self.params.get('pos'))
536536
groups = self.get_groups()

0 commit comments

Comments
 (0)