commit 2c2f25338406354730098c0c7d06cde44b3dff9a
author Scott Baker <smbaker@gmail.com> Tue Sep 29 09:23:35 2015 -0700
committer Scott Baker <smbaker@gmail.com> Tue Sep 29 09:23:35 2015 -0700
tree 467264ab5a0e93ce59aa587cc32a330c7d745ed6
parent b1ee663cb32887e92a035af59c42ef4fd8f3c443

update SiteDeployment with controller in Tosca recipe

xos/tosca/resources/xossite.py

diff --git a/xos/tosca/resources/xossite.py b/xos/tosca/resources/xossite.py
index 997c9e9..0db2705 100644
--- a/xos/tosca/resources/xossite.py
+++ b/xos/tosca/resources/xossite.py
@@ -51,11 +51,17 @@
                     if not controller_name:
                         raise Exception("Controller must be specified in SiteDeployment relationship")
 
-                    controller = self.get_xos_object(Controller, name=controller_name)
+                    controller = self.get_xos_object(Controller, name=controller_name, throw_exception=True)
 
                     existing_sitedeps = SiteDeployment.objects.filter(deployment=deployment, site=obj)
                     if existing_sitedeps:
-                        self.info("SiteDeployment from %s to %s already exists" % (str(obj), str(deployment)))
+                        sd = existing_sitedeps[0]
+                        if sd.controller != controller:
+                            sd.controller = controller
+                            sd.save()
+                            self.info("SiteDeployment from %s to %s updated controller" % (str(obj), str(deployment)))
+                        else:
+                            self.info("SiteDeployment from %s to %s already exists" % (str(obj), str(deployment)))
                     else:
                         sitedep = SiteDeployment(deployment=deployment, site=obj, controller=controller)
                         sitedep.save()