From ed3b65b5326b00e47e1d7695b4d50df35b2340c2 Mon Sep 17 00:00:00 2001
From: John Dewey <john@dewey.ws>
Date: Sat, 7 Dec 2013 10:19:28 -0800
Subject: [PATCH] Made a few canges as I learn more about modules

* Set check_mode to False, am not supporting this ATM.
* Cleaned up delete/create() into single update().
* Return the group_id if created or found so tasks can be chained.
---
 cloud/nova_group | 36 ++++++++++++++++++++++--------------
 1 file changed, 22 insertions(+), 14 deletions(-)

diff --git a/cloud/nova_group b/cloud/nova_group
index 18e00c9c7ba..c96fb725ed5 100644
--- a/cloud/nova_group
+++ b/cloud/nova_group
@@ -157,14 +157,14 @@ class SecurityGroup(NovaGroup):
         self._name = module.params.get('name')
         self._description = module.params.get('description')
 
-    def exists(self):
+    def get(self):
         return self._get_secgroup(self._name)
 
     def create(self):
-        self._sg.create(self._name, self._description)
+        return self._sg.create(self._name, self._description)
 
     def delete(self):
-        self._sg.delete(self._name)
+        return self._sg.delete(self._name)
 
 
 class SecurityGroupRule(NovaGroup):
@@ -219,8 +219,6 @@ class SecurityGroupRule(NovaGroup):
         r = self._concat_security_group_rule(rule)
         if r in self._current_rules:
             return self._current_rules[r]
-        else:
-            return False
 
     def _validate_rules(self):
         for rule in self._rules:
@@ -260,6 +258,14 @@ class SecurityGroupRule(NovaGroup):
                 changed = True
         return changed
 
+    def update(self):
+        changed = False
+        if self.create():
+            changed = True
+        if self.delete():
+            changed = True
+        return changed
+
 
 def main():
     module = AnsibleModule(
@@ -274,7 +280,7 @@ def main():
             region_name=dict(default=None),
             state = dict(default='present', choices=['present', 'absent']),
         ),
-        supports_check_mode=True,
+        supports_check_mode=False,
     )
     login_username = module.params.get('login_username')
     login_password = module.params.get('login_password')
@@ -309,24 +315,26 @@ def main():
     rules = module.params.get('rules')
     state = module.params.get('state')
     security_group = SecurityGroup(nova, module)
-    security_group_rules = SecurityGroupRule(nova, module)
 
     changed = False
-    if security_group.exists():
+    group_id = None
+    group = security_group.get()
+    if group:
+        group_id = group.id
         if state == 'absent':
             security_group.delete()
             changed = True
     elif state == 'present':
-        security_group.create()
+        group = security_group.create()
         changed = True
+        group_id = group.id
 
-    if rules:
-        if security_group_rules.create():
-            changed = True
-        if security_group_rules.delete():
+    if rules is not None:
+        security_group_rules = SecurityGroupRule(nova, module)
+        if security_group_rules.update():
             changed = True
 
-    module.exit_json(changed=changed, group_id=None)
+    module.exit_json(changed=changed, group_id=group_id)
 
 # this is magic, see lib/ansible/module_common.py
 #<<INCLUDE_ANSIBLE_MODULE_COMMON>>