@@ -465,16 +465,14 @@ def __init__(self, module):
465
465
466
466
def validate_params (self ):
467
467
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' )):
470
469
return True
471
470
else :
472
471
self .module .fail_json (
473
472
msg = "When state is present either group_conf should be true or rules must be present but not both"
474
473
)
475
474
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' )):
478
476
return True
479
477
else :
480
478
self .module .fail_json (
@@ -490,6 +488,8 @@ def run(self):
490
488
force = self .params .get ("force" )
491
489
level = self .params .get ("level" )
492
490
rules = self .params .get ("rules" )
491
+ group = self .params .get ("group" )
492
+ group_conf = self .params .get ("group_conf" )
493
493
494
494
if level == "vm" :
495
495
vm = self .get_vm (vmid = self .params .get ('vmid' ))
@@ -510,27 +510,27 @@ def run(self):
510
510
rules_obj = firewall_obj ().rules
511
511
512
512
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 )
514
514
515
515
else :
516
516
firewall_obj = self .proxmox_api .cluster ().firewall
517
517
rules_obj = firewall_obj ().rules
518
518
519
519
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' ))
522
522
if rules is not None :
523
523
self .create_fw_rules (rules_obj = rules_obj , rules = rules , force = force )
524
524
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' ))
527
527
if rules is not None :
528
528
self .update_fw_rules (rules_obj = rules_obj , rules = rules , force = force )
529
529
elif state == "absent" :
530
530
if self .params .get ('pos' ):
531
531
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 )
534
534
else :
535
535
rules = self .get_fw_rules (rules_obj , pos = self .params .get ('pos' ))
536
536
groups = self .get_groups ()
0 commit comments