Update Pure Storage FlashArray hostgroup module (#38503)

Allow for deletion and additon of hosts and volumes to a
volume group.

lib/ansible/modules/storage/purestorage/purefa_hg.py

@@ -41,12 +41,37 @@ extends_documentation_fragment:
 '''
 
 EXAMPLES = r'''
-- name: Create new hostgroup
+- name: Create empty hostgroup
   purefa_hg:
     hostgroup: foo
     fa_url: 10.10.10.2
     api_token: e31060a7-21fc-e277-6240-25983c6c4592
 
+- name: Add hosts and volumes to existing or new hostgroup
+  purefa_hg:
+    hostgroup: foo
+    host:
+      - host1
+      - host2
+    volume:
+      - vol1
+      - vol2
+    fa_url: 10.10.10.2
+    api_token: e31060a7-21fc-e277-6240-25983c6c4592
+
+- name: Delete hosts and volumes from hostgroup
+  purefa_hg:
+    hostgroup: foo
+    host:
+      - host1
+      - host2
+    volume:
+      - vol1
+      - vol2
+    fa_url: 10.10.10.2
+    api_token: e31060a7-21fc-e277-6240-25983c6c4592
+    state: absent
+
 # This will disconnect all hosts and volumes in the hostgroup
 - name: Delete hostgroup
   purefa_hg:
@@ -86,9 +111,9 @@ def get_hostgroup(module, array):
 
     hostgroup = None
 
-    for h in array.list_hgroups():
-        if h["name"] == module.params['hostgroup']:
-            hostgroup = h
+    for host in array.list_hgroups():
+        if host["name"] == module.params['hostgroup']:
+            hostgroup = host
             break
 
     return hostgroup
@@ -98,30 +123,79 @@ def make_hostgroup(module, array):
 
     changed = True
 
-    if not module.check_mode:
-        host = array.create_hgroup(module.params['hostgroup'])
-        if module.params['host']:
-            array.set_hgroup(module.params['hostgroup'], hostlist=module.params['host'])
-        if module.params['volume']:
-            for v in module.params['volume']:
-                array.connect_hgroup(module.params['hostgroup'], v)
+    try:
+        array.create_hgroup(module.params['hostgroup'])
+    except:
+        module.fail_json(msg='Failed to create hostgroup {0}'.format(module.params['hostgroup']))
+    if module.params['host']:
+        array.set_hgroup(module.params['hostgroup'], hostlist=module.params['host'])
+    if module.params['volume']:
+        for vol in module.params['volume']:
+            array.connect_hgroup(module.params['hostgroup'], vol)
     module.exit_json(changed=changed)
 
 
 def update_hostgroup(module, array):
     changed = False
-    hostgroup = module.params['hostgroup']
+    hgroup = get_hostgroup(module, array)
+    volumes = array.list_hgroup_connections(module.params['hostgroup'])
+    if module.params['state'] == "present":
+        if module.params['host']:
+            new_hosts = list(set(module.params['host']).difference(hgroup['hosts']))
+            if new_hosts:
+                try:
+                    array.set_hgroup(module.params['hostgroup'], addhostlist=new_hosts)
+                    changed = True
+                except:
+                    module.fail_josn(msg='Failed to add host(s) to hostgroup')
+        if module.params['volume']:
+            if volumes:
+                current_vols = [vol['vol'] for vol in volumes]
+                new_volumes = list(set(module.params['volume']).difference(set(current_vols)))
+                for cvol in new_volumes:
+                    try:
+                        array.connect_hgroup(module.params['hostgroup'], cvol)
+                        changed = True
+                    except:
+                        changed = False
+            else:
+                for cvol in module.params['volume']:
+                    try:
+                        array.connect_hgroup(module.params['hostgroup'], cvol)
+                        changed = True
+                    except:
+                        changed = False
+    else:
+        if module.params['host']:
+            old_hosts = list(set(module.params['host']).intersection(hgroup['hosts']))
+            if old_hosts:
+                try:
+                    array.set_hgroup(module.params['hostgroup'], remhostlist=old_hosts)
+                    changed = True
+                except:
+                    changed = False
+        if module.params['volume']:
+            old_volumes = list(set(module.params['volume']).difference(set([vol['name'] for vol in volumes])))
+            for cvol in old_volumes:
+                try:
+                    array.disconnect_hgroup(module.params['hostgroup'], cvol)
+                    changed = True
+                except:
+                    changed = False
+
     module.exit_json(changed=changed)
 
 
 def delete_hostgroup(module, array):
     changed = True
-    if not module.check_mode:
-        for vol in array.list_hgroup_connections(module.params['hostgroup']):
-            array.disconnect_hgroup(module.params['hostgroup'], vol["vol"])
-        host = array.get_hgroup(module.params['hostgroup'])
-        array.set_hgroup(module.params['hostgroup'], remhostlist=host['hosts'])
+    for vol in array.list_hgroup_connections(module.params['hostgroup']):
+        array.disconnect_hgroup(module.params['hostgroup'], vol["vol"])
+    host = array.get_hgroup(module.params['hostgroup'])
+    array.set_hgroup(module.params['hostgroup'], remhostlist=host['hosts'])
+    try:
         array.delete_hgroup(module.params['hostgroup'])
+    except:
+        module.fail_json(msg='Failed to delete hostgroup {0}'.format(module.params['hostgroup']))
     module.exit_json(changed=changed)
 
 
@@ -134,7 +208,7 @@ def main():
         volume=dict(type='list'),
     ))
 
-    module = AnsibleModule(argument_spec, supports_check_mode=True)
+    module = AnsibleModule(argument_spec, supports_check_mode=False)
 
     if not HAS_PURESTORAGE:
         module.fail_json(msg='purestorage sdk is required for this module in host')
@@ -145,20 +219,24 @@ def main():
 
     if module.params['host']:
         try:
-            for h in module.params['host']:
-                array.get_host(h)
+            for hst in module.params['host']:
+                array.get_host(hst)
         except:
-            module.fail_json(msg='Host not found')
+            module.fail_json(msg='Host {0} not found'.format(hst))
 
     if module.params['volume']:
         try:
-            for v in module.params['volume']:
-                array.get_volume(v)
+            for vol in module.params['volume']:
+                array.get_volume(vol)
         except:
-            module.fail_json(msg='Volume not found')
+            module.fail_json(msg='Volume {0} not found'.format(vol))
 
     if hostgroup and state == 'present':
         update_hostgroup(module, array)
+    elif hostgroup and module.params['volume'] and state == 'absent':
+        update_hostgroup(module, array)
+    elif hostgroup and module.params['host'] and state == 'absent':
+        update_hostgroup(module, array)
     elif hostgroup and state == 'absent':
         delete_hostgroup(module, array)
     elif hostgroup is None and state == 'absent':
@@ -166,6 +244,5 @@ def main():
     else:
         make_hostgroup(module, array)
 
-
 if __name__ == '__main__':
     main()